Update app.py

app.py

@@ -12,7 +12,8 @@ import httpx
 import time
 import base64
 from gradio_client import Client, handle_file
-
+import cv2
+from moviepy.editor import VideoFileClip
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
 
@@ -46,13 +47,19 @@ 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
     }
@@ -68,14 +75,13 @@ def save_to_gallery(video_path, prompt):
     with open(GALLERY_JSON, "w") as f:
         json.dump(gallery, f, indent=2)
     
-    return new_video_path
-
+    return new_video_path, thumbnail_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 [(item["thumbnail"], item["prompt"], item["video"]) for item in reversed(gallery)]
     return []
 
 
@@ -128,8 +134,9 @@ examples = [
     ["A fantasy map of a fictional world, with detailed terrain and cities.", "q19.webp"]
 ]
 
-def use_prompt(prompt):
-    return prompt
+def use_prompt_and_image(prompt, image):
+    return prompt, image
+
 
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     with gr.Tab("Generate"):
@@ -145,21 +152,23 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
             fs = gr.Slider(minimum=1, maximum=30, step=1, label="FPS", value=8)
             seed = gr.Slider(minimum=0, maximum=1000000, step=1, label="Seed", value=123)
             video_length = gr.Slider(minimum=1, maximum=10, step=1, label="Video Length (seconds)", value=2)
-        
+
         with gr.Row():
             for prompt, image_file in examples:
                 with gr.Column():
-                    gr.Image(image_file, label=prompt[:50] + "...")  # 프롬프트의 처음 50자만 표시
-                    gr.Button("Use this prompt").click(
-                        fn=use_prompt,
+                    gr.Image(image_file, label=prompt[:50] + "...")
+                    gr.Button("Use this example").click(
+                        fn=use_prompt_and_image,
                         inputs=[],
-                        outputs=input_text,
+                        outputs=[input_text, input_image],
                         api_name=False
                     ).then(
-                        lambda x=prompt: x,
+                        lambda p=prompt, i=image_file: (p, i),
                         inputs=[],
-                        outputs=input_text
+                        outputs=[input_text, input_image]
                     )
+        
+
     
     with gr.Tab("Gallery"):
         gallery = gr.Gallery(