Update app.py

app.py

@@ -1,3 +1,6 @@
+import micropip
+await micropip.install(["smolagents", "requests", "pytz", "pyyaml", "beautifulsoup4", "pillow"])
+
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
@@ -9,6 +12,9 @@ from io import BytesIO
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+# Load image generation tool
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
 # Weather Tool (using wttr.in API, no API key needed)
 @tool
 def get_weather(city: str) -> str:
@@ -50,7 +56,10 @@ def generate_image(prompt: str) -> str:
     Args:
         prompt: A description of the image to generate.
     """
-    return image_generation_tool.run(prompt)
+    try:
+        return image_generation_tool.run(prompt)
+    except Exception as e:
+        return f"Error generating image: {str(e)}"
 
 # Timezone Tool (already in your code)
 @tool