Update app.py

app.py

@@ -1033,15 +1033,46 @@ def generate_avatar_animation(portrait_image, driving_audio, guidance_scale, inf
         if not os.path.exists(audio_path):
             return None, None, f"❌ 오디오 파일을 찾을 수 없습니다: {audio_path}"
         
-        # API 호출
-        client = Client(AVATAR_API_URL)
-        result = client.predict(
-            portrait_path,  # 직접 경로 전달
-            audio_path,     # 직접 경로 전달
-            float(guidance_scale),
-            float(inference_steps),
-            api_name="/generate_animation"
-        )
+        # API 연결 재시도 로직
+        max_retries = 3
+        retry_delay = 5  # 초
+        
+        for attempt in range(max_retries):
+            try:
+                logging.info(f"API 연결 시도 {attempt + 1}/{max_retries}")
+                
+                # 타임아웃을 늘린 Client 생성
+                # gradio_client의 타임아웃 설정을 위해 httpx client 설정
+                import httpx
+                client = Client(
+                    AVATAR_API_URL,
+                    timeout=httpx.Timeout(60.0, connect=30.0)  # 연결 타임아웃 30초, 전체 타임아웃 60초
+                )
+                
+                logging.info("API 클라이언트 생성 성공")
+                
+                # API 호출
+                result = client.predict(
+                    portrait_path,  # 직접 경로 전달
+                    audio_path,     # 직접 경로 전달
+                    float(guidance_scale),
+                    float(inference_steps),
+                    api_name="/generate_animation"
+                )
+                
+                # 성공하면 루프 종료
+                break
+                
+            except (httpx.ConnectTimeout, httpx.TimeoutException) as e:
+                logging.warning(f"연결 타임아웃 (시도 {attempt + 1}/{max_retries}): {str(e)}")
+                if attempt < max_retries - 1:
+                    logging.info(f"{retry_delay}초 후 재시도...")
+                    time.sleep(retry_delay)
+                else:
+                    return None, None, "❌ API 서버에 연결할 수 없습니다. 잠시 후 다시 시도해주세요."
+            except Exception as e:
+                # 다른 예외는 즉시 처리
+                raise e
         
         # 임시 파일 삭제 (있는 경우)
         if 'temp_portrait_path' in locals() and os.path.exists(temp_portrait_path):
@@ -1079,6 +1110,12 @@ def generate_avatar_animation(portrait_image, driving_audio, guidance_scale, inf
             logging.error(f"Unexpected API response: {result}")
             return None, None, "❌ API 응답이 예상과 다릅니다."
             
+    except httpx.ConnectTimeout:
+        logging.error("Connection timeout to API server")
+        return None, None, "❌ API 서버 연결 시간 초과. 서버가 응답하지 않습니다."
+    except httpx.TimeoutException:
+        logging.error("Request timeout")
+        return None, None, "❌ 요청 시간 초과. 서버 처리 시간이 너무 깁니다."
     except Exception as e:
         logging.error(f"Avatar generation error: {str(e)}")
         import traceback