Update app.py

app.py

@@ -10,7 +10,7 @@ from bs4 import BeautifulSoup
 
 # Define a function to extract image URLs from DuckDuckGo search
 @tool
-def search_images_for_presentation(query: str) -> list:
+def search_images_for_presentation(query: str) -> str:
     """
     Searches for images related to a given query.
 
@@ -32,7 +32,7 @@ def search_images_for_presentation(query: str) -> list:
     soup = BeautifulSoup(helium.get_driver().page_source, "html.parser")
     images = [img["src"] for img in soup.find_all("img") if "src" in img.attrs]
 
-    return images[:5]  # Return the top 5 image URLs
+    return images[0]  # Return the top 5 image URLs
 
 
 # Initialize agent