Update main.py

main.py

@@ -3,6 +3,7 @@ import re
 import json
 import base64
 import requests
+import random  # အကောင့်တစ်ခုထက်ပိုပါက တစ်လှည့်စီရွေးရန်
 from bs4 import BeautifulSoup
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import StreamingResponse
@@ -16,6 +17,7 @@ app = FastAPI()
 
 # --- Configurations ---
 ACCESS_KEY = os.getenv("ACCESS_KEY", "786969")
+# ဤနေရာတွင် JSON တစ်ခုတည်းဖြစ်စေ၊ [{}, {}] ပုံစံဖြင့် Array ဖြစ်စေ ထည့်နိုင်သည်
 SERVICE_ACCOUNT_JSON_STR = os.getenv("GOOGLE_SERVICE_ACCOUNT_JSON")
 
 @app.get("/")
@@ -23,14 +25,31 @@ def index():
     return {"message": "Proxy is Online!", "usage": "/download?url=[LINK]&key=[YOUR_KEY]"}
 
 def get_google_access_token():
+    """Service Account(s) မှ Access Token ကို Random ရယူခြင်း"""
     if not SERVICE_ACCOUNT_JSON_STR:
+        print("Missing GOOGLE_SERVICE_ACCOUNT_JSON credential!")
         return None
+    
     try:
-        creds_dict = json.loads(SERVICE_ACCOUNT_JSON_STR)
+        # Secret ထဲမှ Data ကို ဖတ်ခြင်း
+        creds_data = json.loads(SERVICE_ACCOUNT_JSON_STR)
+        
+        # အကယ်၍ JSON Array (List) ဖြစ်နေပါက တစ်ခုကို Random ရွေးမည်
+        if isinstance(creds_data, list):
+            selected_cred = random.choice(creds_data)
+        else:
+            selected_cred = creds_data
+
         scopes = ['https://www.googleapis.com/auth/drive.readonly']
-        creds = service_account.Credentials.from_service_account_info(creds_dict, scopes=scopes)
+        creds = service_account.Credentials.from_service_account_info(
+            selected_cred,
+            scopes=scopes
+        )
+        
+        # Token ရယူခြင်း
         auth_req = google.auth.transport.requests.Request()
         creds.refresh(auth_req)
+        
         return creds.token
     except Exception as e:
         print(f"Service Account Auth Error: {e}")
@@ -45,7 +64,7 @@ def get_direct_url(url):
         file_id = fid.group(1)
         
         access_token = get_google_access_token()
-        # ပထမဦးစွာ Service Account Link ကို ပြန်ပေးမည်
+        # API v3 Link
         target = f"https://www.googleapis.com/drive/v3/files/{file_id}?alt=media"
         return target, access_token, file_id
         
@@ -85,36 +104,29 @@ async def download_proxy(request: Request, url: str, key: str = None):
 
     session = requests.Session()
     
-    # ၁။ Service Account ဖြင့် အရင်ကြိုးစားကြည့်မည်
+    # ၁။ Service Account ဖြင့် ကြိုးစားခြင်း
     r = session.get(target_link, headers=headers, stream=True, allow_redirects=True)
     
-    # ၂။ အကယ်၍ 404 တက်ခဲ့လျှင် (Public File ဖြစ်နိုင်သဖြင့်) Fallback လုပ်မည်
+    # ၂။ 404 ဖြစ်ပါက Public Fallback လုပ်ခြင်း
     if r.status_code == 404 and "drive.google.com" in url and file_id:
-        print(f"SA 404. Trying Public Fallback for: {file_id}")
-        
-        # Public Link သုံးမည်ဖြစ်၍ Authorization ကို ဖယ်ထုတ်ရပါမည်
         public_headers = {k: v for k, v in headers.items() if k.lower() != 'authorization'}
         public_url = f"https://drive.google.com/uc?export=download&id={file_id}"
         
-        # Public Link ကို စမ်းခေါ်ခြင်း
         r = session.get(public_url, headers=public_headers, stream=True, allow_redirects=True)
         
-        # GDrive 100MB+ Virus Scan Confirm ကျော်ရန် Logic
+        # Virus Warning ကျော်ခြင်း
         if "text/html" in r.headers.get('Content-Type', '').lower():
             confirm_token = None
             for k, v in session.cookies.items():
                 if k.startswith("download_warning"):
                     confirm_token = v
                     break
-            
             if confirm_token:
                 r = session.get(f"{public_url}&confirm={confirm_token}", headers=public_headers, stream=True)
 
-    # Error ဆက်ရှိနေသေးပါက
     if r.status_code >= 400:
-        raise HTTPException(status_code=r.status_code, detail=f"Failed to fetch file: {r.status_code}")
+        raise HTTPException(status_code=r.status_code, detail=f"Request failed with status {r.status_code}")
 
-    # Header ပြန်ပို့ရန် ပြင်ဆင်ခြင်း
     excluded = ['content-encoding', 'content-length', 'transfer-encoding', 'connection']
     response_headers = {n: v for n, v in r.headers.items() if n.lower() not in excluded}
     response_headers['Accept-Ranges'] = 'bytes'