Update app.py

app.py

@@ -11,6 +11,7 @@ import shutil
 import httpx
 import time
 import base64
+from gradio_client import Client, handle_file
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
@@ -78,6 +79,7 @@ def load_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}")
@@ -85,28 +87,27 @@ def respond(image, prompt, steps, cfg_scale, eta, fs, seed, video_length):
     try:
         # 이미지 파일 처리
         if image is not None:
-            with open(image, "rb") as f:
-                image_data = base64.b64encode(f.read()).decode('utf-8')
+            image_file = handle_file(image)
         else:
-            image_data = None
+            image_file = None
 
         # 비디오 생성 요청
         result = api_client.predict(
-            image_data,  # base64로 인코딩된 이미지 데이터
-            prompt,
-            steps,
-            cfg_scale,
-            eta,
-            fs,
-            seed,
-            video_length,
+            image=image_file,
+            prompt=prompt,
+            steps=steps,
+            cfg_scale=cfg_scale,
+            eta=eta,
+            fs=fs,
+            seed=seed,
+            video_length=video_length,
             api_name="/infer"
         )
         logging.info("API response received: %s", result)
         
         # 결과 확인 및 처리
-        if isinstance(result, str) and result.endswith('.mp4'):
-            saved_video_path = save_to_gallery(result, prompt)
+        if isinstance(result, dict) and 'video' in result:
+            saved_video_path = save_to_gallery(result['video'], prompt)
             return saved_video_path
         else:
             raise ValueError("Unexpected API response format")