Update app.py

app.py

@@ -1,5 +1,4 @@
 from fastapi import FastAPI
-from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 import requests
 import json
@@ -10,28 +9,16 @@ import uvicorn
 TOKEN = os.getenv("POLLINATIONS_TOKEN")
 POLLINATIONS_URL = "https://text.pollinations.ai/openai"
 
-app = FastAPI(
-    title="Pollinations OpenAI API",
-    description="OpenAI Compatible API powered by Pollinations AI",
-    version="1.0.0"
-)
+app = FastAPI(title="Pollinations OpenAI API")
 
-# CORS middleware
+# CORS
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
-    allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# Available models
-MODELS = [
-    "openai", "openai-large", "openai-reasoning", "qwen-coder", "llama", 
-    "llamascout", "mistral", "unity", "gemini", "phi", "deepseek", 
-    "deepseek-reasoning", "rtist", "searchgpt", "evil", "openai-audio"
-]
-
 @app.get("/")
 async def root():
     return {
@@ -39,85 +26,49 @@ async def root():
         "status": "running",
         "endpoints": {
             "models": "/models",
-            "chat_completions": "/chat/completions",
-            "openai_compatible": "/v1/chat/completions"
-        },
-        "usage": {
-            "models": "GET /models",
-            "chat": "POST /chat/completions",
-            "openai": "POST /v1/chat/completions"
+            "chat": "/chat/completions"
         }
     }
 
-@app.get("/health")
-async def health():
-    return {"status": "healthy"}
-
 @app.get("/models")
 async def get_models():
-    """OpenAI compatible models endpoint"""
+    """Models endpoint - यहीं missing था!"""
     return {
         "object": "list",
         "data": [
-            {
-                "id": model,
-                "object": "model",
-                "created": 1677610602,
-                "owned_by": "pollinations"
-            }
-            for model in MODELS
+            {"id": "openai", "object": "model", "owned_by": "pollinations"},
+            {"id": "llama", "object": "model", "owned_by": "pollinations"},
+            {"id": "mistral", "object": "model", "owned_by": "pollinations"},
+            {"id": "gemini", "object": "model", "owned_by": "pollinations"},
+            {"id": "deepseek", "object": "model", "owned_by": "pollinations"}
         ]
     }
 
 @app.post("/chat/completions")
-@app.post("/v1/chat/completions")
 async def chat_completions(data: dict):
-    """OpenAI compatible chat completions endpoint"""
-    
+    """Chat endpoint"""
     model = data.get("model", "openai")
     messages = data.get("messages", [])
     temperature = data.get("temperature", 0.7)
-    max_tokens = data.get("max_tokens", 1000)
-    stream = data.get("stream", False)
     
-    if not messages:
-        return JSONResponse(
-            status_code=400,
-            content={"error": {"message": "messages field is required"}}
-        )
-    
-    # Prepare payload for Pollinations
     payload = {
         "model": model,
         "messages": messages,
         "temperature": temperature,
-        "max_tokens": max_tokens,
-        "referrer": "hf-openai-api"
+        "referrer": "hf-api"
     }
     
-    if stream:
-        payload["stream"] = True
-    
-    # Headers
     headers = {"Content-Type": "application/json"}
     if TOKEN:
         headers["Authorization"] = f"Bearer {TOKEN}"
     
     try:
-        response = requests.post(
-            POLLINATIONS_URL, 
-            json=payload, 
-            headers=headers, 
-            timeout=120
-        )
-        response.raise_for_status()
+        response = requests.post(POLLINATIONS_URL, json=payload, headers=headers, timeout=60)
         result = response.json()
         
-        # Convert to OpenAI format
-        openai_response = {
-            "id": f"chatcmpl-{hash(str(payload)) % 10000000}",
+        return {
+            "id": "chatcmpl-123",
             "object": "chat.completion",
-            "created": 1677858242,
             "model": model,
             "choices": [{
                 "index": 0,
@@ -126,27 +77,10 @@ async def chat_completions(data: dict):
                     "content": result["choices"][0]["message"]["content"]
                 },
                 "finish_reason": "stop"
-            }],
-            "usage": {
-                "prompt_tokens": len(str(messages)),
-                "completion_tokens": len(result["choices"][0]["message"]["content"]),
-                "total_tokens": len(str(messages)) + len(result["choices"][0]["message"]["content"])
-            }
+            }]
         }
-        
-        return openai_response
-        
-    except requests.exceptions.RequestException as e:
-        return JSONResponse(
-            status_code=500,
-            content={"error": {"message": f"Pollinations API error: {str(e)}"}}
-        )
     except Exception as e:
-        return JSONResponse(
-            status_code=500,
-            content={"error": {"message": f"Server error: {str(e)}"}}
-        )
+        return {"error": str(e)}
 
 if __name__ == "__main__":
-    port = int(os.environ.get("PORT", 7860))
-    uvicorn.run(app, host="0.0.0.0", port=port)
+    uvicorn.run(app, host="0.0.0.0", port=7860)