Update main.py

main.py

@@ -25,7 +25,7 @@ ua = UserAgent(fallback='Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/5
 
 # Global Async Client
 client = httpx.AsyncClient(
-    timeout=httpx.Timeout(30.0, read=None), # Timeout ကို 30 စက္ကန့်အထိ တိုးပေးထားပါတယ်
+    timeout=httpx.Timeout(30.0, read=None), 
     follow_redirects=True,
     limits=httpx.Limits(max_connections=100, max_keepalive_connections=20)
 )
@@ -45,7 +45,6 @@ async def get_token_for_account(cred_dict):
     try:
         scopes = ['https://www.googleapis.com/auth/drive.readonly']
         creds = service_account.Credentials.from_service_account_info(cred_dict, scopes=scopes)
-        # Token refresh ကို thread ထဲမှာ run ပါတယ်
         loop = asyncio.get_event_loop()
         auth_req = google.auth.transport.requests.Request()
         await loop.run_in_executor(None, creds.refresh, auth_req)
@@ -76,14 +75,16 @@ async def download_proxy(request: Request, url: str, key: str = None):
                 headers['Referer'] = 'https://www.mediafire.com/'
                 r = await client.get(url, headers=headers)
                 
-                # Regex ဖြင့် Direct Link ကို ရှာဖွေခြင်း (ပိုမိုတိကျစေရန် ပြင်ထားသည်)
+                # Regex ဖြင့် Direct Link ကို ရှာဖွေခြင်း
                 match = re.search(r'href="(https?://[a-zA-Z0-9-]+\.mediafire\.com/download/[^"]+)"', r.text)
                 if match:
                     target_link = match.group(1)
                 else:
                     soup = BeautifulSoup(r.text, 'html.parser')
                     btn = soup.find('a', {'id': 'downloadButton'})
-                    target_link = btn.get('href') if btn else None
+                    if btn and btn.get('href'):
+                        # ပေးလာတဲ့ href မှာ https:// မပါရင် မူလ URL နဲ့ ပေါင်းပြီး အပြည့်အစုံဖြစ်အောင် ပြုလုပ်ခြင်း
+                        target_link = urllib.parse.urljoin(url, btn.get('href'))
             except Exception as e:
                 print(f"MediaFire Error: {e}")
         
@@ -140,24 +141,20 @@ async def stream_file(target_url, range_header):
         raise HTTPException(status_code=500, detail=str(e))
 
 async def process_response(r):
-    # မူလလာတဲ့ content-disposition ကို ဖယ်ထုတ်ဖို့ excluded ထဲ ထည့်ပါမယ်
     excluded = ['content-encoding', 'content-length', 'transfer-encoding', 'connection', 'content-disposition']
     response_headers = {n: v for n, v in r.headers.items() if n.lower() not in excluded}
     
-    # Video Seeking အတွက် မရှိမဖြစ်လိုအပ်သော Headers များ
     response_headers['Accept-Ranges'] = 'bytes'
     if 'Content-Length' in r.headers:
         response_headers['Content-Length'] = r.headers['Content-Length']
     if 'Content-Range' in r.headers:
         response_headers['Content-Range'] = r.headers['Content-Range']
 
-    # Direct Download ဖြစ်စေရန်
     response_headers['Content-Disposition'] = 'attachment'
 
-    # Connection များကို Memory Leak မဖြစ်စေဘဲ သေချာပိတ်ပေးမည့် Generator
     async def stream_generator():
         try:
-            async for chunk in r.aiter_bytes(chunk_size=131072): # 128KB chunks
+            async for chunk in r.aiter_bytes(chunk_size=131072):
                 yield chunk
         finally:
             await r.aclose()