Update main.py

main.py

@@ -71,13 +71,12 @@ async def download_proxy(request: Request, url: str, key: str = None):
     # --- MediaFire ---
     if "mediafire.com" in clean_url:
         try:
-            # Player ကလာတဲ့ Request ကို Browser ကလာသလို အသွင်ဖျက်ရန်
             browser_headers = {
                 'User-Agent': ua.random,
                 'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8',
                 'Accept-Language': 'en-US,en;q=0.5',
+                'Referer': 'https://www.mediafire.com/',
                 'Connection': 'keep-alive',
-                'Upgrade-Insecure-Requests': '1',
             }
             
             r = await client.get(clean_url, headers=browser_headers)
@@ -85,27 +84,33 @@ async def download_proxy(request: Request, url: str, key: str = None):
                 raise HTTPException(status_code=r.status_code, detail="MediaFire page access denied")
 
             target_link = None
-            # Method 1: Regex for Direct Download Link
+            # Method 1: Regex
             match = re.search(r"https?://download[0-9]+\.mediafire\.com/[a-zA-Z0-9_-]+/[a-zA-Z0-9_-]+/[^\s'\"]+", r.text)
             if match:
                 target_link = match.group(0).replace('"', '').replace("'", "")
             
-            # Method 2: BS4 Fallback
+            # Method 2: BS4
             if not target_link:
                 soup = BeautifulSoup(r.text, 'html.parser')
                 btn = soup.find('a', {'id': 'downloadButton'})
                 if btn and btn.get('href'):
                     target_link = btn.get('href')
 
+            # --- အရေးကြီးသောပြင်ဆင်ချက်: Relative URL ကို Absolute URL သို့ပြောင်းခြင်း ---
+            if target_link:
+                if target_link.startswith("//"):
+                    target_link = f"https:{target_link}"
+                elif target_link.startswith("/"):
+                    target_link = f"https://www.mediafire.com{target_link}"
+            
             if not target_link:
                 raise HTTPException(status_code=404, detail="Direct link extraction failed")
             
-            # Direct Link ကို Stream လုပ်တဲ့အခါ Referer ထည့်ပေးခြင်းဖြင့် 500 Error ကို ကျော်လွှားနိုင်သည်
             return await stream_file(target_link, range_header, referer=clean_url)
             
         except Exception as e:
             print(f"MediaFire Error: {str(e)}")
-            raise HTTPException(status_code=500, detail=f"Internal Server Error: {str(e)}")
+            raise HTTPException(status_code=500, detail=str(e))
 
     # --- Google Drive ---
     elif "drive.google.com" in clean_url:
@@ -139,25 +144,27 @@ async def download_proxy(request: Request, url: str, key: str = None):
 async def stream_file(target_url, range_header, referer=None):
     headers = {'User-Agent': ua.random}
     if range_header: headers['Range'] = range_header
-    if referer: headers['Referer'] = referer # MediaFire အတွက် အရေးကြီးသည်
+    if referer: headers['Referer'] = referer
     
     try:
         req = client.build_request("GET", target_url, headers=headers)
         r = await client.send(req, stream=True)
-        # အကယ်၍ မူရင်း Server က 403/500 ပြန်လာလျှင် Proxy ကနေ Error ပြန်ရန်
         if r.status_code >= 400:
             await r.aclose()
-            raise HTTPException(status_code=r.status_code, detail="Remote server rejected the request")
+            raise HTTPException(status_code=r.status_code, detail=f"Remote server returned {r.status_code}")
         return await process_response(r)
     except Exception as e:
+        print(f"Stream Error: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
 async def process_response(r):
+    # Player များအတွက် မရှိမဖြစ်လိုအပ်သော Header များ
     safe_headers = [
         'content-type', 'content-length', 'content-range', 
         'accept-ranges', 'content-disposition', 'cache-control'
     ]
     response_headers = {n: v for n, v in r.headers.items() if n.lower() in safe_headers}
+    
     if 'accept-ranges' not in [k.lower() for k in response_headers.keys()]:
         response_headers['Accept-Ranges'] = 'bytes'