Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import dashscope
 from dashscope import Generation
@@ -8,6 +8,7 @@ from dashscope import Generation
 headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
 dashscope.api_key = headers
 print(dashscope.api_key)
+
 app = FastAPI()
 
 math_messages = []
@@ -43,3 +44,9 @@ async def solve_math(query: MathQuery):
     answer = get_math_response(user_question)
     return {"question": user_question, "answer": answer}
 
+# Optional: Handle 404 errors gracefully without defining additional routes
+@app.exception_handler(HTTPException)
+async def http_exception_handler(request, exc):
+    if exc.status_code == 404:
+        return {"detail": "Not Found. Use /math-solver to submit a math question."}
+    return await request.app.default_exception_handler(request, exc)