Update app.py

app.py

@@ -10,35 +10,35 @@ from Gradio_UI import GradioUI
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 # Replace this part in app.py
 @tool
-def my_custom_tool(query: str, generate_visual: bool) -> str:
-    """A tool that searches for information and optionally generates visuals
+def my_custom_tool(search_topic: str, max_results: int) -> str:
+    """A tool that performs a web search and returns formatted results
     Args:
-        query: The search query to find information about
-        generate_visual: Whether to generate a visual representation
+        search_topic: The topic to search for
+        max_results: Number of results to return (1-3)
     """
     try:
-        # Initialize search tool
         search_tool = DuckDuckGoSearchTool()
+        results = search_tool(search_topic)
         
-        # Get results
-        results = search_tool(query)
-        
-        # Format response
         if not results:
-            return f"No information found for '{query}'"
+            return f"No information found about '{search_topic}'"
             
-        result = results[0]
-        response = f"{result['title']}\n{result['snippet']}"
+        # Limit number of results
+        max_results = min(max(1, max_results), 3)
+        results = results[:max_results]
         
-        # Add visual if requested
-        if generate_visual:
-            image_generation_tool(query)
-            response += "\n(A visual has been generated)"
+        # Format response like the time zone example
+        formatted_results = []
+        for result in results:
+            formatted_results.append(
+                f"Title: {result['title']}\n"
+                f"Summary: {result['snippet']}"
+            )
             
-        return response
+        return "\n\n".join(formatted_results)
         
     except Exception as e:
-        return f"Error while searching for '{query}': {str(e)}"
+        return f"Error searching for '{search_topic}': {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: