Update app.py

app.py

@@ -1,46 +1,50 @@
 # main.py
-import requests
-import tempfile
 import shutil
-import os
-from fastapi import FastAPI, Query
+import tempfile
+import requests
+from fastapi import FastAPI, Query, HTTPException
 from fastapi.responses import FileResponse
+import os
 from urllib.parse import urlparse
 
-app = FastAPI(title="Neon AI RawLink Proxy")
-
-TEMP_DIR = tempfile.gettempdir()
-
-def safe_filename(url: str) -> str:
-    """Generate a safe filename from URL."""
-    parsed = urlparse(url)
-    name = os.path.basename(parsed.path)
-    if not name.lower().endswith(".mp4"):
-        name += ".mp4"
-    return name
+app = FastAPI(title="Telegram-Friendly Video Proxy")
 
 @app.get("/download")
-def download_endpoint(url: str = Query(..., description="Catbox/Litter video URL")):
+def download_video(url: str = Query(..., description="Public video URL")):
     """
-    Accepts a video URL, downloads it temporarily, 
-    and serves it with proper headers so Telegram sees a download.
+    Downloads a video from a streaming/HTML-wrapped URL and serves it
+    as a raw file Telegram can accept.
     """
-    try:
-        filename = safe_filename(url)
-        tmp_path = os.path.join(TEMP_DIR, filename)
+    # Validate URL
+    parsed = urlparse(url)
+    if not parsed.scheme.startswith("http"):
+        raise HTTPException(status_code=400, detail="Invalid URL")
 
+    # Extract file extension for naming
+    _, ext = os.path.splitext(parsed.path)
+    if ext.lower() not in [".mp4", ".mkv", ".webm", ".mov", ".avi"]:
+        ext = ".mp4"  # default fallback
+
+    # Temp file to download
+    tmp_file = tempfile.mktemp(suffix=ext)
+
+    try:
         # Download the video
         with requests.get(url, stream=True, timeout=60) as r:
             r.raise_for_status()
-            with open(tmp_path, "wb") as f:
+            with open(tmp_file, "wb") as f:
                 shutil.copyfileobj(r.raw, f)
 
-        # Serve file with download headers
+        # Serve with Content-Disposition so Telegram sees it as a file
         return FileResponse(
-            tmp_path,
+            tmp_file,
             media_type="application/octet-stream",
-            filename=filename,
+            filename=f"video{ext}",
+            headers={"Content-Disposition": f'attachment; filename="video{ext}"'}
         )
 
     except Exception as e:
-        return {"ok": False, "error": str(e)}
+        # Clean up in case of failure
+        if os.path.exists(tmp_file):
+            os.remove(tmp_file)
+        raise HTTPException(status_code=500, detail=f"Failed to download file: {repr(e)}")