Update app.py

app.py

@@ -75,6 +75,9 @@ def find_top_images(text_query):
 
     return all_results
 
+def enlarge_image(image_url):
+    return gr.Image.update(value=image_url, interactive=False, visible=True)
+
 # Gradio interface setup
 with gr.Blocks() as demo:
     gr.Markdown("Query Results using OpenAI CLIP (openai/clip-vit-large-patch14-336)")
@@ -82,12 +85,18 @@ with gr.Blocks() as demo:
 
     # Output row for the model
     with gr.Row():
-        outputs = [gr.Image(label=f"{model_name} - Top Image {i+1}", width=200, height=200) for i in range(5)]
+        image_elements = []
+        for i in range(5):
+            with gr.Column():
+                image = gr.Image(label=f"Image {i+1}", width=200, height=200)
+                image_elements.append(image)
+                # Add a button to enlarge the image
+                gr.Button(f"Enlarge Image {i+1}").click(fn=enlarge_image, inputs=image, outputs=image)
 
     text_input.submit(
         fn=find_top_images,
         inputs=text_input,
-        outputs=outputs
+        outputs=image_elements
     )
 
 demo.launch()