Update app.py

removed the image generation tool

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool, image_generation_tool
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
@@ -6,6 +6,12 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
+image_generation_tool = tool.from_space(
+    "black-forest-labs/FLUX.1-schnell",
+    name="image_generator",
+    description="Generate an image from a prompt"
+)
+
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -52,45 +58,8 @@ def my_custom_tool(arg1: str, arg2: int) -> str:
     
     return "\n".join(messages)
 
-@tool
-def promote_chakwal(arg1: str, arg2: int) -> str:
-    """
-    A tool that promotes tourism and investment in Chakwal by fetching live information 
-    and generating a visual representation of the city's attractions.
-    
-    Args:
-        arg1: A custom note or call-to-action message.
-        arg2: The number of search results to include from DuckDuckGo regarding tourism/investment.
-    """
-    output_lines = []
-    
-    # Use DuckDuckGoSearchTool to fetch current news or information on Chakwal tourism and investment
-    duck_tool = DuckDuckGoSearchTool()
-    search_query = "Chakwal tourism investment latest news attractions"
-    search_results = duck_tool.run(search_query)
-    
-    # Process and include search results (assuming a list of result dictionaries)
-    if isinstance(search_results, list):
-        output_lines.append("Latest Information on Chakwal:")
-        for idx, result in enumerate(search_results[:arg2]):
-            title = result.get("title", "No Title")
-            snippet = result.get("snippet", "No snippet available.")
-            output_lines.append(f"{idx+1}. {title} - {snippet}")
-    else:
-        output_lines.append(f"Search Results: {search_results}")
-    
-    # Use the text-to-image generation tool to create an attractive image prompt for Chakwal
-    image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-    image_prompt = "A stunning view of Chakwal with lush landscapes, historic landmarks, and vibrant local culture"
-    generated_image = image_generation_tool.run(image_prompt)
-    output_lines.append("\nVisual Promotion:")
-    output_lines.append(f"Generated Image URL/Result: {generated_image}")
-    
-    # Append any additional note from arg1
-    if arg1:
-        output_lines.append(f"\nAdditional Note: {arg1}")
-    
-    return "\n".join(output_lines)
+
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: