Update app.py

app.py

@@ -1001,48 +1001,82 @@ def generate_avatar_animation(portrait_image, driving_audio, guidance_scale, inf
         # 상태 업데이트
         status = "아바타 생성 시작..."
         
-        # 이미지 저장
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as fp:
-            portrait_path = fp.name
-            if isinstance(portrait_image, np.ndarray):
-                Image.fromarray(portrait_image).save(portrait_path)
-            else:
-                Image.open(portrait_image).save(portrait_path)
+        # 이미지 처리 - filepath로 받으므로 직접 사용
+        portrait_path = portrait_image
+        
+        # 이미지가 실제로 존재하는지 확인
+        if not os.path.exists(portrait_path):
+            # 만약 numpy array나 PIL 이미지로 왔다면 저장
+            with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as fp:
+                temp_portrait_path = fp.name
+                if isinstance(portrait_image, np.ndarray):
+                    Image.fromarray(portrait_image).save(temp_portrait_path)
+                elif isinstance(portrait_image, Image.Image):
+                    portrait_image.save(temp_portrait_path)
+                else:
+                    return None, None, "❌ 이미지 형식을 인식할 수 없습니다."
+                portrait_path = temp_portrait_path
         
         # 오디오 경로 확인
-        if isinstance(driving_audio, str):
-            audio_path = driving_audio
-        else:
-            audio_path = driving_audio
+        audio_path = driving_audio
         
-        logging.info(f"Portrait: {portrait_path}")
-        logging.info(f"Audio: {audio_path}")
+        # 경로 로깅
+        logging.info(f"Portrait path: {portrait_path}")
+        logging.info(f"Audio path: {audio_path}")
         logging.info(f"Guidance Scale: {guidance_scale}")
         logging.info(f"Inference Steps: {inference_steps}")
         
+        # 파일 존재 확인
+        if not os.path.exists(portrait_path):
+            return None, None, f"❌ 이미지 파일을 찾을 수 없습니다: {portrait_path}"
+        
+        if not os.path.exists(audio_path):
+            return None, None, f"❌ 오디오 파일을 찾을 수 없습니다: {audio_path}"
+        
         # API 호출
         client = Client(AVATAR_API_URL)
         result = client.predict(
-            handle_file(portrait_path),
-            handle_file(audio_path),
+            portrait_path,  # 직접 경로 전달
+            audio_path,     # 직접 경로 전달
             float(guidance_scale),
             float(inference_steps),
             api_name="/generate_animation"
         )
         
-        # 임시 파일 삭제
-        os.unlink(portrait_path)
+        # 임시 파일 삭제 (있는 경우)
+        if 'temp_portrait_path' in locals() and os.path.exists(temp_portrait_path):
+            os.unlink(temp_portrait_path)
         
+        # 결과 처리
         if result and len(result) >= 2:
             animation_result = result[0]
             comparison_result = result[1]
             
-            # 비디오 경로 추출
-            animation_video = animation_result.get("video") if isinstance(animation_result, dict) else None
-            comparison_video = comparison_result.get("video") if isinstance(comparison_result, dict) else None
+            # 결과가 dict인지 직접 경로인지 확인
+            if isinstance(animation_result, dict):
+                animation_video = animation_result.get("video")
+            else:
+                animation_video = animation_result
+                
+            if isinstance(comparison_result, dict):
+                comparison_video = comparison_result.get("video")
+            else:
+                comparison_video = comparison_result
+            
+            # 비디오 파일 존재 확인
+            if animation_video and os.path.exists(str(animation_video)):
+                logging.info(f"Animation video created: {animation_video}")
+            else:
+                logging.warning(f"Animation video not found or invalid: {animation_video}")
+                
+            if comparison_video and os.path.exists(str(comparison_video)):
+                logging.info(f"Comparison video created: {comparison_video}")
+            else:
+                logging.warning(f"Comparison video not found or invalid: {comparison_video}")
             
             return animation_video, comparison_video, "✅ 아바타 애니메이션 생성 완료!"
         else:
+            logging.error(f"Unexpected API response: {result}")
             return None, None, "❌ API 응답이 예상과 다릅니다."
             
     except Exception as e:
@@ -1493,7 +1527,8 @@ with demo:
                         
                         avatar_portrait = gr.Image(
                             label="초상화 이미지 (아무 비율 가능)",
-                            type="filepath"
+                            type="filepath",
+                            sources=["upload"]
                         )
                         
                     with gr.Group(elem_classes="panel-box"):