Update app.py

app.py

@@ -65,17 +65,18 @@ def run_agent(query: str):
 # Gradio interface to show images
 def gradio_interface(query: str):
     # Call the agent to run the query and return the image
-    image_path = run_agent(query)
+    image_paths = run_agent(query)
+    images = [Image.open(image_path) for image_path in image_paths]
     
     # Return the image to Gradio for display
-    return image_path
+    return images
 
 
 # Set up Gradio interface
 gr.Interface(
     fn=gradio_interface,
     inputs="text",  # Text input for the search query
-    outputs=gr.Image(),  # Image output to display the fetched image
+    outputs=gr.Gallery(label="Found Images").style(grid=3),  # Image output to display the fetched image
     title="Figure, Image & Logo Finder",
     description="Enter a query to search for relevant images, logos, or figures for your presentation."
 ).launch()