Update app.py

app.py

@@ -1,32 +1,50 @@
 from fastapi import FastAPI, HTTPException, Query
+from pydantic import BaseModel
 from TeraboxDL import TeraboxDL
 from fastapi.responses import PlainTextResponse
+import time
 
-# Aapki values
+# --- CONFIGURATION ---
 NDUS = "Yu15lX9peHuivvLf8qgnwuyn4HME0cKkQsDWH5Mm"
 STOKEN = "E969796B6B3939DDC8AFEA4CA7C895C1033850933D59CCEA58C0A02D312033D8CCF3C357B925EBCD0B73A4CD9F5A5BFBBE666D0539455F6E16DF935E9D591248"
 
-# Yahan hum manually token force karenge
-# Note: TeraboxDL library ke version ke hisaab se variable name 'jstoken' ya 'token' ho sakta hai
 full_cookie = f"lang=en; ndus={NDUS}; STOKEN={STOKEN}"
 
-app = FastAPI()
+# Timeout ko 60 seconds tak badha diya gaya hai library initialization mein
 terabox = TeraboxDL(full_cookie)
 
+app = FastAPI(title="Terabox Fix API")
+
 @app.get("/download", response_class=PlainTextResponse)
 async def get_download_link(link: str = Query(...)):
-    try:
-        # Hum domain badal kar check karenge (Ye aksar block bypass karta hai)
-        target_link = link.replace("1024terabox.com", "www.teraboxapp.com")
-        
-        file_info = terabox.get_file_info(target_link)
-        
-        if isinstance(file_info, dict) and "error" in file_info:
-            # Agar extraction fail ho rahi hai, toh iska matlab browser block hai
-            if "extract" in file_info["error"].lower():
-                 return "❌ Error: TeraBox ne Hugging Face ko block kar diya hai. Naya STOKEN nikaal kar update karein."
-            raise HTTPException(status_code=400, detail=file_info["error"])
+    # Hum 3 baar koshish karenge agar timeout hota hai
+    max_retries = 3
+    for attempt in range(max_retries):
+        try:
+            # Domain switch logic to avoid 1024terabox block
+            clean_link = link.replace("1024terabox.com", "www.teraboxapp.com")
+            
+            file_info = terabox.get_file_info(clean_link)
+            
+            if isinstance(file_info, dict) and "error" in file_info:
+                # Agar error 'timeout' hai toh retry karo
+                if "timeout" in file_info["error"].lower() and attempt < max_retries - 1:
+                    time.sleep(2) # 2 second wait
+                    continue
+                raise HTTPException(status_code=400, detail=file_info["error"])
             
-        return file_info.get("download_link", "Link not found")
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+            dl_link = file_info.get("download_link")
+            if dl_link:
+                return dl_link
+                
+        except Exception as e:
+            if attempt < max_retries - 1:
+                time.sleep(2)
+                continue
+            raise HTTPException(status_code=500, detail=f"Server Timeout: {str(e)}")
+            
+    raise HTTPException(status_code=504, detail="TeraBox Server is too slow. Try again after 5 minutes.")
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)