Update app.py

app.py

@@ -1165,9 +1165,12 @@ def generate_image(prompt: str):
         if len(clean_prompt) < 3:
             return None, None
 
-        # 여기서 API 호출 시도
+        # Client 객체 생성 시 timeout 설정 (선택사항)
+        client = Client(IMAGE_API_URL, timeout=30)  # 여기서 timeout 설정
+        
         try:
-            res = Client(IMAGE_API_URL).predict(
+            # predict 함수에서는 timeout 인자 제거
+            res = client.predict(
                 prompt      = clean_prompt,
                 width       = 768,
                 height      = 768,
@@ -1175,30 +1178,15 @@ def generate_image(prompt: str):
                 inference_steps = 30,
                 seed        = 3,
                 do_img2img  = False,
-                api_name    = "/generate_image",
-                timeout     = 30
+                api_name    = "/generate_image"
+                # timeout 매개변수 제거함
             )
         except Exception as api_err:
             st.error(f"이미지 API 연결 실패: {api_err}")
             logging.error(f"이미지 API 연결 오류: {api_err}")
             return None, None
 
-        if not res:
-            logging.warning("이미지 API가 빈 응답을 반환함")
-            return None, None
-            
-        raw = res[0] if isinstance(res, list) else res
-        
-        # 이미지 데이터 처리 (base64 인 경우)
-        if isinstance(raw, str) and raw.startswith("data:image"):
-            raw = raw.split(",", 1)[1]
-            try:
-                import base64
-                img_bytes = base64.b64decode(raw)
-                return img_bytes, clean_prompt
-            except Exception as e:
-                logging.error(f"base64 디코딩 실패: {e}")
-                return None, None
+
         
         # URL인 경우
         elif isinstance(raw, str) and raw.startswith("http"):