Update app.py

app.py

@@ -47,19 +47,13 @@ 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")
-    thumbnail_path = os.path.join(GALLERY_DIR, f"{timestamp}_thumb.jpg")
     
     # 비디오 파일 복사
     shutil.copy2(video_path, new_video_path)
     
-    # 썸네일 생성
-    video = VideoFileClip(new_video_path)
-    video.save_frame(thumbnail_path, t=0)  # 첫 프레임을 썸네일로 저장
-    
     # 갤러리 정보 저장
     gallery_info = {
         "video": new_video_path,
-        "thumbnail": thumbnail_path,
         "prompt": prompt,
         "timestamp": timestamp
     }
@@ -75,16 +69,15 @@ def save_to_gallery(video_path, prompt):
     with open(GALLERY_JSON, "w") as f:
         json.dump(gallery, f, indent=2)
     
-    return new_video_path, thumbnail_path
+    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["thumbnail"], item["prompt"]) for item in reversed(gallery)]
+        return [(item["video"], 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}")
@@ -183,7 +176,6 @@ 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",
@@ -192,23 +184,24 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
             columns=[5],
             rows=[3],
             object_fit="contain",
-            height="auto"
+            height="auto",
+            preview=True
         )
         selected_video = gr.Video(label="Selected Video")
         refresh_btn = gr.Button("Refresh Gallery")
-
+    
 
     def update_gallery():
         return load_gallery()
 
     def show_video(evt: gr.SelectData):
-        if isinstance(evt.value, tuple) and len(evt.value) > 1:
-            return evt.value[0]  # 첫 번째 요소가 비디오 경로일 가능성이 높습니다
+        if isinstance(evt.value, tuple) and len(evt.value) > 0:
+            return evt.value[0]  # 첫 번째 요소가 비디오 경로입니다
         elif isinstance(evt.value, str):
             return evt.value  # 값이 직접 비디오 경로일 수 있습니다
         else:
             print(f"Unexpected evt.value structure: {evt.value}")
-            return None  # 또는 기본 비디오 경로를 반환
+            return None
     
     refresh_btn.click(fn=update_gallery, inputs=None, outputs=gallery)
     demo.load(fn=update_gallery, inputs=None, outputs=gallery)