Update app.py

app.py

@@ -1041,13 +1041,8 @@ def generate_avatar_animation(portrait_image, driving_audio, guidance_scale, inf
             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초
-                )
+                # 간단한 Client 생성 (timeout 매개변수 없이)
+                client = Client(AVATAR_API_URL)
                 
                 logging.info("API 클라이언트 생성 성공")
                 
@@ -1063,16 +1058,17 @@ def generate_avatar_animation(portrait_image, driving_audio, guidance_scale, inf
                 # 성공하면 루프 종료
                 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
+                logging.warning(f"API 호출 오류 (시도 {attempt + 1}/{max_retries}): {str(e)}")
+                if "Connection" in str(e) or "Timeout" in str(e):
+                    if attempt < max_retries - 1:
+                        logging.info(f"{retry_delay}초 후 재시도...")
+                        time.sleep(retry_delay)
+                    else:
+                        return None, None, "❌ API 서버에 연결할 수 없습니다. 잠시 후 다시 시도해주세요."
+                else:
+                    # 다른 예외는 즉시 처리
+                    raise e
         
         # 임시 파일 삭제 (있는 경우)
         if 'temp_portrait_path' in locals() and os.path.exists(temp_portrait_path):
@@ -1110,12 +1106,6 @@ 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