Update app.py

app.py

@@ -51,7 +51,19 @@ pdf_service = PDFService(model_service)
 faq_service = FAQService(model_service)
 chat_service = ChatService(model_service, data_service, pdf_service, faq_service)
 
-# API endpoints
+import math
+from fastapi.responses import JSONResponse
+
+# Helper function to sanitize data
+def sanitize_response(data):
+    if isinstance(data, dict):
+        return {k: sanitize_response(v) for k, v in data.items()}
+    elif isinstance(data, list):
+        return [sanitize_response(item) for item in data]
+    elif isinstance(data, float) and (math.isnan(data) or math.isinf(data)):
+        return None  # Replace NaN/Infinity with None or another default value
+    return data
+
 @app.post("/api/chat")
 async def chat_endpoint(
     background_tasks: BackgroundTasks,
@@ -59,19 +71,31 @@ async def chat_endpoint(
     api_key: str = Depends(get_api_key)
 ):
     try:
+        # Call the chat service to get the data
         response, updated_history, search_results = await chat_service.chat(
             user_input.user_input,
             user_input.chat_history
         )
-        return {
+
+        # Build the response dictionary
+        response_data = {
             "status": "success",
             "response": response,
             "chat_history": updated_history,
             "search_results": search_results
         }
+
+        # Sanitize the response to ensure JSON compliance
+        sanitized_data = sanitize_response(response_data)
+
+        # Return the sanitized response
+        return JSONResponse(content=sanitized_data)
+
     except Exception as e:
+        # Log and raise an error with details
         logger.error(f"Error in chat endpoint: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail="An internal server error occurred.")
+
 
 @app.post("/api/search")
 async def search_endpoint(
@@ -80,7 +104,12 @@ async def search_endpoint(
 ):
     try:
         results = await data_service.search(query.query, query.top_k)
-        return {"results": results}
+                # Sanitize the response to ensure JSON compliance
+        sanitized_data = sanitize_response(results)
+
+        # Return the sanitized response
+        return JSONResponse(content=sanitized_data)
+         
     except Exception as e:
         logger.error(f"Error in search endpoint: {e}")
         raise HTTPException(status_code=500, detail=str(e))
@@ -153,7 +182,7 @@ if __name__ == "__main__":
     
     # Create and launch Gradio interface
     demo = create_gradio_interface()
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=8080)
     
     # Start FastAPI server
     #uvicorn.run(app, host="0.0.0.0", port=8000)