Update app.py

app.py

@@ -7,6 +7,8 @@ from datetime import datetime
 import tempfile
 import numpy as np
 from PIL import Image
+import shutil
+from gradio_client import Client, File
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
 
@@ -28,8 +30,6 @@ def save_to_gallery(video_path, prompt):
     # 비디오 파일 복사
     shutil.copy2(video_path, new_video_path)
     
-
-    
     # 갤러리 정보 저장
     gallery_info = {
         "video": new_video_path,
@@ -50,6 +50,7 @@ def save_to_gallery(video_path, prompt):
     
     return new_video_path
 
+
 def load_gallery():
     if os.path.exists(GALLERY_JSON):
         with open(GALLERY_JSON, "r") as f:
@@ -57,9 +58,6 @@ 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}")
@@ -67,16 +65,16 @@ def respond(image, prompt, steps, cfg_scale, eta, fs, seed, video_length):
     try:
         # 이미지 처리
         if isinstance(image, str):  # 파일 경로인 경우
-            image_path = image
+            image_file = File(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)
-                image_path = temp_file.name
+                image_file = File(temp_file.name)
 
         # 비디오 생성 요청
         result = api_client.predict(
-            image_path,
+            image_file,
             prompt,
             steps,
             cfg_scale,
@@ -90,7 +88,7 @@ def respond(image, prompt, steps, cfg_scale, eta, fs, seed, video_length):
         
         # 임시 파일 삭제 (numpy array였을 경우에만)
         if not isinstance(image, str):
-            os.unlink(image_path)
+            os.unlink(image_file.name)
         
         # 결과 확인 및 처리
         if isinstance(result, str) and result.endswith('.mp4'):
@@ -121,7 +119,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="numpy")
+            input_image = gr.Image(label="Upload an image", type="filepath")
             input_text = gr.Textbox(label="Enter your prompt for video generation")
             output_video = gr.Video(label="Generated Video")