Update app.py

app.py

@@ -11,36 +11,34 @@ from Gradio_UI import GradioUI
 # 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 related images
+    """A tool that searches for information and optionally generates visuals
     Args:
         query: The search query to find information about
         generate_visual: Whether to generate a visual representation
     """
     try:
-        # First do web search
+        # Initialize search tool
         search_tool = DuckDuckGoSearchTool()
-        print(f"Searching for: {query}")  # Debug print
-        search_results = search_tool(query)
         
-        if not search_results:
-            return "No information found for your query."
+        # Get results
+        results = search_tool(query)
         
-        # Get first result
-        result = search_results[0]
-        response = f"Here's what I found:\n{result['title']}\n{result['snippet']}"
+        # Format response
+        if not results:
+            return f"No information found for '{query}'"
+            
+        result = results[0]
+        response = f"{result['title']}\n{result['snippet']}"
         
-        # Generate image if requested
+        # Add visual if requested
         if generate_visual:
-            try:
-                image_generation_tool(query)
-                response += "\n\nI've generated a visual for your query."
-            except Exception as img_err:
-                response += f"\n\nNote: Image generation failed: {str(img_err)}"
-                
+            image_generation_tool(query)
+            response += "\n(A visual has been generated)"
+            
         return response
         
     except Exception as e:
-        return f"An error occurred while searching: {str(e)}"
+        return f"Error while searching for '{query}': {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: