Update app.py

app.py

@@ -9,32 +9,35 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1: str, arg2: int) -> str:
-    """A tool that searches for 3 PDF files related to the user's request using DuckDuckGoSearchTool.
+def search_image_urls(query: str, num_images: int) -> str:
+    """A tool that searches for image URLs using DuckDuckGo.
     
     Args:
-        arg1: the first argument (string), interpreted as the search query.
-        arg2: the second argument (integer), potentially controlling how many results to fetch.
+        query: The user-provided search keyword(s), e.g. 'cat'
+        num_images: The number of image URLs to return
     """
-    # We'll use DuckDuckGoSearchRun from langchain to perform the actual search.
-    search = DuckDuckGoSearchRun()
-
-    # Construct the query, searching specifically for PDF documents.
-    query = f"{arg1} filetype:pdf"
-
-    # Use arg2 to limit the number of results to fetch (if you prefer, you can hardcode num_results=3).
-    num_results = min(arg2, 3) if arg2 > 0 else 3
+    from duckduckgo_search import ddg_images
+    
+    # Validate number of images
+    if num_images < 1:
+        num_images = 1
+    if num_images > 5:
+        # Let's limit to 5 for demonstration
+        num_images = 5
+
+    # Perform the image search
+    results = ddg_images(query, max_results=num_images)
+    if not results:
+        return f"No images found for '{query}'."
 
-    # Execute the search
-    results = search.run(query, num_results=num_results)
+    # Format results in a neat way
+    output_lines = []
+    for index, result in enumerate(results, start=1):
+        thumbnail = result.get("image", "No image URL found")
+        title = result.get("title", "No title")
+        output_lines.append(f"{index}. {title}\n   {thumbnail}")
 
-    # If results is a string describing the search outcome, just return it.
-    # If results is a list of links, you might parse them. For simplicity, we'll cast it to string and return.
-    if not results:
-        return "No PDF files found."
-    
-    # Convert the results to a string for returning; you can format or parse further if desired.
-    return f"Top {num_results} PDF results for '{arg1}':\n\n{str(results)}"
+    return "\n\n".join(output_lines)
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: