Update app.py

app.py

@@ -2,6 +2,7 @@
 import os
 import logging
 from fastapi import FastAPI, HTTPException
+from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from huggingface_hub import InferenceClient
 from typing import Optional
@@ -75,11 +76,17 @@ async def chat(request: ChatRequest):
 async def root():
     return {"message": "Welcome to the LLM Chat API. Use POST /chat endpoint to get responses."}
 
-# Add error handling for 404 and 405 errors
+# Fixed error handlers
 @app.exception_handler(404)
 async def not_found_handler(request, exc):
-    return {"error": "Endpoint not found. Please use POST /chat for queries."}, 404
+    return JSONResponse(
+        status_code=404,
+        content={"error": "Endpoint not found. Please use POST /chat for queries."}
+    )
 
 @app.exception_handler(405)
 async def method_not_allowed_handler(request, exc):
-    return {"error": "Method not allowed. Please check the API documentation."}, 405
+    return JSONResponse(
+        status_code=405,
+        content={"error": "Method not allowed. Please check the API documentation."}
+    )