feat: improve error handling

app.py

@@ -6,11 +6,16 @@ from utils.llm_caller import LLMCaller
 import asyncio
 import time
 from datetime import datetime
+from fastapi import FastAPI, HTTPException
+import logging
 
 app = FastAPI()
 data_importer = DataImporter()
 agent = LLMCaller()
 
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 @app.get("/")
 def root():
     """Root endpoint - Hugging Face checks this"""
@@ -29,11 +34,6 @@ def health_check():
         "service": "PAN-SEA Travel Planning API"
     }
 
-@app.get("/ready")
-def ready_check():
-    """Simple ready check for Hugging Face"""
-    return {"ready": True, "status": "ok"}
-
 @app.get("/v1")
 def greet_json():
     start_time = time.time()
@@ -57,30 +57,37 @@ def generate_trip_plan(request: PlanRequest):
                             preparation=trip_plan.preparation,
                             meta={"status": "success", "timestamp": datetime.utcnow().isoformat()})
     except Exception as e:
-        print(f"Error in generate_trip_plan: {e}")
-        # Return error response
-        return PlanResponse(
-            tripOverview=f"Error: {str(e)}",
-            query_params=request,
-            retrieved_data=[],
-            trip_plan=TripPlan(tripOverview="Error occurred", timeline=[], spots=[], budget={}, permits={}, safety={}, contact={}, safetyContacts={}),
-            preparation=None,
-            meta={"status": "error", "error": str(e)}
+        logger.error(f"Error in generate_trip_plan: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail={
+                "error": "Internal server error",
+                "message": "Failed to generate trip plan",
+                "details": str(e)
+            }
         )
 
+
 @app.post("/v1/addYoutubeLink", response_model=YoutubeLinkResponse)
 def add_youtube_link(request: YoutubeLinkRequest):
     try:
         data_importer.insert_from_youtube(request.video_id)
-    except Exception as e:
         return YoutubeLinkResponse(
-            message="Failed to add YouTube link",
-            video_url= None
+                    message="add successfully",
+                    video_url=f"https://www.youtube.com/watch?v={request.video_id}"
+                )
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=f"Invalid video ID: {str(e)}")
+    except Exception as e:
+        logger.error(f"Error adding YouTube link: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail={
+                "error": "Internal server error",
+                "message": "Failed to add YouTube link",
+                "details": str(e)
+            }
         )
-    return YoutubeLinkResponse(
-        message="add successfully",
-        video_url=f"https://www.youtube.com/watch?v={request.video_id}"
-    )
 
 @app.post("/v1/searchSimilar", response_model=list[dict])
 def search_similar(request: YoutubeLinkRequest):
@@ -88,15 +95,33 @@ def search_similar(request: YoutubeLinkRequest):
         results = data_importer.search_similar(query=request.video_id)
         return results
     except Exception as e:
-        print(f"Error during search: {e}")
-        return {"error": "Search failed"}
-    return []
+        logger.error(f"Error during search: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail={
+                "error": "Search failed",
+                "message": "Unable to search similar content",
+                "details": str(e)
+            }
+        )
 
 @app.post("/v1/basicChat", response_model=str)
 def basic_chat(request: ChatRequest):
-    user_message = request.message
-    llm_response = asyncio.run(agent.basic_query(
-        user_prompt=user_message
-    ))
-    return llm_response
+    try:
+        user_message = request.message
+        llm_response = asyncio.run(agent.basic_query(
+            user_prompt=user_message
+        ))
+        return llm_response
+    except Exception as e:
+        logger.error(f"Error in basic_chat: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail={
+                "error": "Internal server error",
+                "message": "Failed to process chat request",
+                "details": str(e)
+            }
+        )
+