Update app.py

app.py

@@ -71,13 +71,15 @@ def save_to_gallery(video_path, prompt):
     
     return new_video_path
 
+
+    
 def load_gallery():
     if os.path.exists(GALLERY_JSON):
         with open(GALLERY_JSON, "r") as f:
             gallery = json.load(f)
-        return [(item["video"], item["prompt"]) for item in reversed(gallery)]
+        return [{"image": item["video"], "caption": item["prompt"]} for item in reversed(gallery)]
     return []
-
+    
 def respond(image, prompt, steps, cfg_scale, eta, fs, seed, video_length):
     logging.info(f"Received prompt: {prompt}, steps: {steps}, cfg_scale: {cfg_scale}, "
                  f"eta: {eta}, fs: {fs}, seed: {seed}, video_length: {video_length}")
@@ -102,11 +104,9 @@ def respond(image, prompt, steps, cfg_scale, eta, fs, seed, video_length):
             api_name="/infer"
         )
         logging.info("API response received: %s", result)
-        
-
 
         if isinstance(result, dict) and 'video' in result:
-            saved_video_path, _ = save_to_gallery(result['video'], prompt)
+            saved_video_path = save_to_gallery(result['video'], prompt)
             return saved_video_path
         else:
             raise ValueError("예상치 못한 API 응답 형식")
@@ -176,6 +176,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                         inputs=[],
                         outputs=[input_text, input_image]
                     )
+
     with gr.Tab("Gallery"):
         gallery = gr.Gallery(
             label="Generated Videos",
@@ -186,7 +187,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
             object_fit="contain",
             height="auto",
             preview=True
-        )
+        ).style(grid=3)
         selected_video = gr.Video(label="Selected Video")
         refresh_btn = gr.Button("Refresh Gallery")
     
@@ -195,13 +196,14 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         return load_gallery()
 
     def show_video(evt: gr.SelectData):
-        if isinstance(evt.value, tuple) and len(evt.value) > 0:
-            return evt.value[0]  # 첫 번째 요소가 비디오 경로입니다
+        if isinstance(evt.value, dict) and 'image' in evt.value:
+            return evt.value['image']['path']
         elif isinstance(evt.value, str):
-            return evt.value  # 값이 직접 비디오 경로일 수 있습니다
+            return evt.value
         else:
-            print(f"Unexpected evt.value structure: {evt.value}")
+            logging.error(f"Unexpected evt.value structure: {evt.value}")
             return None
+
     
     refresh_btn.click(fn=update_gallery, inputs=None, outputs=gallery)
     demo.load(fn=update_gallery, inputs=None, outputs=gallery)