Update app.py

app.py

@@ -1,12 +1,12 @@
 import os
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, Request
 from pydantic import BaseModel
 import dashscope
 from dashscope import Generation
 
 # Set up the API key for Dashscope
-dashscope.api_key = os.getenv('HF_TOKEN')  # Ensure the environment variable is set
-print(dashscope.api_key)
+YOUR_API_TOKEN = os.getenv('HF_TOKEN')
+dashscope.api_key = YOUR_API_TOKEN
 
 app = FastAPI()
 
@@ -43,9 +43,6 @@ 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)
+@app.get("/")
+def root():
+    return {"message": "Welcome to the FastAPI Math Solver!"}