Update app.py

app.py

@@ -1,18 +1,39 @@
 import os
 import subprocess
+import logging
 
 from fastapi import FastAPI, Request, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
+# Optional: Configure CORS if needed
+origins = [
+    # Add allowed origins if you implement a frontend later
+]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,  # Adjust as needed
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 # Load your API key from the environment (defaults to "change_me")
 OLLAMA_API_KEY = os.environ.get("OLLAMA_API_KEY", "change_me")
 
-@app.post("/generate")
+@app.post("/api/generate")
 async def generate(request: Request):
     """Endpoint that generates text based on the prompt."""
     # 1. Check API key
     auth_header = request.headers.get("Authorization")
+    logger.info(f"Received Authorization header: {auth_header}")
+
     if not auth_header or not auth_header.startswith("Bearer "):
         raise HTTPException(status_code=401, detail="Missing or invalid Authorization header")
 
@@ -44,6 +65,7 @@ async def generate(request: Request):
         output, error = process.communicate(timeout=60)  # Timeout after 60 seconds
 
         if process.returncode != 0:
+            logger.error(f"Ollama Error: {error.strip()}")
             raise Exception(error.strip())
 
     except subprocess.TimeoutExpired:
@@ -51,15 +73,21 @@ async def generate(request: Request):
         raise HTTPException(status_code=500, detail="Ollama request timed out")
 
     except Exception as e:
+        logger.error(f"Ollama Exception: {str(e)}")
         raise HTTPException(
             status_code=500,
             detail=f"Ollama error: {str(e)}"
         )
 
+    logger.info("Ollama response successfully generated.")
     return {"response": output.strip()}
 
+@app.get("/health")
+async def health():
+    """Health check endpoint."""
+    return {"status": "OK"}
+
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8080)  # Changed from 7860 to 8080
-
+    uvicorn.run(app, host="0.0.0.0", port=8080)