Update main.py

main.py

@@ -28,15 +28,13 @@ ua = UserAgent(fallback='Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/5
 MEDIAFIRE_CACHE = {}
 CACHE_TTL = 1800 
 
-# တစ်ချိန်တည်းမှာ MediaFire ဆီ သွားမယ့် Request အရေအတွက်ကို ကန့်သတ်ရန် (Stability အတွက်)
-MAX_CONCURRENT_REQUESTS = 20
+MAX_CONCURRENT_REQUESTS = 30
 semaphore = asyncio.Semaphore(MAX_CONCURRENT_REQUESTS)
 
-# Global Client with optimized pooling
 client = httpx.AsyncClient(
     timeout=httpx.Timeout(30.0, read=None), 
     follow_redirects=True,
-    limits=httpx.Limits(max_connections=500, max_keepalive_connections=50)
+    limits=httpx.Limits(max_connections=500, max_keepalive_connections=100)
 )
 
 @app.get("/")
@@ -80,13 +78,26 @@ async def scrape_mediafire(url):
             r = await temp_client.get(url)
             if r.status_code != 200: return None
             
-            # Regex patterns
+            # Direct link search
+            target = None
             match = re.search(r'https?://download[^\s"\']+mediafire\.com/[^\s"\']+', r.text)
-            if match: return match.group(0).strip().replace("'", "").replace('"', '')
+            if match: 
+                target = match.group(0).strip().replace("'", "").replace('"', '')
             
-            soup = BeautifulSoup(r.text, 'html.parser')
-            btn = soup.find('a', {'id': 'downloadButton'}) or soup.find('a', {'aria-label': re.compile(r'Download', re.I)})
-            if btn and btn.get('href'): return btn.get('href')
+            if not target:
+                soup = BeautifulSoup(r.text, 'html.parser')
+                btn = soup.find('a', {'id': 'downloadButton'}) or soup.find('a', {'aria-label': re.compile(r'Download', re.I)})
+                if btn: target = btn.get('href')
+
+            # --- URL Fix Section ---
+            if target:
+                # အကယ်၍ // နဲ့စရင် https: ထည့်မယ်
+                if target.startswith("//"): 
+                    target = f"https:{target}"
+                # အကယ်၍ /file/... လို့ Relative path နဲ့လာရင် MediaFire domain ထည့်ပေးမယ်
+                elif target.startswith("/"):
+                    target = f"https://www.mediafire.com{target}"
+                return target
     except: pass
     return None
 
@@ -102,14 +113,13 @@ async def download_proxy(request: Request, url: str, key: str = None):
 
     # --- MediaFire Section ---
     if "mediafire.com" in clean_url:
-        async with semaphore: # Request တွေကို Queue စနစ်နဲ့ ထိန်းချုပ်ခြင်း
+        async with semaphore:
             cached = MEDIAFIRE_CACHE.get(clean_url)
             target_link = cached['link'] if (cached and (current_time - cached['time']) < CACHE_TTL) else None
             
             if not target_link:
                 target_link = await scrape_mediafire(clean_url)
                 if target_link:
-                    if target_link.startswith("//"): target_link = f"https:{target_link}"
                     MEDIAFIRE_CACHE[clean_url] = {'link': target_link, 'time': current_time}
 
             if not target_link:
@@ -155,6 +165,10 @@ async def download_proxy(request: Request, url: str, key: str = None):
         return await stream_file(clean_url, range_header, filename)
 
 async def stream_file(target_url, range_header, filename, referer=None):
+    # Ensure full URL for httpx
+    if not target_url.startswith("http"):
+        raise HTTPException(status_code=500, detail=f"Malformed URL: {target_url}")
+
     headers = {'User-Agent': ua.random}
     if range_header: headers['Range'] = range_header
     if referer: headers['Referer'] = referer