Update app.py

app.py

@@ -1,250 +1,115 @@
+"""
+Gradio UI that calls the remote animation server.
+Set ANIM_API_URL env-var when deploying:
+   ANIM_API_URL=http://211.233.58.201:7788/
+"""
+
+import os, time, logging
+from datetime import datetime
+
 import gradio as gr
+import httpx
 from gradio_client import Client, handle_file
-import logging
-import traceback
-from datetime import datetime
-import json
 
-# 로깅 설정
 logging.basicConfig(
-    level=logging.DEBUG,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+    level=logging.INFO,
+    format="%(asctime)s [%(levelname)s] %(message)s"
 )
-logger = logging.getLogger(__name__)
+log = logging.getLogger(__name__)
 
-# API 엔드포인트
-API_URL = "http://211.233.58.201:7788/"
+# ------------------------------------------------------------------ #
+# 기본 접속 주소 – 내부(Loopback) 여부에 따라 자동 판단 + 환경변수 재정의
+# ------------------------------------------------------------------ #
+_DEFAULT_URL = "http://127.0.0.1:7788/"
+REMOTE_URL   = "http://211.233.58.201:7788/"
+API_URL = os.getenv("ANIM_API_URL", REMOTE_URL)
+
+if API_URL == REMOTE_URL and os.getenv("HF_SPACE") == "true":
+    # HuggingFace 스페이스 컨테이너 안 → 외부 서버 접속
+    pass
+elif "127.0.0.1" in API_URL and os.getenv("HF_SPACE") == "true":
+    raise RuntimeError("HF Space 내부에서는 공인 IP 또는 도메인으로 ANIM_API_URL을 지정해야 합니다.")
+
+# ------------------------------------------------------------------ #
+def make_client() -> Client:
+    """커넥션 타임아웃을 30s로 늘려 재시도 가능성 확보"""
+    timeout = httpx.Timeout(connect=30.0, read=120.0)
+    return Client(API_URL, timeout=timeout, retries=3)
 
 def test_api_connection():
-    """API 연결 테스트"""
+    now = datetime.now().strftime("%H:%M:%S")
     try:
-        client = Client(API_URL)
-        logger.info(f"Successfully connected to API: {API_URL}")
-        return True, "API 연결 성공!"
+        client = make_client()
+        # readiness-probe
+        ready = client.get('/healthz')['ready']
+        msg = f"[{now}] 서버 연결 성공 ✅ (ready={ready})"
+        log.info(msg)
+        return True, msg
     except Exception as e:
-        logger.error(f"Failed to connect to API: {str(e)}")
-        return False, f"API 연결 실패: {str(e)}"
+        msg = f"[{now}] 서버 연결 실패 ❌ : {e}"
+        log.error(msg)
+        return False, msg
 
+# ------------------------------------------------------------------ #
 def generate_animation(image, audio, guidance_scale, steps, progress=gr.Progress()):
-    """애니메이션 생성 함수"""
-    logger.info("=== 애니메이션 생성 시작 ===")
-    logs = []
-    
+    start = datetime.now().strftime("%H:%M:%S")
+    logs = [f"[{start}] 요청 시작"]
     try:
-        # 입력 파라미터 로깅
-        log_msg = f"입력 파라미터:\n- Image: {image}\n- Audio: {audio}\n- Guidance Scale: {guidance_scale}\n- Steps: {steps}"
-        logger.info(log_msg)
-        logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] {log_msg}")
-        
-        # 입력 검증
-        if image is None:
-            error_msg = "이미지가 제공되지 않았습니다."
-            logger.error(error_msg)
-            logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] ERROR: {error_msg}")
-            return None, None, "\n".join(logs)
-        
-        if audio is None:
-            error_msg = "오디오가 제공되지 않았습니다."
-            logger.error(error_msg)
-            logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] ERROR: {error_msg}")
-            return None, None, "\n".join(logs)
-        
-        # Progress 업데이트
-        progress(0.1, desc="API 클라이언트 초기화 중...")
-        
-        # API 클라이언트 생성
-        client = Client(API_URL)
-        logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] API 클라이언트 생성 완료")
-        
-        # Progress 업데이트
-        progress(0.3, desc="파일 핸들 생성 중...")
-        
-        # 파일 핸들 생성
-        image_handle = handle_file(image)
-        audio_handle = handle_file(audio)
-        
-        log_msg = f"파일 핸들 생성 완료:\n- Image handle: {type(image_handle)}\n- Audio handle: {type(audio_handle)}"
-        logger.debug(log_msg)
-        logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] {log_msg}")
-        
-        # Progress 업데이트
-        progress(0.5, desc="API 호출 중... (이 과정은 시간이 걸릴 수 있습니다)")
-        
-        # API 호출
-        logger.info("API 호출 시작")
-        logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] API 호출 시작...")
-        
+        if image is None or audio is None:
+            raise ValueError("이미지와 오디오를 모두 업로드하세요.")
+
+        progress(0.05, desc="파일 준비")
+        client = make_client()
+        progress(0.15, desc="서버 호출 중… (최대 2-3분 소요)")
+
         result = client.predict(
-            image_path=image_handle,
-            audio_path=audio_handle,
+            image_path=handle_file(image),
+            audio_path=handle_file(audio),
             guidance_scale=guidance_scale,
             steps=steps,
             api_name="/generate_animation"
         )
-        
-        # Progress 업데이트
-        progress(0.9, desc="결과 처리 중...")
-        
-        # 결과 로깅
-        log_msg = f"API 호출 성공!\n결과 타입: {type(result)}\n결과 길이: {len(result) if isinstance(result, (list, tuple)) else 'N/A'}"
-        logger.info(log_msg)
-        logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] {log_msg}")
-        
-        # 결과 상세 로깅
+
+        progress(0.95, desc="결과 정리")
         if isinstance(result, (list, tuple)) and len(result) >= 2:
-            for i, item in enumerate(result):
-                log_msg = f"결과[{i}]: {type(item)} - {str(item)[:100]}..."
-                logger.debug(log_msg)
-                logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] {log_msg}")
-            
-            animation_result = result[0]
-            comparison_result = result[1]
-            
-            # 비디오 경로 추출
-            animation_video = None
-            comparison_video = None
-            
-            if isinstance(animation_result, dict) and 'video' in animation_result:
-                animation_video = animation_result['video']
-                logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] 애니메이션 비디오 경로: {animation_video}")
-            
-            if isinstance(comparison_result, dict) and 'video' in comparison_result:
-                comparison_video = comparison_result['video']
-                logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] 비교 비디오 경로: {comparison_video}")
-            
-            logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] === 애니메이션 생성 완료! ===")
-            
-            return animation_video, comparison_video, "\n".join(logs)
+            anim, comp = result[0], result[1]
+            logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] 성공")
+            return anim, comp, "\n".join(logs)
         else:
-            error_msg = f"예상치 못한 결과 형식: {type(result)}"
-            logger.error(error_msg)
-            logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] ERROR: {error_msg}")
-            return None, None, "\n".join(logs)
-            
+            raise RuntimeError(f"예상치 못한 반환 형식: {type(result)}")
     except Exception as e:
-        error_msg = f"오류 발생: {str(e)}"
-        logger.error(error_msg)
-        logger.error(traceback.format_exc())
-        logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] ERROR: {error_msg}")
-        logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] 상세 오류:\n{traceback.format_exc()}")
+        logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] 오류: {e}")
         return None, None, "\n".join(logs)
 
-# Gradio 인터페이스 생성
-with gr.Blocks(title="Animation Generator API Test") as demo:
-    gr.Markdown("""
-    # 🎬 Animation Generator API Test Interface
-    
-    이 인터페이스는 `http://211.233.58.201:7788/` API를 테스트하기 위한 도구입니다.
-    
-    ## 사용 방법:
-    1. 포트레이트 이미지를 업로드하세요
-    2. 드라이빙 오디오 파일을 업로드하세요
-    3. Guidance Scale과 Inference Steps를 조정하세요
-    4. "Generate Animation" 버튼을 클릭하세요
-    """)
-    
-    # API 연결 상태 확인
-    with gr.Row():
-        with gr.Column():
-            connection_status = gr.Textbox(label="API 연결 상태", interactive=False)
-            check_connection_btn = gr.Button("API 연결 테스트", variant="secondary")
-    
+# ------------------------------------------------------------------ #
+with gr.Blocks(title="Animation Generator Client") as demo:
+    gr.Markdown("# 🎬 Animation Generator – Client UI")
+
+    # health check
+    status_box = gr.Textbox(label="API 상태", interactive=False)
+    gr.Button("서버 연결 테스트").click(
+        test_api_connection,
+        outputs=[status_box, status_box]
+    )
+
     gr.Markdown("---")
-    
-    # 입력 섹션
     with gr.Row():
         with gr.Column():
-            image_input = gr.Image(
-                label="Portrait Image (any aspect ratio)",
-                type="filepath",
-                elem_id="image_input"
-            )
-            audio_input = gr.Audio(
-                label="Driving Audio",
-                type="filepath",
-                elem_id="audio_input"
-            )
-        
-        with gr.Column():
-            guidance_scale = gr.Slider(
-                minimum=1,
-                maximum=10,
-                value=3,
-                step=0.1,
-                label="Guidance Scale",
-                info="Controls the strength of the guidance"
-            )
-            steps = gr.Slider(
-                minimum=1,
-                maximum=50,
-                value=10,
-                step=1,
-                label="Inference Steps",
-                info="Number of denoising steps"
-            )
-            
-            generate_btn = gr.Button("🚀 Generate Animation", variant="primary", size="lg")
-    
-    # 결과 섹션
-    gr.Markdown("## 📽️ Results")
-    with gr.Row():
-        with gr.Column():
-            animation_output = gr.Video(
-                label="Animation Result",
-                elem_id="animation_output"
+            img_in  = gr.Image(type="filepath", label="Portrait")
+            aud_in  = gr.Audio(type="filepath", label="Audio")
+            scale   = gr.Slider(1, 10, value=3.0, step=0.1, label="Guidance Scale")
+            steps   = gr.Slider(5, 30, value=10, step=1, label="Inference Steps")
+            gr.Button("🚀 Generate").click(
+                generate_animation,
+                inputs=[img_in, aud_in, scale, steps],
+                outputs=["animation_out", "comp_out", "log_out"]
             )
         with gr.Column():
-            comparison_output = gr.Video(
-                label="Side-by-Side Comparison",
-                elem_id="comparison_output"
-            )
-    
-    # 로그 섹션
-    with gr.Accordion("📋 실행 로그", open=True):
-        logs_output = gr.Textbox(
-            label="Logs",
-            lines=10,
-            max_lines=20,
-            interactive=False,
-            elem_id="logs"
-        )
-    
-    # 예제 섹션
-    gr.Markdown("## 🎯 Examples")
-    gr.Examples(
-        examples=[
-            ["example_portrait.jpg", "example_audio.wav", 3.0, 10],
-            ["example_portrait2.jpg", "example_audio2.wav", 2.5, 15],
-        ],
-        inputs=[image_input, audio_input, guidance_scale, steps],
-        outputs=[animation_output, comparison_output, logs_output],
-        fn=generate_animation,
-        cache_examples=False
-    )
-    
-    # 이벤트 핸들러
-    check_connection_btn.click(
-        fn=test_api_connection,
-        outputs=[connection_status, connection_status]
-    )
-    
-    generate_btn.click(
-        fn=generate_animation,
-        inputs=[image_input, audio_input, guidance_scale, steps],
-        outputs=[animation_output, comparison_output, logs_output]
-    )
-    
-    # 페이지 로드 시 API 연결 테스트
-    demo.load(
-        fn=test_api_connection,
-        outputs=[connection_status, connection_status]
-    )
+            gr.Video(label="Animation Result", elem_id="animation_out")
+            gr.Video(label="Side-by-side",    elem_id="comp_out")
+
+    with gr.Accordion("로그", open=False):
+        gr.Textbox(label="실행 로그", lines=12, max_lines=20, interactive=False, elem_id="log_out")
 
-# 앱 실행
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        debug=True
-    )
+    demo.queue(max_size=4).launch(server_name="0.0.0.0", server_port=7860, show_api=False)