Update main.py

main.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, HTTPException
+from fastapi.responses import RedirectResponse # Import RedirectResponse
 import google.generativeai as genai
 from fastapi.middleware.cors import CORSMiddleware
 import requests
@@ -72,13 +73,11 @@ def get_weather(location: str) -> dict:
 # Tool execution function
 def execute_tool(tool_name: str, parameters: dict) -> str:
     if tool_name == "get_weather":
-        # The 'London' fallback will now only be used if Gemini genuinely doesn't provide a location
         result = get_weather(parameters.get("location", "London"))
         if "error" in result:
             return result["error"]
 
         try:
-            # Correct parsing for OpenWeatherMap API
             location_name = result.get("name", "Unknown City")
             country = result.get("sys", {}).get("country", "Unknown Country")
             temp_c = result.get("main", {}).get("temp")
@@ -91,7 +90,6 @@ def execute_tool(tool_name: str, parameters: dict) -> str:
             if None in [temp_c, humidity, wind_speed_ms, feels_like_c]:
                 return "Incomplete weather data received from API."
 
-            # Convert wind speed from m/s to km/h
             wind_kph = wind_speed_ms * 3.6
 
             return (
@@ -116,7 +114,6 @@ def process_with_tools(query: str) -> str:
     try:
         response = model.generate_content(query)
 
-        # Check if Gemini wants to call a function
         if response.candidates and response.candidates[0].content.parts:
             for part in response.candidates[0].content.parts:
                 if part.function_call:
@@ -124,7 +121,6 @@ def process_with_tools(query: str) -> str:
                     
                     function_args = {}
                     if hasattr(part.function_call, 'args'):
-                        # Directly convert MapComposite or dict-like objects to dict
                         if hasattr(part.function_call.args, 'items'):
                             function_args = dict(part.function_call.args)
                         elif isinstance(part.function_call.args, str):
@@ -145,15 +141,14 @@ def process_with_tools(query: str) -> str:
                         tool_result = execute_tool("get_weather", function_args)
                         print(f"Tool execution result: {tool_result}")
 
-                        # Send the tool result back to Gemini for a final, natural language response
                         chat_session = model.start_chat()
-                        chat_session.send_message(query) # Send user's original query to establish context
+                        chat_session.send_message(query)
                         
                         response_with_tool_output = chat_session.send_message(
                             genai.protos.Part(
                                 function_response=genai.protos.FunctionResponse(
                                     name="get_weather",
-                                    response={"result": tool_result} # Wrap the string result in a dict for FunctionResponse
+                                    response={"result": tool_result}
                                 )
                             )
                         )
@@ -161,7 +156,6 @@ def process_with_tools(query: str) -> str:
                     else:
                         return f"Unknown tool requested by AI: {function_name}"
                 elif part.text:
-                    # Gemini provided a direct text response (no tool needed or tool not applicable)
                     return part.text
         return "No coherent response from AI (neither text nor function call)."
 
@@ -194,10 +188,17 @@ gradio_app = gr.Interface(
 # Mount Gradio on FastAPI
 app = gr.mount_gradio_app(app, gradio_app, path="/ui")
 
-# Health check
+# === NEW CODE STARTS HERE ===
+# Redirect the root path to the Gradio UI
 @app.get("/")
+async def redirect_to_gradio():
+    return RedirectResponse(url="/ui")
+
+# Optional: Keep the health check at a different path if still desired
+@app.get("/health_status")
 def health_check():
     return {"status": "active", "components": ["fastapi", "gemini", "weather-api", "gradio"]}
+# === NEW CODE ENDS HERE ===
 
 if __name__ == "__main__":
     import uvicorn