Update app.py

app.py

@@ -1,6 +1,6 @@
 """
 Gradio UI that calls the remote animation server.
-환경변수 ANIM_API_URL 로 서버 주소를 설정하세요
+ANIM_API_URL 환경변수를 설정해 서버 주소를 주입하세요.
    ANIM_API_URL=http://211.233.58.201:7788/
 """
 
@@ -11,39 +11,32 @@ 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__)
 
-# ───────────────────── 서버 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)
+# ──────────────── 서버 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)
 
-# 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나 도메인으로 설정해야 합니다."
+        "HF Space 컨테이너 안에서는 공인 IP나 도메인을 ANIM_API_URL 로 지정해야 합니다."
     )
 
-# ──────────────────────── 헬퍼 함수 ───────────────────────── #
-def make_client() -> Client:
-    """
-    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)
+# ──────────────── 헬스체크 함수 ─────────────────────── #
+TIMEOUT = httpx.Timeout(connect=30.0, read=120.0, write=120.0, pool=30.0)
 
 def test_api_connection():
     now = datetime.now().strftime("%H:%M:%S")
     try:
-        client = make_client()
-        ready  = client.get("/healthz")["ready"]
-        msg    = f"[{now}] 서버 연결 성공 ✅ (ready={ready})"
+        resp  = httpx.get(f"{API_URL.rstrip('/')}/healthz", timeout=TIMEOUT)
+        ready = resp.json().get("ready", False)
+        msg   = f"[{now}] 서버 연결 성공 ✅ (ready={ready})"
         log.info(msg)
         return True, msg
     except Exception as e:
@@ -51,17 +44,19 @@ 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}] 요청 시작"]
+
     try:
         if image is None or audio is None:
             raise ValueError("이미지와 오디오를 모두 업로드하세요.")
 
         progress(0.05, desc="파일 준비")
-        client = make_client()
+        client = Client(API_URL)                # ← timeout 인자 삭제
 
-        progress(0.15, desc="서버 호출 중… (최대 2~3분)")
+        progress(0.15, desc="서버 호출 중… (수 분 소요 가능)")
         result = client.predict(
             image_path=handle_file(image),
             audio_path=handle_file(audio),
@@ -82,7 +77,7 @@ def generate_animation(image, audio, guidance_scale, steps, progress=gr.Progress
         logs.append(f"[{datetime.now().strftime('%H:%M:%S')}] 오류: {e}")
         return None, None, "\n".join(logs)
 
-# ────────────────────────── UI 정의 ───────────────────────── #
+# ─────────────────────── UI 정의 ────────────────────── #
 with gr.Blocks(title="Animation Generator Client") as demo:
     gr.Markdown("# 🎬 Animation Generator – Client UI")
 
@@ -107,14 +102,13 @@ with gr.Blocks(title="Animation Generator Client") as demo:
     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]
     )
 
-# ────────────────────────── 실행 ─────────────────────────── #
+# ─────────────────────── 실행 ───────────────────────── #
 if __name__ == "__main__":
     demo.queue(max_size=4).launch(
         server_name="0.0.0.0",