Update app.py

app.py

@@ -26,8 +26,9 @@ def save_to_gallery(video_path, prompt):
     new_video_path = os.path.join(GALLERY_DIR, f"{timestamp}.mp4")
     
     # 비디오 파일 복사
-    with open(video_path, "rb") as src, open(new_video_path, "wb") as dst:
-        dst.write(src.read())
+    shutil.copy2(video_path, new_video_path)
+    
+
     
     # 갤러리 정보 저장
     gallery_info = {
@@ -58,23 +59,24 @@ def load_gallery():
 
 
 
+
 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}")
     
     try:
-        # 이미지를 임시 파일로 저장
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
-            if isinstance(image, np.ndarray):
+        # 이미지 처리
+        if isinstance(image, str):  # 파일 경로인 경우
+            image_path = image
+        else:  # numpy array인 경우
+            with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
                 img = Image.fromarray(image.astype('uint8'), 'RGB')
                 img.save(temp_file.name)
-            else:
-                temp_file.write(image)
-            temp_file_path = temp_file.name
+                image_path = temp_file.name
 
         # 비디오 생성 요청
         result = api_client.predict(
-            temp_file_path,  # 직접 파일 경로 전달
+            image_path,
             prompt,
             steps,
             cfg_scale,
@@ -86,8 +88,9 @@ def respond(image, prompt, steps, cfg_scale, eta, fs, seed, video_length):
         )
         logging.info("API response received: %s", result)
         
-        # 임시 파일 삭제
-        os.unlink(temp_file_path)
+        # 임시 파일 삭제 (numpy array였을 경우에만)
+        if not isinstance(image, str):
+            os.unlink(image_path)
         
         # 결과 확인 및 처리
         if isinstance(result, str) and result.endswith('.mp4'):
@@ -118,7 +121,7 @@ def use_prompt(prompt):
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     with gr.Tab("Generate"):
         with gr.Row():
-            input_image = gr.Image(label="Upload an image", type="filepath")
+            input_image = gr.Image(label="Upload an image", type="numpy")
             input_text = gr.Textbox(label="Enter your prompt for video generation")
             output_video = gr.Video(label="Generated Video")