Update main.py

main.py

@@ -7,6 +7,7 @@ import random
 from bs4 import BeautifulSoup
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import StreamingResponse
+from fake_useragent import UserAgent
 import uvicorn
 
 # Google Auth Libraries
@@ -19,6 +20,9 @@ app = FastAPI()
 ACCESS_KEY = os.getenv("ACCESS_KEY", "0000")
 SERVICE_ACCOUNT_JSON_STR = os.getenv("GOOGLE_SERVICE_ACCOUNT_JSON")
 
+# User-Agent Generator ကို တစ်ခါတည်း တည်ဆောက်ထားပါ
+ua = UserAgent()
+
 @app.get("/")
 def index():
     return {"message": "Proxy is Online!", "usage": "/download?url=[LINK]&key=[YOUR_KEY]"}
@@ -31,7 +35,7 @@ def get_all_service_accounts():
         data = json.loads(SERVICE_ACCOUNT_JSON_STR)
         if isinstance(data, list):
             return data
-        return [data] # တစ်ခုတည်းဆိုလျှင် list ထဲထည့်ပေးမည်
+        return [data]
     except Exception as e:
         print(f"JSON Parsing Error: {e}")
         return []
@@ -61,15 +65,27 @@ async def download_proxy(request: Request, url: str, key: str = None):
     # --- Google Drive မဟုတ်သော Link များအတွက် (MediaFire, Dropbox, etc.) ---
     if "drive.google.com" not in url:
         target_link = None
+        
+        # Headers အတွက် Random User-Agent ထုတ်ပေးခြင်း
+        current_headers = {
+            'User-Agent': ua.random,
+            'Accept-Language': 'en-US,en;q=0.9',
+        }
+
         if "mediafire.com" in url:
             try:
-                r = requests.get(url, headers={'User-Agent': 'Mozilla/5.0'}, timeout=10)
+                # MediaFire အတွက် Referer ထည့်ပေးခြင်းက ပိုစိတ်ချရသည်
+                current_headers['Referer'] = 'https://www.mediafire.com/'
+                r = requests.get(url, headers=current_headers, timeout=10)
                 soup = BeautifulSoup(r.text, 'html.parser')
                 btn = soup.find('a', {'id': 'downloadButton'})
                 target_link = btn.get('href') if btn else None
-            except: pass
+            except Exception as e:
+                print(f"MediaFire Error: {e}")
+        
         elif "dropbox.com" in url:
             target_link = url.replace("?dl=0", "").split("?")[0] + "?dl=1"
+            
         elif "1drv.ms" in url or "onedrive.live.com" in url:
             encoded_url = base64.b64encode(bytes(url, 'utf-8')).decode('utf-8').replace('=', '').replace('/', '_').replace('+', '-')
             target_link = f"https://api.onedrive.com/v1.0/shares/u!{encoded_url}/root/content"
@@ -85,9 +101,8 @@ async def download_proxy(request: Request, url: str, key: str = None):
         raise HTTPException(status_code=400, detail="Invalid Google Drive Link")
 
     accounts = get_all_service_accounts()
-    random.shuffle(accounts) # အကောင့်များကို တစ်လှည့်စီဖြစ်အောင် နေရာပြောင်းသည်
+    random.shuffle(accounts)
 
-    # အကောင့်တစ်ခုချင်းစီဖြင့် ဒေါင်းရန် ကြိုးစားကြည့်မည်
     for account in accounts:
         token = get_token_for_account(account)
         if not token: continue
@@ -99,22 +114,19 @@ async def download_proxy(request: Request, url: str, key: str = None):
 
         r = requests.get(api_link, headers=headers, stream=True, allow_redirects=True)
         
-        # အကယ်၍ အောင်မြင်ပါက (200 သို့မဟုတ် 206 Partial Content) ပြန်ပို့မည်
         if r.status_code in [200, 206]:
             return process_response(r)
         
-        # 403 သို့မဟုတ် 404 ဖြစ်ပါက နောက်အကောင့်တစ်ခုဖြင့် ထပ်စမ်းမည်
         print(f"Account {account.get('client_email')} failed with {r.status_code}. Trying next...")
 
-    # --- အကောင့်အားလုံးမရပါက (သို့မဟုတ် အကောင့်မရှိပါက) Public Fallback လုပ်မည် ---
+    # --- Public Fallback ---
     public_url = f"https://drive.google.com/uc?export=download&id={file_id}"
-    headers = {}
+    headers = {'User-Agent': ua.random} # Public link မှာလည်း UA သုံးခြင်း
     if request.headers.get('range'):
         headers['Range'] = request.headers.get('range')
         
     r = requests.get(public_url, headers=headers, stream=True, allow_redirects=True)
     
-    # Virus Warning ကျော်ခြင်း
     if "text/html" in r.headers.get('Content-Type', '').lower():
         confirm_token = None
         for k, v in r.cookies.items():
@@ -129,7 +141,14 @@ async def download_proxy(request: Request, url: str, key: str = None):
     raise HTTPException(status_code=r.status_code, detail="All service accounts and public download failed.")
 
 async def stream_file(target_url, range_header):
-    headers = {'Range': range_header} if range_header else {}
+    # ဖိုင်ကို တိုက်ရိုက်ဆွဲထုတ်ရာတွင်လည်း Identity မျိုးစုံ သုံးနိုင်ရန်
+    headers = {
+        'User-Agent': ua.random,
+        'Referer': 'https://www.mediafire.com/' if "mediafire" in target_url else ''
+    }
+    if range_header:
+        headers['Range'] = range_header
+        
     r = requests.get(target_url, headers=headers, stream=True, allow_redirects=True)
     if r.status_code >= 400:
         raise HTTPException(status_code=r.status_code, detail="External link fetch failed")