Update main.py

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, Query, Header
+from fastapi import FastAPI, HTTPException, Query, Request
 from fastapi.responses import StreamingResponse
 import requests
 from bs4 import BeautifulSoup
@@ -6,14 +6,12 @@ import re
 import base64
 
 app = FastAPI()
+ACCESS_KEY = "969786"
 
 @app.get("/")
 def index():
     return {"message": "Proxy is Online!", "usage": "/download?url=[LINK]&key=[YOUR_KEY]"}
 
-# ကိုယ်တိုင်သုံးမည့် Password သတ်မှတ်ရန်
-ACCESS_KEY = "969786" 
-
 def get_direct_url(url):
     if "drive.google.com" in url:
         fid = re.search(r'/(?:d|file/d|open\?id=)/([a-zA-Z0-9_-]+)', url)
@@ -33,30 +31,42 @@ def get_direct_url(url):
     return None
 
 @app.get("/download")
-def download_proxy(url: str, key: str = None):
-    # Password စစ်ဆေးခြင်း
+async def download_proxy(request: Request, url: str, key: str = None):
     if key != ACCESS_KEY:
-        raise HTTPException(status_code=403, detail="Access Denied: Invalid Key")
+        raise HTTPException(status_code=403, detail="Access Denied")
 
     target_link = get_direct_url(url)
     if not target_link:
-        raise HTTPException(status_code=400, detail="Unsupported or Invalid Link")
-
-    try:
-        r = requests.get(target_link, stream=True, allow_redirects=True)
-        file_name = "downloaded_file"
-        if "content-disposition" in r.headers:
-            fname_match = re.findall('filename="(.+)"', r.headers['content-disposition'])
-            if fname_match: file_name = fname_match[0].strip('"')
-
-        def iterfile():
-            for chunk in r.iter_content(chunk_size=1024*1024):
-                if chunk: yield chunk
-
-        return StreamingResponse(iterfile(), media_type="application/octet-stream", 
-                                 headers={"Content-Disposition": f"attachment; filename={file_name}"})
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=400, detail="Invalid Link")
+
+    # Client ဆီကလာတဲ့ Range Header ကို ရယူခြင်း (Video ကျော်ကြည့်ရန်အတွက်)
+    range_header = request.headers.get('range')
+    headers = {}
+    if range_header:
+        headers['Range'] = range_header
+
+    # Target Server (MediaFire/GDrive) ဆီကို Range Header နဲ့ လှမ်းတောင်းခြင်း
+    r = requests.get(target_link, headers=headers, stream=True, allow_redirects=True)
+    
+    # Header များကို ပြန်လည်ပေးပို့ရန် ပြင်ဆင်ခြင်း
+    excluded_headers = ['content-encoding', 'content-length', 'transfer-encoding', 'connection']
+    response_headers = {name: value for name, value in r.headers.items() if name.lower() not in excluded_headers}
+    
+    # Video ကျော်ကြည့်နိုင်ရန် 'Accept-Ranges': 'bytes' ထည့်ပေးရပါမည်
+    response_headers['Accept-Ranges'] = 'bytes'
+    if 'Content-Length' in r.headers:
+        response_headers['Content-Length'] = r.headers['Content-Length']
+
+    def iterfile():
+        for chunk in r.iter_content(chunk_size=1024*1024):
+            if chunk: yield chunk
+
+    return StreamingResponse(
+        iterfile(),
+        status_code=r.status_code,
+        headers=response_headers,
+        media_type=r.headers.get('Content-Type', 'application/octet-stream')
+    )
 
 if __name__ == "__main__":
     import uvicorn