Update app.py

app.py

@@ -1158,63 +1158,38 @@ def process_uploaded_files(uploaded_files):
 
 # ───────────────────────────── Image Generation ──────────────────────
 def generate_image(prompt: str):
-    """이미지 생성 함수"""
     if not prompt:
         return None, None
-    
     try:
         clean_prompt = prompt.strip("\"'").strip()
         if len(clean_prompt) < 3:
             return None, None
-
-        # Gradio 클라이언트 생성 (timeout은 클라이언트 생성 시 설정)
-        client = Client(IMAGE_API_URL)
-        
-        try:
-            res = client.predict(
-                prompt = clean_prompt,
-                width = 768,
-                height = 768,
-                guidance = 3.5,
-                inference_steps = 30,
-                seed = 3,
-                do_img2img = False,
-                api_name = "/generate_image"
-            )
-        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
-        
-        # 응답 타입에 따른 처리
-        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
-        elif isinstance(raw, str) and raw.startswith("http"):
-            return raw, clean_prompt
+        logging.info(f"Sending image generation request with prompt: {clean_prompt}")
+
+        res = Client(IMAGE_API_URL).predict(
+            prompt=clean_prompt,
+            width=768,
+            height=768,
+            guidance=3.5,
+            inference_steps=30,
+            seed=3,
+            do_img2img=False,
+            init_image=None,
+            image2image_strength=0.8,
+            resize_img=True,
+            api_name="/generate_image"
+        )
+        if res and len(res) >= 2 and res[0]:
+            logging.info("Successfully received image data")
+            return res[0], clean_prompt
         else:
-            # 바이너리 데이터이거나 다른 형식일 경우
-            return raw, clean_prompt
-            
+            logging.warning(f"Invalid response format from image API: {res}")
+            return None, None
     except Exception as e:
-        logging.error(f"이미지 생성 오류: {e}", exc_info=True)
+        logging.error(f"Image generation error: {str(e)}", exc_info=True)
         return None, None
 
 
-
 # ───────────────────────────── Kaggle API ─────────────────────────────
 @st.cache_resource
 def check_kaggle_availability():