Update app.py

app.py

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