Update app.py

app.py

@@ -4,7 +4,8 @@ import os
 import logging
 import json
 from datetime import datetime
-
+import shutil
+from moviepy.editor import VideoFileClip
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
 
@@ -18,23 +19,16 @@ GALLERY_JSON = "/home/user/app/gallery.json"
 # 갤러리 디렉토리 생성
 os.makedirs(GALLERY_DIR, exist_ok=True)
 
-def save_to_gallery(video_path, prompt):
+def save_to_gallery(image_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)
+    new_image_path = os.path.join(GALLERY_DIR, f"{timestamp}.png")
     
-    # 썸네일 생성
-    video = VideoFileClip(new_video_path)
-    video.save_frame(thumbnail_path, t=0)  # 첫 프레임을 썸네일로 저장
-    video.close()
+    # 이미지 파일 복사
+    shutil.copy2(image_path, new_image_path)
     
     # 갤러리 정보 저장
     gallery_info = {
-        "video": new_video_path,
-        "thumbnail": thumbnail_path,
+        "image": new_image_path,
         "prompt": prompt,
         "timestamp": timestamp
     }
@@ -50,13 +44,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, thumbnail_path
+    return new_image_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"], item["video"]) for item in reversed(gallery)]
+        return [(item["image"], item["prompt"]) for item in reversed(gallery)]
     return []
 
 def respond(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):