Update app.py

app.py

@@ -177,38 +177,27 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                         outputs=[input_text, input_image]
                     )
 
+
+
     with gr.Tab("Gallery"):
-        gallery = gr.Gallery(
-            label="Generated Videos",
-            show_label=False,
-            elem_id="gallery",
-            columns=5,
-            rows=3,
-            object_fit="contain",
-            height="auto",
-            preview=True
-        )
+        gallery_html = gr.HTML()
         selected_video = gr.Video(label="Selected Video")
         refresh_btn = gr.Button("Refresh Gallery")
     
-
-
     def update_gallery():
-        return load_gallery()
+        gallery_items = load_gallery()
+        html = "<div style='display: flex; flex-wrap: wrap;'>"
+        for item in gallery_items:
+            html += f"<div style='margin: 10px;'><video width='200' height='200' controls><source src='{item['image']}' type='video/mp4'></video><p>{item['caption']}</p></div>"
+        html += "</div>"
+        return html
 
     def show_video(evt: gr.SelectData):
-        if isinstance(evt.value, dict) and 'image' in evt.value:
-            return evt.value['image']['path']
-        elif isinstance(evt.value, str):
-            return evt.value
-        else:
-            logging.error(f"Unexpected evt.value structure: {evt.value}")
-            return None
+        return evt.target.src
 
-    
-    refresh_btn.click(fn=update_gallery, inputs=None, outputs=gallery)
-    demo.load(fn=update_gallery, inputs=None, outputs=gallery)
-    gallery.select(show_video, None, selected_video)
+    refresh_btn.click(fn=update_gallery, inputs=None, outputs=gallery_html)
+    demo.load(fn=update_gallery, inputs=None, outputs=gallery_html)
+    gallery_html.click(show_video, None, selected_video)
 
     input_text.submit(
         fn=respond, 
@@ -217,11 +206,9 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     ).then(
         fn=update_gallery,
         inputs=None,
-        outputs=gallery
+        outputs=gallery_html
     )
 
 if __name__ == "__main__":
     demo.launch()
-
-
-
+