Update app.py

app.py

@@ -8,27 +8,21 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 @tool
-def fetch_duckduckgo_summary(query: str) -> str:
+def dream_image_generator(dream_desc: str) -> str:
     """
-    Fetches a short summary from web search results.
+    Generates an image of an dream based on user description.
 
     Args:
-        query (str): The search query.
+        dream_desc (str): Dream description.
 
     Returns:
-        str: A brief summary of the top search results.
+        str: URL or path to the AI-generated dream image.
     """
     try:
         # Perform a DuckDuckGo search
-        search_tool = DuckDuckGoSearchTool()
-        search_results = search_tool(query=query, max_results=3)
+        dream_image = image_generation_tool(f"A surreal dream {dream_desc}")
 
-        # Extract the top results
-        if search_results and isinstance(search_results, list):
-            summary = "\n".join([f"{res['title']}: {res.get('body', 'No description')}" for res in search_results])
-            return summary
-        else:
-            return "No relevant results found."
+        return dream_image
 
     except Exception as e:
         return f"Error fetching search results: {str(e)}"
@@ -70,7 +64,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[generate_image_from_wikipedia, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
+    tools=[dream_image_generator, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,