Update app.py

app.py

@@ -9,18 +9,11 @@ import uvicorn
 from g4f.client import Client
 from g4f.Provider import __providers__
 
-# =========================================================
-# CONFIG
-# =========================================================
-
 MODEL = "gpt-4o-mini"
+
 MAX_RETRIES = 10
 TIMEOUT = 60
 
-# =========================================================
-# FILTER SAFE PROVIDERS
-# =========================================================
-
 SAFE_PROVIDERS = []
 
 for provider in __providers__:
@@ -29,27 +22,21 @@ for provider in __providers__:
 
         name = provider.__name__.lower()
 
-        # must work
         if not getattr(provider, "working", False):
             continue
 
-        # skip auth providers
         if getattr(provider, "needs_auth", False):
             continue
 
-        # skip browser providers
         if getattr(provider, "use_nodriver", False):
             continue
 
-        # blocked providers
         blocked = [
             "openai",
             "qwen",
             "copilot",
             "gemini",
             "claude",
-            "deepseek",
-            "perplexity",
         ]
 
         if any(x in name for x in blocked):
@@ -64,15 +51,8 @@ random.shuffle(SAFE_PROVIDERS)
 
 print(f"[+] SAFE PROVIDERS: {len(SAFE_PROVIDERS)}")
 
-# =========================================================
-# CLIENT
-# =========================================================
-
 client = Client()
 
-# =========================================================
-# SMART SYSTEM
-# =========================================================
 
 class SmartG4F:
 
@@ -81,14 +61,10 @@ class SmartG4F:
         self.good = []
         self.bad = {}
 
-    # -----------------------------------------------------
-
     def mark_bad(self, provider, cooldown=300):
 
         self.bad[provider.__name__] = time.time() + cooldown
 
-    # -----------------------------------------------------
-
     def is_bad(self, provider):
 
         expire = self.bad.get(provider.__name__)
@@ -102,8 +78,6 @@ class SmartG4F:
 
         return True
 
-    # -----------------------------------------------------
-
     def provider_pool(self):
 
         providers = []
@@ -119,8 +93,6 @@ class SmartG4F:
 
         return providers
 
-    # -----------------------------------------------------
-
     def ask(self, prompt):
 
         errors = []
@@ -156,6 +128,9 @@ class SmartG4F:
 
                 text = response.choices[0].message.content
 
+                if not text:
+                    raise Exception("Empty response")
+
                 if provider not in self.good:
                     self.good.append(provider)
 
@@ -176,40 +151,23 @@ class SmartG4F:
                     "error": str(e)
                 })
 
-                # 429
                 if "429" in err:
 
-                    self.mark_bad(
-                        provider,
-                        cooldown=600
-                    )
-
+                    self.mark_bad(provider, 600)
                     time.sleep(random.randint(10, 20))
 
-                # cloudflare
                 elif "cloudflare" in err:
 
-                    self.mark_bad(
-                        provider,
-                        cooldown=1200
-                    )
-
+                    self.mark_bad(provider, 1200)
                     time.sleep(20)
 
-                # timeout
                 elif "timeout" in err:
 
-                    self.mark_bad(
-                        provider,
-                        cooldown=300
-                    )
+                    self.mark_bad(provider, 300)
 
                 else:
 
-                    self.mark_bad(
-                        provider,
-                        cooldown=180
-                    )
+                    self.mark_bad(provider, 180)
 
                 continue
 
@@ -218,24 +176,15 @@ class SmartG4F:
             "errors": errors
         }
 
-# =========================================================
-# FASTAPI
-# =========================================================
 
 app = FastAPI()
 
 ai = SmartG4F()
 
-# =========================================================
-# REQUEST MODEL
-# =========================================================
 
 class Query(BaseModel):
     prompt: str
 
-# =========================================================
-# ROUTES
-# =========================================================
 
 @app.get("/")
 async def home():
@@ -245,16 +194,12 @@ async def home():
         "providers": len(SAFE_PROVIDERS)
     }
 
-# ---------------------------------------------------------
 
 @app.post("/ask")
 async def ask(query: Query):
 
     return ai.ask(query.prompt)
 
-# =========================================================
-# START
-# =========================================================
 
 if __name__ == "__main__":