Update app.py

app.py

@@ -31,7 +31,7 @@ API_KEY = "sk-your-secret-key"
 # timeout لمنع التعليق الأبدي
 REQUEST_TIMEOUT = 45
 
-# إعادة المحاولة
+# retry خفيف
 MAX_RETRIES = 2
 
 
@@ -41,7 +41,7 @@ MAX_RETRIES = 2
 
 app = FastAPI(
     title="Universal AI Gateway",
-    version="4.1.0"
+    version="4.2.0"
 )
 
 
@@ -114,7 +114,7 @@ async def root():
     return {
         "status": "online",
         "service": "Universal AI Gateway",
-        "version": "4.1.0"
+        "version": "4.2.0"
     }
 
 
@@ -127,41 +127,79 @@ async def get_models():
 
     models_data = []
 
+    # =================================================
+    # MODELS THAT WORK WELL
+    # =================================================
+
     fallback_models = [
+
+        # GPT
         "gpt-4o-mini",
         "gpt-4o",
         "gpt-4",
         "gpt-3.5-turbo",
+
+        # Claude
         "claude-3-haiku",
+
+        # Llama
         "llama-3.1-70b",
+
+        # Mixtral
         "mixtral-8x7b",
+
+        # Deepseek
         "deepseek-chat",
-        "gemini-pro"
+
+        # Gemini
+        "gemini-pro",
+
+        # =================================================
+        # COHERE FAMILY
+        # =================================================
+
+        "command-r",
+        "command-r-plus",
+        "command-r7b",
+        "command",
+        "command-nightly",
+
+        # Additional Cohere-style names
+        "cohere-command-r",
+        "cohere-command-r-plus",
     ]
 
+    added_models = set()
+
     try:
 
         if hasattr(g4f.models, "_all_models"):
 
             all_models = list(g4f.models._all_models)
 
-            for model in all_models[:50]:
+            for model in all_models[:100]:
 
-                models_data.append({
-                    "id": str(model),
-                    "object": "model",
-                    "created": int(time.time()),
-                    "owned_by": "g4f"
-                })
+                model_name = str(model)
+
+                if model_name not in added_models:
+
+                    models_data.append({
+                        "id": model_name,
+                        "object": "model",
+                        "created": int(time.time()),
+                        "owned_by": "g4f"
+                    })
+
+                    added_models.add(model_name)
 
     except Exception as e:
 
         logger.error(f"Models error: {e}")
 
-    # fallback
-    if not models_data:
+    # fallback models
+    for model in fallback_models:
 
-        for model in fallback_models:
+        if model not in added_models:
 
             models_data.append({
                 "id": model,
@@ -170,6 +208,8 @@ async def get_models():
                 "owned_by": "g4f"
             })
 
+            added_models.add(model)
+
     return {
         "object": "list",
         "data": models_data
@@ -198,6 +238,7 @@ async def safe_completion(
 
             client = Client()
 
+            # timeout لمنع التعليق الأبدي
             response = await asyncio.wait_for(
                 asyncio.to_thread(
                     client.chat.completions.create,
@@ -208,6 +249,10 @@ async def safe_completion(
                 timeout=REQUEST_TIMEOUT
             )
 
+            logger.info(
+                f"Success | model={model}"
+            )
+
             return response
 
         except asyncio.TimeoutError:
@@ -215,7 +260,7 @@ async def safe_completion(
             last_error = "Request timeout"
 
             logger.warning(
-                f"Timeout attempt {attempt + 1}"
+                f"Timeout | model={model}"
             )
 
         except Exception as e:
@@ -223,7 +268,7 @@ async def safe_completion(
             last_error = e
 
             logger.warning(
-                f"Attempt failed {attempt + 1}: {e}"
+                f"Attempt failed {attempt + 1} | {e}"
             )
 
         await asyncio.sleep(1)
@@ -322,7 +367,7 @@ async def chat_completions(
                             f"Chunk error: {chunk_error}"
                         )
 
-                # provider لم يرسل أي شيء
+                # provider فتح stream بدون محتوى
                 if not has_content:
 
                     error_payload = {