Update app.py

app.py

@@ -5,7 +5,7 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 from bs4 import BeautifulSoup
-from smolagents import GradioUI
+from Gradio_UI import GradioUI
 import requests
 from io import BytesIO
 from PIL import Image
@@ -86,9 +86,24 @@ def get_all_matches_for_a_date(date: str) -> str:
     except Exception as e:
         return f"Error fetching matches for {date}: {str(e)}"
 
-        
 web_serach_tool = DuckDuckGoSearchTool()
 
+@tool
+def generate_image(imageprompt: str) -> Image.Image:
+    """A tool that generates an image based on a given prompt.
+    Args:
+        imageprompt: The description of the image to generate.
+    Returns:
+        The generated image as a PIL Image object.
+    """
+    url = f"https://image.pollinations.ai/prompt/{imageprompt}"
+    response = requests.get(url)
+    
+    if response.status_code == 200:
+        return Image.open(BytesIO(response.content))
+    else:
+        raise Exception("Failed to generate image.")
+
 
 @tool
 def get_news(topic: str) -> str:
@@ -118,12 +133,16 @@ model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',#
 custom_role_conversions=None,
 )
 
+
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[get_all_matches_for_a_date, get_news, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_all_matches_for_a_date,get_news, generate_image, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,