Update app.py

app.py

@@ -4,7 +4,9 @@ import os
 import logging
 import json
 from datetime import datetime
-
+import tempfile
+import numpy as np
+from PIL import Image
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
 
@@ -53,14 +55,24 @@ def load_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}")
     
     try:
+        # 이미지를 임시 파일로 저장
+        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
+            if isinstance(image, np.ndarray):
+                img = Image.fromarray(image.astype('uint8'), 'RGB')
+                img.save(temp_file.name)
+            else:
+                temp_file.write(image)
+            temp_file_path = temp_file.name
+
         # 비디오 생성 요청
         result = api_client.predict(
-            image=handle_file(image),
+            image=temp_file_path,
             prompt=prompt,
             steps=steps,
             cfg_scale=cfg_scale,
@@ -72,6 +84,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)
+        
         # 결과 확인 및 처리
         if isinstance(result, str) and result.endswith('.mp4'):
             saved_video_path = save_to_gallery(result, prompt)