Update app.py

app.py

@@ -33,24 +33,52 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+# @tool
+# def image_generation_tool(prompt: str) -> Image:
+#    """A tool that generates images using the FLUX.1-schnell model.
+#    Args:
+#        prompt: A string containing the image generation prompt.
+#    Returns:
+#        str: Status message indicating success or error of image generation.
+#    """
+#    try:
+#        tool = Tool.from_space(
+#            "black-forest-labs/FLUX.1-schnell",
+#            name="image_generator",
+#            description="Generate an image from a prompt"
+#        )
+#        result = tool.generate(prompt)
+#        return f"Successfully generated image from prompt: {prompt}"
+#    except Exception as e:
+#        return f"Error generating image from prompt '{prompt}': {str(e)}"
+
+
 @tool
-def image_generation_tool(prompt: str) -> Image:
-   """A tool that generates images using the FLUX.1-schnell model.
+def get_weather(city: str) -> dict:
+   """Fetches current weather data for a given city using open-meteo API.
    Args:
-       prompt: A string containing the image generation prompt.
+       city: Name of the city to get weather for.
    Returns:
-       str: Status message indicating success or error of image generation.
+       dict: Weather data including temperature, humidity, etc.
    """
    try:
-       tool = Tool.from_space(
-           "black-forest-labs/FLUX.1-schnell",
-           name="image_generator",
-           description="Generate an image from a prompt"
-       )
-       result = tool.generate(prompt)
-       return f"Successfully generated image from prompt: {prompt}"
+       # Get coordinates for city using Nominatim
+       geolocator = Nominatim(user_agent="weather_tool")
+       location = geolocator.geocode(city)
+       
+       if not location:
+           raise Exception(f"Could not find coordinates for {city}")
+
+       # Build open-meteo API URL
+       url = f"https://api.open-meteo.com/v1/forecast?latitude={location.latitude}&longitude={location.longitude}&current=temperature_2m,relative_humidity_2m,precipitation,wind_speed_10m"
+       
+       response = requests.get(url)
+       response.raise_for_status()
+       
+       return response.json()['current']
    except Exception as e:
-       return f"Error generating image from prompt '{prompt}': {str(e)}"
+       raise Exception(f"Error fetching weather data: {str(e)}")
+
 
 final_answer = FinalAnswerTool()