Update app.py

app.py

@@ -13,13 +13,19 @@ from dotenv import load_dotenv
 load_dotenv()
 
 # Initialize FastAPI app
-app = FastAPI(title="MultiChatAI to OpenAI API Wrapper")
+app = FastAPI(
+    title="MultiChatAI to OpenAI API Wrapper",
+    description="API wrapper for MultiChatAI with OpenAI-compatible endpoints",
+    version="1.0.0",
+    docs_url="/docs",
+    redoc_url=None
+)
 
 # Configuration
 API_KEY_NAME = "X-API-KEY"
 API_KEYS = os.getenv("API_KEYS", "").split(",")  # Comma-separated list from .env
 
-# Configure CORS - adjust these based on your needs
+# Configure CORS
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -33,15 +39,9 @@ api_key_header = APIKeyHeader(name=API_KEY_NAME, auto_error=False)
 
 async def get_api_key(api_key: str = Security(api_key_header)):
     if not api_key:
-        raise HTTPException(
-            status_code=401, 
-            detail="API key is missing"
-        )
+        raise HTTPException(status_code=401, detail="API key is missing")
     if api_key not in API_KEYS:
-        raise HTTPException(
-            status_code=401, 
-            detail="Invalid API key"
-        )
+        raise HTTPException(status_code=401, detail="Invalid API key")
     return api_key
 
 # Request models
@@ -56,9 +56,14 @@ class ChatCompletionRequest(BaseModel):
     max_tokens: int = None
 
 # Health check endpoint
-@app.get("/")
+@app.get("/", include_in_schema=False)
 async def health_check():
-    return {"status": "OK", "service": "MultiChatAI Proxy", "timestamp": datetime.now().isoformat()}
+    return {
+        "status": "OK",
+        "service": "MultiChatAI Proxy",
+        "timestamp": datetime.now().isoformat(),
+        "environment": os.getenv("ENVIRONMENT", "development")
+    }
 
 # Main API endpoint
 @app.post("/v1/chat/completions")
@@ -85,26 +90,16 @@ async def chat_completion(
             "customModelId": ""
         }
         
-        headers = {
-            "accept": "application/json",
-            "content-type": "application/json",
-        }
-        
         # Call MultiChatAI API
         response = requests.post(
             "https://www.multichatai.com/api/chat/deepinfra",
-            headers=headers,
+            headers={"Content-Type": "application/json"},
             json=multi_chat_body,
             timeout=30
         )
         
-        if not response.ok:
-            raise HTTPException(
-                status_code=response.status_code,
-                detail=f"MultiChatAI API error: {response.text}"
-            )
+        response.raise_for_status()
         
-        # Format response in OpenAI style
         return JSONResponse({
             "id": f"chatcmpl-{uuid.uuid4()}",
             "object": "chat.completion",
@@ -127,10 +122,29 @@ async def chat_completion(
         
     except requests.Timeout:
         raise HTTPException(status_code=504, detail="Upstream service timeout")
+    except requests.RequestException as e:
+        raise HTTPException(
+            status_code=502,
+            detail=f"Upstream service error: {str(e)}"
+        )
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
 # Add this if you need to support OPTIONS requests
 @app.options("/v1/chat/completions")
 async def options_handler():
-    return JSONResponse(content={}, status_code=200)
+    return JSONResponse(content={}, status_code=200)
+
+# For production deployment
+def get_application():
+    return app
+
+# For running locally
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(
+        "app:app",
+        host="0.0.0.0",
+        port=int(os.getenv("PORT", 7860)),
+        reload=os.getenv("RELOAD", "false").lower() == "true"
+    )