Update app.py

app.py

@@ -54,7 +54,7 @@ def generate_animation(image, audio, guidance_scale, steps, progress=gr.Progress
             raise ValueError("이미지와 오디오를 모두 업로드하세요.")
 
         progress(0.05, desc="파일 준비")
-        client = Client(API_URL)                # ← timeout 인자 삭제
+        client = Client(API_URL)
 
         progress(0.15, desc="서버 호출 중… (수 분 소요 가능)")
         result = client.predict(
@@ -66,15 +66,42 @@ def generate_animation(image, audio, guidance_scale, steps, progress=gr.Progress
         )
 
         progress(0.95, desc="결과 정리")
+        
+        # 결과 처리 - dict 형태 처리 추가
+        def extract_video_path(obj):
+            """비디오 객체에서 경로 추출"""
+            if isinstance(obj, str):
+                return obj
+            elif isinstance(obj, dict):
+                # Gradio의 FileData dict 처리
+                if 'path' in obj:
+                    return obj['path']
+                elif 'url' in obj:
+                    return obj['url']
+                elif 'name' in obj:
+                    return obj['name']
+                else:
+                    log.warning(f"Unexpected dict structure: {obj.keys()}")
+                    return None
+            else:
+                log.warning(f"Unexpected type: {type(obj)}")
+                return None
+        
         if isinstance(result, (list, tuple)) and len(result) >= 2:
-            anim, comp = result[0], result[1]
-            logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] 성공")
-            return anim, comp, "\n".join(logs)
+            anim_path = extract_video_path(result[0])
+            comp_path = extract_video_path(result[1])
+            
+            if anim_path and comp_path:
+                logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] 성공")
+                return anim_path, comp_path, "\n".join(logs)
+            else:
+                raise RuntimeError(f"비디오 경로 추출 실패: {result}")
         else:
             raise RuntimeError(f"예상치 못한 반환 형식: {type(result)}")
 
     except Exception as e:
         logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] 오류: {e}")
+        log.error(f"Animation generation error: {e}", exc_info=True)
         return None, None, "\n".join(logs)
 
 # ─────────────────────── UI 정의 ────────────────────── #
@@ -114,4 +141,4 @@ if __name__ == "__main__":
         server_name="0.0.0.0",
         server_port=7860,
         show_api=False
-    )
+    )