Update app.py

app.py

@@ -74,15 +74,7 @@ def get_provider(name: str):
         return None
 
 REAL_PROVIDERS: Dict[str, Any] = {
-    "Perplexity": get_provider("Perplexity") or get_provider("PerplexityAi"),
-    "Copilot": get_provider("Copilot"),
     "Qwen": get_provider("Qwen"),
-    "Blackbox": get_provider("Blackbox"),
-    "DeepSeek": get_provider("DeepSeek"),
-    "HuggingFace": get_provider("HuggingFace"),
-    "Cloudflare": get_provider("Cloudflare"),
-    "You": get_provider("You"),
-    "Bing": get_provider("Bing"),
 }
 
 REAL_PROVIDERS = {k: v for k, v in REAL_PROVIDERS.items() if v}
@@ -91,15 +83,7 @@ REAL_PROVIDERS = {k: v for k, v in REAL_PROVIDERS.items() if v}
 # MODELS
 # =====================================================
 PROVIDER_MODELS_FALLBACK: Dict[str, List[str]] = {
-    "Perplexity": ["sonar", "sonar-pro", "gpt-4o", "llama-3"],
-    "Copilot": ["gpt-4o", "gpt-4", "turbo"],
     "Qwen": ["Qwen/Qwen3.6-35B-A3B"],
-    "Blackbox": ["gpt-4o", "claude-3", "gemini-pro", "llama-3"],
-    "DeepSeek": ["deepseek-chat", "deepseek-coder"],
-    "HuggingFace": ["meta-llama", "mistral", "qwen", "gemma"],
-    "Cloudflare": ["llama-3", "mistral", "qwen"],
-    "You": ["gpt-4o", "claude", "llama-3"],
-    "Bing": ["gpt-4o", "gpt-4"],
 }
 
 CLAUDE_MODEL_MAP: Dict[str, str] = {
@@ -167,7 +151,7 @@ def discover_provider_models(provider_obj: Any, provider_name: str) -> List[str]
     if not candidates:
         candidates = PROVIDER_MODELS_FALLBACK.get(
             provider_name,
-            ["gpt-4o"]
+            ["Qwen/Qwen3.6-35B-A3B"]
         )
 
     seen = set()
@@ -250,8 +234,9 @@ def ask(message: str, history, provider_name: str, model_name: str):
         yield ""
         return
 
-    # تحويل نماذج Claude إلى Qwen3.6-35B-A3B
-    model_name = CLAUDE_MODEL_MAP.get(model_name, model_name)
+    # تحويل نماذج Claude الوهمية إلى Qwen3.6-35B-A3B
+    model_name = CLAUDE_MODEL_MAP.get(model_name, "Qwen/Qwen3.6-35B-A3B")
+    provider_name = "Qwen"
 
     key = f"{provider_name}|{model_name}|{message}"
 
@@ -297,70 +282,49 @@ def ask(message: str, history, provider_name: str, model_name: str):
         "content": message
     })
 
-    fallback_providers = [
-        provider_name,
-        "Perplexity",
-        "Copilot",
-        "Blackbox",
-        "DeepSeek",
-        "Bing",
-        "You",
-        "Qwen"
-    ]
-
-    used = []
-
-    for pname in fallback_providers:
-        if pname in used:
-            continue
+    pobj = REAL_PROVIDERS.get("Qwen")
 
-        used.append(pname)
+    if not pobj:
+        yield "❌ Qwen provider not available."
+        return
 
-        pobj = REAL_PROVIDERS.get(pname)
+    if "Qwen" not in _PROVIDER_MODEL_CACHE:
+        _PROVIDER_MODEL_CACHE["Qwen"] = discover_provider_models(pobj, "Qwen")
 
-        if not pobj:
-            continue
+    model_candidates = ["Qwen/Qwen3.6-35B-A3B"] + [
+        x for x in _PROVIDER_MODEL_CACHE["Qwen"]
+        if x != "Qwen/Qwen3.6-35B-A3B"
+    ]
 
-        if pname not in _PROVIDER_MODEL_CACHE:
-            _PROVIDER_MODEL_CACHE[pname] = discover_provider_models(
-                pobj,
-                pname
+    for m in model_candidates[:12]:
+        try:
+            stream = g4f.ChatCompletion.create(
+                model=m,
+                provider=pobj,
+                messages=msgs,
+                stream=True
             )
 
-        model_candidates = [model_name] + [
-            x for x in _PROVIDER_MODEL_CACHE[pname]
-            if x != model_name
-        ]
-
-        for m in model_candidates[:12]:
-            try:
-                stream = g4f.ChatCompletion.create(
-                    model=m,
-                    provider=pobj,
-                    messages=msgs,
-                    stream=True
-                )
-
-                text = ""
+            text = ""
 
-                for chunk in stream:
-                    c = clean_stream(chunk)
+            for chunk in stream:
+                c = clean_stream(chunk)
 
-                    if c is None or c == "":
-                        continue
+                if c is None or c == "":
+                    continue
 
-                    text += c
-                    yield text
+                text += c
+                yield text
 
-                if text.strip():
-                    _cache_set(key, text)
-                    return
+            if text.strip():
+                _cache_set(key, text)
+                return
 
-            except Exception as e:
-                logger.warning(e)
-                continue
+        except Exception as e:
+            logger.warning(e)
+            continue
 
-    yield "❌ Failed with all providers."
+    yield "❌ Failed with Qwen provider."
 
 # =====================================================
 # FASTAPI
@@ -372,10 +336,22 @@ API_KEY = os.getenv("API_KEY", "mysecretkey123")
 
 class ChatRequest(BaseModel):
     message: str
-    provider: str = "Perplexity"
-    model: str = "sonar"
+    provider: str = "Qwen"
+    model: str = "claude-sonnet-4-6"
     history: List[Any] = []
 
+# =====================================================
+# MCP SUPPORT
+# =====================================================
+class MCPToolCall(BaseModel):
+    tool: str
+    parameters: Dict[str, Any] = {}
+
+class MCPRequest(BaseModel):
+    messages: List[Dict[str, Any]]
+    tools: List[Dict[str, Any]] = []
+    model: str = "claude-sonnet-4-6"
+
 def verify_api_key(request: Request):
     auth = request.headers.get("Authorization")
     if auth and auth.startswith("Bearer "):
@@ -401,7 +377,9 @@ async def root():
             "GET /health": "التحقق من صحة الخادم",
             "GET /providers": "قائمة المزودين والنماذج (يتطلب مفتاح)",
             "POST /chat": "إرسال رسالة والحصول على رد (يتطلب مفتاح)",
-            "POST /chat/stream": "إرسال رسالة والحصول على رد متدفق (يتطلب مفتاح)"
+            "POST /chat/stream": "إرسال رسالة والحصول على رد متدفق (يتطلب مفتاح)",
+            "POST /v1/messages": "MCP / Claude-compatible endpoint (يتطلب مفتاح)",
+            "GET /v1/models": "قائمة النماذج بصيغة OpenAI (يتطلب مفتاح)"
         },
         "authentication": "Bearer YOUR_API_KEY or X-API-Key: YOUR_API_KEY",
         "status": "✅ Server is working"
@@ -444,6 +422,78 @@ async def get_providers(request: Request):
 
     return JSONResponse(content={"providers": providers_info})
 
+# =====================================================
+# MCP / Claude-compatible endpoints
+# =====================================================
+
+@app.get("/v1/models")
+async def list_models(request: Request):
+    verify_api_key(request)
+    models = [
+        {"id": "claude-opus-4-6",   "object": "model", "owned_by": "anthropic"},
+        {"id": "claude-sonnet-4-6", "object": "model", "owned_by": "anthropic"},
+        {"id": "claude-haiku-4-5",  "object": "model", "owned_by": "anthropic"},
+    ]
+    return JSONResponse(content={"object": "list", "data": models})
+
+@app.post("/v1/messages")
+async def mcp_messages(request: Request):
+    verify_api_key(request)
+
+    body = await request.json()
+    model = body.get("model", "claude-sonnet-4-6")
+    messages = body.get("messages", [])
+    stream = body.get("stream", False)
+    max_tokens = body.get("max_tokens", 4096)
+
+    # استخراج آخر رسالة كـ message والباقي كـ history
+    if not messages:
+        raise HTTPException(status_code=400, detail="messages is required")
+
+    last_msg = messages[-1]
+    history = messages[:-1]
+
+    # دعم content كـ string أو list
+    if isinstance(last_msg.get("content"), list):
+        user_text = " ".join(
+            block.get("text", "")
+            for block in last_msg["content"]
+            if isinstance(block, dict) and block.get("type") == "text"
+        )
+    else:
+        user_text = last_msg.get("content", "")
+
+    if stream:
+        async def generate_stream():
+            full = ""
+            for chunk in ask(user_text, history, "Qwen", model):
+                delta = chunk[len(full):]
+                full = chunk
+                if delta:
+                    event = {
+                        "type": "content_block_delta",
+                        "delta": {"type": "text_delta", "text": delta}
+                    }
+                    yield f"data: {json.dumps(event)}\n\n"
+            yield f"data: {json.dumps({'type': 'message_stop'})}\n\n"
+
+        return StreamingResponse(generate_stream(), media_type="text/event-stream")
+
+    else:
+        full_response = ""
+        for chunk in ask(user_text, history, "Qwen", model):
+            full_response = chunk
+
+        return JSONResponse(content={
+            "id": f"msg_{int(time.time())}",
+            "type": "message",
+            "role": "assistant",
+            "model": model,
+            "content": [{"type": "text", "text": full_response}],
+            "stop_reason": "end_turn",
+            "usage": {"input_tokens": 0, "output_tokens": 0}
+        })
+
 if __name__ == "__main__":
     import uvicorn
     port = int(os.getenv("PORT", "7860"))