Update app.py

app.py

@@ -1,12 +1,22 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
-import datetime
-import requests
-import pytz
-import yaml
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 from tools.final_answer import FinalAnswerTool
+from Gradio_UI import GradioUI  # assuming you created this file
 
-from Gradio_UI import GradioUI
+# You must define or import image_generator
+image_generator = HfApiModel("stabilityai/stable-diffusion-2")  # Example model
 
+# Create the agent and register tools
+tools = [
+    DuckDuckGoSearchTool(),
+    FinalAnswerTool(),
+    image_generator  # This acts as a tool
+]
 
-image = image_generator(prompt=" A high-quality, photorealistic image of a cat sleeping on a windowsill.")
-final_answer(image)
+agent = CodeAgent(tools=tools)
+
+# Optional: You can run a test outside Gradio
+# image = image_generator(prompt="A high-quality, photorealistic image of a cat sleeping on a windowsill.")
+# final_answer(image)
+
+# Launch Gradio UI
+GradioUI(agent).launch()