Update app.py

app.py

@@ -1,9 +1,8 @@
 import time
 import random
 import traceback
-import uuid
 
-from fastapi import FastAPI, Header, HTTPException
+from fastapi import FastAPI, Header
 from pydantic import BaseModel
 import uvicorn
 
@@ -15,7 +14,6 @@ MODEL = "gpt-4o-mini"
 MAX_RETRIES = 10
 TIMEOUT = 60
 
-# ضع المفتاح هنا
 OPENAI_API_KEY = "sk-your-key"
 
 SAFE_PROVIDERS = []
@@ -97,7 +95,7 @@ class SmartG4F:
 
         return providers
 
-    def ask(self, messages):
+    def ask(self, prompt):
 
         errors = []
 
@@ -121,7 +119,12 @@ class SmartG4F:
                 response = client.chat.completions.create(
                     model=MODEL,
                     provider=provider,
-                    messages=messages,
+                    messages=[
+                        {
+                            "role": "user",
+                            "content": prompt
+                        }
+                    ],
                     timeout=TIMEOUT,
                 )
 
@@ -181,18 +184,8 @@ app = FastAPI()
 ai = SmartG4F()
 
 
-# =========================
-# OpenAI Compatible Models
-# =========================
-
-class Message(BaseModel):
-    role: str
-    content: str
-
-
-class ChatRequest(BaseModel):
-    model: str = MODEL
-    messages: list[Message]
+class Query(BaseModel):
+    prompt: str
 
 
 @app.get("/")
@@ -204,67 +197,19 @@ async def home():
     }
 
 
-# =========================
-# OpenAI Endpoint
-# =========================
-
-@app.post("/v1/chat/completions")
-async def chat_completions(
-    request: ChatRequest,
+@app.post("/ask")
+async def ask(
+    query: Query,
     authorization: str = Header(None)
 ):
 
-    # التحقق من المفتاح
-    if OPENAI_API_KEY:
-
-        if not authorization:
-            raise HTTPException(
-                status_code=401,
-                detail="Missing Authorization header"
-            )
-
-        token = authorization.replace("Bearer ", "")
-
-        if token != OPENAI_API_KEY:
-            raise HTTPException(
-                status_code=401,
-                detail="Invalid API key"
-            )
-
-    result = ai.ask(
-        [m.dict() for m in request.messages]
-    )
-
-    if not result["success"]:
-
-        raise HTTPException(
-            status_code=500,
-            detail=result
-        )
-
-    response_text = result["response"]
-
-    return {
-        "id": f"chatcmpl-{uuid.uuid4().hex}",
-        "object": "chat.completion",
-        "created": int(time.time()),
-        "model": request.model,
-        "choices": [
-            {
-                "index": 0,
-                "message": {
-                    "role": "assistant",
-                    "content": response_text
-                },
-                "finish_reason": "stop"
-            }
-        ],
-        "usage": {
-            "prompt_tokens": 0,
-            "completion_tokens": 0,
-            "total_tokens": 0
+    if authorization != f"Bearer {OPENAI_API_KEY}":
+        return {
+            "success": False,
+            "error": "Invalid API Key"
         }
-    }
+
+    return ai.ask(query.prompt)
 
 
 if __name__ == "__main__":