Create app.py

app.py

@@ -0,0 +1,152 @@
+from fastapi import FastAPI
+from fastapi.responses import JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
+import requests
+import json
+import os
+import uvicorn
+
+# Configuration
+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"
+)
+
+# CORS middleware
+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 {
+        "message": "Pollinations OpenAI Compatible API",
+        "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"
+        }
+    }
+
+@app.get("/health")
+async def health():
+    return {"status": "healthy"}
+
+@app.get("/models")
+async def get_models():
+    """OpenAI compatible models endpoint"""
+    return {
+        "object": "list",
+        "data": [
+            {
+                "id": model,
+                "object": "model",
+                "created": 1677610602,
+                "owned_by": "pollinations"
+            }
+            for model in MODELS
+        ]
+    }
+
+@app.post("/chat/completions")
+@app.post("/v1/chat/completions")
+async def chat_completions(data: dict):
+    """OpenAI compatible chat completions 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"
+    }
+    
+    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()
+        result = response.json()
+        
+        # Convert to OpenAI format
+        openai_response = {
+            "id": f"chatcmpl-{hash(str(payload)) % 10000000}",
+            "object": "chat.completion",
+            "created": 1677858242,
+            "model": model,
+            "choices": [{
+                "index": 0,
+                "message": {
+                    "role": "assistant",
+                    "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)}"}}
+        )
+
+if __name__ == "__main__":
+    port = int(os.environ.get("PORT", 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)