Update app.py

app.py

@@ -46,14 +46,15 @@ os.makedirs(GALLERY_DIR, exist_ok=True)
 
 def save_to_gallery(video_path, prompt):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    new_video_path = os.path.join(GALLERY_DIR, f"{timestamp}.mp4")
+    new_video_name = f"{timestamp}.mp4"
+    new_video_path = os.path.join(GALLERY_DIR, new_video_name)
     
     # 비디오 파일 복사
     shutil.copy2(video_path, new_video_path)
     
     # 갤러리 정보 저장
     gallery_info = {
-        "video": new_video_path,
+        "video": f"/file={new_video_path}",  # 상대 경로 사용
         "prompt": prompt,
         "timestamp": timestamp
     }
@@ -70,7 +71,7 @@ def save_to_gallery(video_path, prompt):
         json.dump(gallery, f, indent=2)
     
     return new_video_path
-
+    
 def load_gallery():
     if os.path.exists(GALLERY_JSON):
         with open(GALLERY_JSON, "r") as f:
@@ -78,8 +79,7 @@ def load_gallery():
         return [(item["video"], item["prompt"]) for item in reversed(gallery)]
     return []
 
-def update_gallery():
-    return load_gallery()
+
 
 def respond(image, prompt, steps, cfg_scale, eta, fs, seed, video_length):
     logging.info(f"Received prompt: {prompt}, steps: {steps}, cfg_scale: {cfg_scale}, "
@@ -187,21 +187,29 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                     )
 
     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"
-        )
+        gallery_html = gr.HTML()
         selected_video = gr.Video(label="Selected Video")
         refresh_btn = gr.Button("Refresh Gallery")
 
-    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)
+    def update_gallery():
+        gallery_items = load_gallery()
+        html = "<div style='display: flex; flex-wrap: wrap;'>"
+        for video_path, prompt in gallery_items:
+            html += f"""
+            <div style='margin: 10px; text-align: center;'>
+                <video width='200' height='200' controls>
+                    <source src='{video_path}' type='video/mp4'>
+                </video>
+                <p>{prompt[:50]}...</p>
+            </div>
+            """
+        html += "</div>"
+        return html
+
+    refresh_btn.click(fn=update_gallery, inputs=None, outputs=gallery_html)
+    demo.load(fn=update_gallery, inputs=None, outputs=gallery_html)
+    
+
 
     input_text.submit(
         fn=respond,