Update app.py

app.py

@@ -1,48 +1,49 @@
 """
 Gradio UI that calls the remote animation server.
-Set ANIM_API_URL env-var when deploying:
+환경변수 ANIM_API_URL 로 서버 주소를 설정하세요
    ANIM_API_URL=http://211.233.58.201:7788/
 """
 
-import os, time, logging
+import os, logging
 from datetime import datetime
 
 import gradio as gr
 import httpx
 from gradio_client import Client, handle_file
 
+# ────────────────────────── 로깅 ──────────────────────────── #
 logging.basicConfig(
     level=logging.INFO,
     format="%(asctime)s [%(levelname)s] %(message)s"
 )
 log = logging.getLogger(__name__)
 
-# ------------------------------------------------------------------ #
-# 기본 접속 주소 – 내부(Loopback) 여부에 따라 자동 판단 + 환경변수 재정의
-# ------------------------------------------------------------------ #
+# ───────────────────── 서버 URL 결정 로직 ───────────────────── #
 _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)
+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을 지정해야 합니다.")
+# Hugging Face Space 컨테이너 안에서 127.0.0.1 사용하면 외부 접속 불가
+if "127.0.0.1" in API_URL and os.getenv("HF_SPACE") == "true":
+    raise RuntimeError(
+        "HF Space 내부에서는 ANIM_API_URL 을 공인 IP나 도메인으로 설정해야 합니다."
+    )
 
-# ------------------------------------------------------------------ #
+# ──────────────────────── 헬퍼 함수 ───────────────────────── #
 def make_client() -> Client:
-    """커넥션 타임아웃을 30s로 늘려 재시도 가능성 확보"""
-    timeout = httpx.Timeout(connect=30.0, read=120.0)
+    """
+    httpx.Timeout 은 0.27+ 에서 4개 또는 default 모두 지정해야 함
+    connect 30s • read/write 120s • pool 30s
+    """
+    timeout = httpx.Timeout(connect=30.0, read=120.0, write=120.0, pool=30.0)
     return Client(API_URL, timeout=timeout, retries=3)
 
 def test_api_connection():
     now = datetime.now().strftime("%H:%M:%S")
     try:
         client = make_client()
-        # readiness-probe
-        ready = client.get('/healthz')['ready']
-        msg = f"[{now}] 서버 연결 성공 ✅ (ready={ready})"
+        ready  = client.get("/healthz")["ready"]
+        msg    = f"[{now}] 서버 연결 성공 ✅ (ready={ready})"
         log.info(msg)
         return True, msg
     except Exception as e:
@@ -50,18 +51,17 @@ def test_api_connection():
         log.error(msg)
         return False, msg
 
-# ------------------------------------------------------------------ #
 def generate_animation(image, audio, guidance_scale, steps, progress=gr.Progress()):
     start = datetime.now().strftime("%H:%M:%S")
-    logs = [f"[{start}] 요청 시작"]
+    logs  = [f"[{start}] 요청 시작"]
     try:
         if image is None or audio is None:
             raise ValueError("이미지와 오디오를 모두 업로드하세요.")
 
         progress(0.05, desc="파일 준비")
         client = make_client()
-        progress(0.15, desc="서버 호출 중… (최대 2-3분 소요)")
 
+        progress(0.15, desc="서버 호출 중… (최대 2~3분)")
         result = client.predict(
             image_path=handle_file(image),
             audio_path=handle_file(audio),
@@ -77,41 +77,47 @@ def generate_animation(image, audio, guidance_scale, steps, progress=gr.Progress
             return anim, comp, "\n".join(logs)
         else:
             raise RuntimeError(f"예상치 못한 반환 형식: {type(result)}")
+
     except Exception as e:
         logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] 오류: {e}")
         return None, None, "\n".join(logs)
 
-# ------------------------------------------------------------------ #
+# ────────────────────────── UI 정의 ───────────────────────── #
 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)
     test_btn   = gr.Button("서버 연결 테스트")
     test_btn.click(test_api_connection, outputs=[status_box, status_box])
 
     gr.Markdown("---")
+
     with gr.Row():
         with gr.Column():
-            img_in  = gr.Image(type="filepath", label="Portrait")
-            aud_in  = gr.Audio(type="filepath", label="Audio")
+            img_in  = gr.Image(type="filepath", label="Portrait Image")
+            aud_in  = gr.Audio(type="filepath", label="Driving 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")
-            gen_btn = gr.Button("🚀 Generate")          # 버튼을 변수로
+            gen_btn = gr.Button("🚀 Generate")
         with gr.Column():
             anim_out = gr.Video(label="Animation Result")
-            comp_out = gr.Video(label="Side-by-side")
-    with gr.Accordion("로그", open=False):
-        log_out = gr.Textbox(label="실행 로그", lines=12, max_lines=20, interactive=False)
+            comp_out = gr.Video(label="Side-by-Side")
 
-    # ⭐ click 연결 ­– outputs에 **컴포넌트 변수** 사용
+    with gr.Accordion("실행 로그", open=False):
+        log_out = gr.Textbox(label="Logs", lines=12, max_lines=20, interactive=False)
+
+    # 버튼 이벤트 연결
     gen_btn.click(
         generate_animation,
         inputs=[img_in, aud_in, scale, steps],
-        outputs=[anim_out, comp_out, log_out]          # ← 문자열 X
+        outputs=[anim_out, comp_out, log_out]
     )
 
+# ────────────────────────── 실행 ─────────────────────────── #
 if __name__ == "__main__":
-    demo.queue(max_size=4).launch(server_name="0.0.0.0",
-                                  server_port=7860,
-                                  show_api=False)
+    demo.queue(max_size=4).launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_api=False
+    )