Create app.py

app.py

@@ -0,0 +1,129 @@
+import os
+import g4f
+import logging
+from fastapi import FastAPI, HTTPException, Header, Request
+from fastapi.responses import StreamingResponse
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from typing import List, Optional
+
+# --- Rate Limiting (Proxy Aware) ---
+from slowapi import Limiter, _rate_limit_exceeded_handler
+from slowapi.errors import RateLimitExceeded
+
+def get_real_ip(request: Request) -> str:
+    """استخراج عنوان IP الحقيقي للمستخدم خلف خوادم Hugging Face"""
+    forwarded_for = request.headers.get("X-Forwarded-For")
+    if forwarded_for:
+        return forwarded_for.split(",")[0].strip()
+    return request.client.host if request.client else "127.0.0.1"
+
+limiter = Limiter(key_func=get_real_ip)
+
+# --- Logging ---
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# --- App Setup ---
+app = FastAPI(title="G4F Gateway", version="1.0", docs_url=None, redoc_url=None) # إخفاء مسارات التوثيق في الإنتاج لتقليل السطح الهجومي
+app.state.limiter = limiter
+app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
+
+API_KEY = os.getenv("API_KEY", "your_fallback_secret")
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=False,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# --- Models ---
+class ChatMessage(BaseModel):
+    role: str
+    content: str
+
+class ChatRequest(BaseModel):
+    model: str = "gpt-4o-mini"
+    messages: List[ChatMessage]
+    stream: bool = False
+    provider: Optional[str] = None
+
+# --- Auth ---
+def verify_key(auth: str):
+    if not auth:
+        raise HTTPException(status_code=401, detail="Missing Authorization header")
+    
+    parts = auth.split()
+    if len(parts) != 2 or parts[0] != "Bearer":
+        raise HTTPException(status_code=401, detail="Malformed Authorization header")
+    
+    if parts[1] != API_KEY:
+        raise HTTPException(status_code=401, detail="Invalid API Key")
+
+# --- Routes ---
+@app.post("/v1/chat/completions")
+@limiter.limit("20/minute")
+async def chat(request: Request, body: ChatRequest, authorization: str = Header(None)):
+    verify_key(authorization)
+    
+    messages = [{"role": m.role, "content": m.content} for m in body.messages]
+    
+    provider = None
+    if body.provider:
+        provider = getattr(g4f.Provider, body.provider, None)
+        if provider is None:
+            raise HTTPException(status_code=400, detail="Invalid provider")
+            
+    try:
+        # التعامل الآمن مع Streaming
+        if body.stream:
+            def generate():
+                try:
+                    response = g4f.ChatCompletion.create(
+                        model=body.model,
+                        messages=messages,
+                        provider=provider,
+                        stream=True
+                    )
+                    for chunk in response:
+                        if chunk:
+                            yield f"data: {chunk}\n\n"
+                    yield "data: [DONE]\n\n"
+                except Exception as stream_e:
+                    logger.error(f"Streaming failure: {stream_e}")
+                    yield f"data: {{\"error\": \"Provider stream disconnected\"}}\n\n"
+                    
+            return StreamingResponse(generate(), media_type="text/event-stream")
+
+        # الاستجابة العادية (Asynchronous)
+        response = await g4f.ChatCompletion.create_async(
+            model=body.model,
+            messages=messages,
+            provider=provider
+        )
+        
+        return {
+            "id": "chatcmpl-custom",
+            "object": "chat.completion",
+            "choices": [
+                {
+                    "index": 0,
+                    "message": {
+                        "role": "assistant",
+                        "content": response
+                    },
+                    "finish_reason": "stop"
+                }
+            ]
+        }
+        
+    except Exception as e:
+        logger.exception(f"G4F Exception: {e}")
+        raise HTTPException(status_code=503, detail="Provider currently unavailable")
+
+@app.get("/")
+@limiter.limit("5/minute")
+def health(request: Request):
+    return {"status": "online"}