Update app.py

app.py

@@ -12,23 +12,29 @@ logging.basicConfig(level=logging.INFO)
 api_client = Client("http://211.233.58.202:7960/")
 
 # 갤러리 저장 디렉토리 설정
-GALLERY_DIR = "gallery"
-GALLERY_JSON = "gallery.json"
+GALLERY_DIR = "/home/user/app/gallery"
+GALLERY_JSON = "/home/user/app/gallery.json"
 
 # 갤러리 디렉토리 생성
 os.makedirs(GALLERY_DIR, exist_ok=True)
 
-def save_to_gallery(image_path, prompt):
+def save_to_gallery(video_path, prompt):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    new_image_path = os.path.join(GALLERY_DIR, f"{timestamp}.png")
+    new_video_path = os.path.join(GALLERY_DIR, f"{timestamp}.mp4")
+    thumbnail_path = os.path.join(GALLERY_DIR, f"{timestamp}_thumb.jpg")
     
-    # 이미지 파일 복사
-    with open(image_path, "rb") as src, open(new_image_path, "wb") as dst:
-        dst.write(src.read())
+    # 비디오 파일 복사
+    shutil.copy2(video_path, new_video_path)
+    
+    # 썸네일 생성
+    video = VideoFileClip(new_video_path)
+    video.save_frame(thumbnail_path, t=0)  # 첫 프레임을 썸네일로 저장
+    video.close()
     
     # 갤러리 정보 저장
     gallery_info = {
-        "image": new_image_path,
+        "video": new_video_path,
+        "thumbnail": thumbnail_path,
         "prompt": prompt,
         "timestamp": timestamp
     }
@@ -44,13 +50,13 @@ def save_to_gallery(image_path, prompt):
     with open(GALLERY_JSON, "w") as f:
         json.dump(gallery, f, indent=2)
     
-    return new_image_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["image"], item["prompt"]) for item in reversed(gallery)]
+        return [(item["thumbnail"], item["prompt"], item["video"]) for item in reversed(gallery)]
     return []
 
 def respond(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):