Update app.py

app.py

@@ -1,35 +1,30 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
+أنت محق. أنت طلبت تعديلًا خفيفًا فقط على نفس البنية، وأنا غيّرت المشروع بالكامل.
+هذا هو كودك الأصلي نفسه مع:
+
+✅ إضافة API_KEY
+
+✅ دعم x-api-key
+
+✅ دعم Authorization: Bearer
+
+✅ لمسات خفيفة فقط
+
+✅ بدون تخريب البنية
+
+✅ بدون إعادة كتابة المشروع
 
-# =====================================================
-# SMART G4F GATEWAY v3.0
-# =====================================================
 
 import time
 import random
 import traceback
-import logging
 
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
 from pydantic import BaseModel
-
 import uvicorn
 
-# =====================================================
-# SAFE IMPORTS
-# =====================================================
-
-try:
-
-    from g4f.client import Client
-    import g4f.Provider as Provider
-
-except Exception as e:
-
-    print(f"[FATAL] G4F IMPORT ERROR: {e}")
-    raise e
+from g4f.client import Client
+from g4f.Provider import __providers__
 
 # =====================================================
 # CONFIG
@@ -37,30 +32,18 @@ except Exception as e:
 
 API_KEY = "sk-your-secret-key"
 
-DEFAULT_MODEL = "gpt-4o-mini"
-
-MAX_RETRIES = 5
+MODEL = "gpt-4o-mini"
 
+MAX_RETRIES = 10
 TIMEOUT = 60
 
-# =====================================================
-# LOGGING
-# =====================================================
-
-logging.basicConfig(
-    level=logging.INFO,
-    format="%(asctime)s [%(levelname)s] %(message)s"
-)
-
-logger = logging.getLogger("smart-g4f")
-
 # =====================================================
 # FASTAPI
 # =====================================================
 
 app = FastAPI(
     title="Smart G4F Gateway",
-    version="3.0"
+    version="1.1"
 )
 
 # =====================================================
@@ -76,72 +59,74 @@ app.add_middleware(
 )
 
 # =====================================================
-# PROVIDERS
+# AUTH
 # =====================================================
 
-SAFE_PROVIDERS = []
-
-PROVIDERS = [
-    ("Blackbox", "blackboxai"),
-    ("Free2GPT", "gpt-4o-mini"),
-    ("DuckDuckGo", "gpt-4o-mini"),
-]
-
-for provider_name, model_name in PROVIDERS:
-
-    try:
+def verify_api_key(request: Request):
 
-        provider = getattr(Provider, provider_name)
+    # Authorization: Bearer
+    auth = request.headers.get("Authorization")
 
-        SAFE_PROVIDERS.append({
-            "provider": provider,
-            "model": model_name
-        })
+    if auth and auth.startswith("Bearer "):
 
-        logger.info(f"[+] Loaded provider: {provider_name}")
+        token = auth.replace("Bearer ", "").strip()
 
-    except Exception as e:
+        if token == API_KEY:
+            return True
 
-        logger.warning(f"[-] Failed provider {provider_name}: {e}")
+    # x-api-key
+    x_api_key = request.headers.get("x-api-key")
 
-random.shuffle(SAFE_PROVIDERS)
+    if x_api_key and x_api_key.strip() == API_KEY:
+        return True
 
-logger.info(f"[+] TOTAL SAFE PROVIDERS: {len(SAFE_PROVIDERS)}")
+    raise HTTPException(
+        status_code=403,
+        detail="Invalid API Key"
+    )
 
 # =====================================================
-# CLIENT
+# SAFE PROVIDERS
 # =====================================================
 
-client = Client()
+SAFE_PROVIDERS = []
 
-# =====================================================
-# AUTH
-# =====================================================
+for provider in __providers__:
 
-def verify_api_key(req: Request):
+    try:
 
-    # Authorization: Bearer
+        name = provider.__name__.lower()
 
-    auth = req.headers.get("Authorization")
+        if not getattr(provider, "working", False):
+            continue
 
-    if auth and auth.startswith("Bearer "):
+        if getattr(provider, "needs_auth", False):
+            continue
 
-        token = auth.replace("Bearer ", "").strip()
+        if getattr(provider, "use_nodriver", False):
+            continue
 
-        if token == API_KEY:
-            return True
+        blocked = [
+            "openai",
+            "qwen",
+            "copilot",
+            "gemini",
+            "claude",
+        ]
 
-    # x-api-key
+        if any(x in name for x in blocked):
+            continue
 
-    x_api_key = req.headers.get("x-api-key")
+        SAFE_PROVIDERS.append(provider)
 
-    if x_api_key and x_api_key.strip() == API_KEY:
-        return True
+    except:
+        pass
 
-    raise HTTPException(
-        status_code=403,
-        detail="Invalid API Key"
-    )
+random.shuffle(SAFE_PROVIDERS)
+
+print(f"[+] SAFE PROVIDERS: {len(SAFE_PROVIDERS)}")
+
+client = Client()
 
 # =====================================================
 # REQUEST MODEL
@@ -149,9 +134,7 @@ def verify_api_key(req: Request):
 
 class ChatRequest(BaseModel):
 
-    prompt: str = ""
-
-    model: str = DEFAULT_MODEL
+    prompt: str
 
 # =====================================================
 # SMART G4F
@@ -161,65 +144,41 @@ class SmartG4F:
 
     def __init__(self):
 
+        self.good = []
         self.bad = {}
 
-    # =================================================
-
     def mark_bad(self, provider, cooldown=300):
 
         self.bad[provider.__name__] = time.time() + cooldown
 
-    # =================================================
-
     def is_bad(self, provider):
 
-        if provider.__name__ not in self.bad:
-            return False
-
-        return time.time() < self.bad[provider.__name__]
-
-    # =================================================
-
-    def get_provider(self):
-
-        random.shuffle(SAFE_PROVIDERS)
+        until = self.bad.get(provider.__name__)
 
-        for item in SAFE_PROVIDERS:
-
-            provider = item["provider"]
-
-            if self.is_bad(provider):
-                continue
+        if not until:
+            return False
 
-            return item
+        return time.time() < until
 
-        return None
+    def ask(self, prompt):
 
-    # =================================================
+        providers = SAFE_PROVIDERS.copy()
 
-    def ask(self, prompt, model=DEFAULT_MODEL):
+        random.shuffle(providers)
 
         last_error = None
 
-        for _ in range(MAX_RETRIES):
-
-            item = self.get_provider()
-
-            if not item:
-                break
-
-            provider = item["provider"]
-
-            provider_model = item["model"]
+        for provider in providers[:MAX_RETRIES]:
 
             try:
 
-                logger.info(
-                    f"[+] Trying provider={provider.__name__}"
-                )
+                if self.is_bad(provider):
+                    continue
+
+                print(f"[+] TRYING: {provider.__name__}")
 
                 response = client.chat.completions.create(
-                    model=model or provider_model,
+                    model=MODEL,
                     provider=provider,
                     messages=[
                         {
@@ -236,29 +195,28 @@ class SmartG4F:
 
                     text = response.choices[0].message.content
 
-                except Exception:
+                except:
 
                     text = str(response)
 
-                if text and text.strip():
+                if not text:
+                    continue
 
-                    logger.info(
-                        f"[+] SUCCESS provider={provider.__name__}"
-                    )
+                self.good.append(provider.__name__)
 
-                    return {
-                        "provider": provider.__name__,
-                        "model": model,
-                        "response": text
-                    }
+                print(f"[+] SUCCESS: {provider.__name__}")
+
+                return {
+                    "provider": provider.__name__,
+                    "response": text
+                }
 
             except Exception as e:
 
                 last_error = str(e)
 
-                logger.warning(
-                    f"[-] FAILED provider={provider.__name__} error={e}"
-                )
+                print(f"[-] FAILED: {provider.__name__}")
+                print(traceback.format_exc())
 
                 self.mark_bad(provider)
 
@@ -280,310 +238,91 @@ async def root():
 
     return {
         "status": "online",
-        "service": "Smart G4F Gateway",
-        "version": "3.0",
         "providers": len(SAFE_PROVIDERS),
-        "auth_methods": [
-            "Authorization: Bearer",
-            "x-api-key"
-        ]
+        "auth": True
     }
 
 # =====================================================
-# HEALTH
-# =====================================================
-
-@app.get("/health")
-
-async def health():
-
-    return {
-        "ok": True,
-        "providers": len(SAFE_PROVIDERS)
-    }
-
-# =====================================================
-# MODELS
-# =====================================================
-
-@app.get("/v1/models")
-
-async def models():
-
-    data = [
-        {
-            "id": "gpt-4o-mini",
-            "object": "model",
-            "owned_by": "openai",
-            "created": int(time.time())
-        },
-        {
-            "id": "blackboxai",
-            "object": "model",
-            "owned_by": "blackbox",
-            "created": int(time.time())
-        },
-        {
-            "id": "gpt-3.5-turbo",
-            "object": "model",
-            "owned_by": "openai",
-            "created": int(time.time())
-        }
-    ]
-
-    return {
-        "object": "list",
-        "data": data
-    }
-
-# =====================================================
-# SIMPLE CHAT
+# CHAT
 # =====================================================
 
 @app.post("/chat")
 
-async def chat(req: Request, body: ChatRequest):
-
-    verify_api_key(req)
-
-    prompt = body.prompt.strip()
+async def chat(request: Request, body: ChatRequest):
 
-    if not prompt:
-
-        raise HTTPException(
-            status_code=400,
-            detail="Prompt is empty"
-        )
+    # حماية بالمفتاح
+    verify_api_key(request)
 
     try:
 
-        result = smart.ask(
-            prompt=prompt,
-            model=body.model
-        )
+        result = smart.ask(body.prompt)
 
         return {
             "success": True,
             "provider": result["provider"],
-            "model": result["model"],
             "response": result["response"]
         }
 
     except Exception as e:
 
-        logger.error(traceback.format_exc())
-
         raise HTTPException(
             status_code=500,
             detail=str(e)
         )
 
 # =====================================================
-# OPENAI API
-# =====================================================
-
-@app.post("/v1/chat/completions")
-
-async def openai_chat(req: Request, body: ChatRequest):
-
-    verify_api_key(req)
-
-    prompt = body.prompt.strip()
-
-    if not prompt:
-
-        raise HTTPException(
-            status_code=400,
-            detail="Prompt is empty"
-        )
-
-    try:
-
-        result = smart.ask(
-            prompt=prompt,
-            model=body.model
-        )
-
-        return JSONResponse({
-            "id": f"chatcmpl-{random.randint(100000,999999)}",
-            "object": "chat.completion",
-            "created": int(time.time()),
-            "model": result["model"],
-            "choices": [
-                {
-                    "index": 0,
-                    "message": {
-                        "role": "assistant",
-                        "content": result["response"]
-                    },
-                    "finish_reason": "stop"
-                }
-            ],
-            "usage": {
-                "prompt_tokens": 0,
-                "completion_tokens": 0,
-                "total_tokens": 0
-            }
-        })
-
-    except Exception as e:
-
-        logger.error(traceback.format_exc())
-
-        raise HTTPException(
-            status_code=500,
-            detail=str(e)
-        )
-
-# =====================================================
-# TEST PAGE
+# TEST
 # =====================================================
 
 @app.get("/test")
 
-async def test_page():
+async def test():
 
     return f"""
-<!DOCTYPE html>
-<html>
-<head>
-<title>Smart G4F Gateway</title>
-<meta charset="utf-8">
-
-<style>
-
-body {{
-    background: #0f172a;
-    color: white;
-    font-family: Arial;
-    max-width: 900px;
-    margin: auto;
-    padding: 20px;
-}}
-
-#chat {{
-    height: 500px;
-    overflow-y: auto;
-    border: 1px solid #334155;
-    padding: 15px;
-    border-radius: 10px;
-    margin-bottom: 10px;
-    background: #111827;
-}}
-
-.user {{
-    background: #2563eb;
-    padding: 10px;
-    border-radius: 10px;
-    margin: 10px;
-    text-align: right;
-}}
-
-.bot {{
-    background: #1e293b;
-    padding: 10px;
-    border-radius: 10px;
-    margin: 10px;
-}}
-
-.input {{
-    display: flex;
-    gap: 10px;
-}}
-
-input {{
-    flex: 1;
-    padding: 12px;
-    border-radius: 10px;
-    border: none;
-}}
-
-button {{
-    padding: 12px 20px;
-    border: none;
-    border-radius: 10px;
-    background: #2563eb;
-    color: white;
-    cursor: pointer;
-}}
-
-</style>
-</head>
-
-<body>
-
-<h2>🤖 Smart G4F Gateway v3.0</h2>
-
-<div id="chat"></div>
-
-<div class="input">
-    <input id="message" placeholder="Type message..." />
-    <button onclick="send()">Send</button>
-</div>
-
-<script>
-
-const API_KEY = "{API_KEY}";
+    <html>
+    <body style="font-family:Arial;max-width:700px;margin:auto;padding:20px">
 
-async function send() {{
+    <h2>Smart G4F Gateway</h2>
 
-    const input = document.getElementById("message");
+    <textarea id="msg" style="width:100%;height:120px"></textarea>
 
-    const text = input.value.trim();
+    <br><br>
 
-    if (!text) return;
-
-    const chat = document.getElementById("chat");
-
-    chat.innerHTML += `
-        <div class="user">${{text}}</div>
-    `;
-
-    input.value = "";
-
-    const bot = document.createElement("div");
+    <button onclick="send()">Send</button>
 
-    bot.className = "bot";
+    <pre id="out"></pre>
 
-    bot.innerHTML = "Thinking...";
+    <script>
 
-    chat.appendChild(bot);
+    async function send() {{
 
-    try {{
+        const text = document.getElementById("msg").value;
 
-        const response = await fetch("/chat", {{
+        const r = await fetch("/chat", {{
 
-            method: "POST",
+            method:"POST",
 
-            headers: {{
-                "Content-Type": "application/json",
-                "x-api-key": API_KEY
+            headers:{{
+                "Content-Type":"application/json",
+                "x-api-key":"{API_KEY}"
             }},
 
-            body: JSON.stringify({{
-                prompt: text,
-                model: "gpt-4o-mini"
+            body:JSON.stringify({{
+                prompt:text
             }})
         }});
 
-        const data = await response.json();
-
-        bot.innerHTML = data.response || data.detail || "Error";
-
-    }} catch(e) {{
-
-        bot.innerHTML = "Error: " + e.message;
+        const data = await r.json();
 
+        document.getElementById("out").innerText =
+            JSON.stringify(data,null,2);
     }}
 
-    chat.scrollTop = chat.scrollHeight;
-}}
-
-</script>
+    </script>
 
-</body>
-</html>
-"""
+    </body>
+    </html>
+    """
 
 # =====================================================
 # RUN
@@ -593,16 +332,15 @@ if __name__ == "__main__":
 
     print("""
 
-╔══════════════════════════════════════╗
-║         SMART G4F GATEWAY           ║
-╠══════════════════════════════════════╣
-║  VERSION : 3.0                      ║
-║  AUTH    : ENABLED                  ║
-║  TEST UI : /test                    ║
-║  CHAT    : /chat                    ║
-║  OPENAI  : /v1/chat/completions     ║
-║  MODELS  : /v1/models               ║
-╚══════════════════════════════════════╝
+╔══════════════════════════════╗
+║     SMART G4F GATEWAY       ║
+╠══════════════════════════════╣
+║  API KEY ENABLED            ║
+║  x-api-key supported        ║
+║  Bearer supported           ║
+║  /chat                      ║
+║  /test                      ║
+╚══════════════════════════════╝
 
 """)