Update app.py

app.py

@@ -719,9 +719,23 @@ def _save_temp_image(img_bytes: bytes, suffix=".png") -> str:
 
 def generate_flux_image_via_api(prompt: str) -> Optional[str]:
     if not FLUX_API_ENABLED or not flux_api_client:
+        logger.error("FLUX API is not enabled or client not initialized")
         return None
 
     try:
+        logger.info(f"[FLUX] Calling API with prompt length: {len(prompt)}")
+        
+        # 빈 이미지 데이터 생성 (필수 파라미터를 위한 더미 데이터)
+        dummy_image = {
+            "path": None,
+            "url": None,
+            "size": None,
+            "orig_name": None,
+            "mime_type": None,
+            "is_stream": False,
+            "meta": {}
+        }
+        
         result = flux_api_client.predict(
             prompt=prompt,
             width=768,
@@ -730,42 +744,61 @@ def generate_flux_image_via_api(prompt: str) -> Optional[str]:
             inference_steps=8,
             seed=random.randint(1, 1_000_000),
             do_img2img=False,
-            api_name="/generate_image",   # ← 엔드포인트 확인 필수!
+            init_image=dummy_image,  # ← 필수 파라미터 추가
+            image2image_strength=0.8,  # 기본값
+            resize_img=True,  # 기본값
+            api_name="/generate_image",
         )
         
-        logger.info(f"[FLUX] raw result type={type(result)} value={str(result)[:200]}")
-
-        # 1) tuple 형식 (기존)
-        if isinstance(result, tuple) and result:
-            candidate = result[0]
-
-        # 2) 문자열 URL
-        elif isinstance(result, str) and result.startswith(("http://", "https://")):
-            candidate = _save_temp_image(requests.get(result).content)
+        logger.info(f"[FLUX] API response type: {type(result)}")
+        logger.info(f"[FLUX] API response (first 200 chars): {str(result)[:200]}")
 
-        # 3) base64 문자열
-        elif isinstance(result, str) and len(result) > 100:
-            try:
-                candidate = _save_temp_image(base64.b64decode(result.split(",")[-1]))
-            except Exception:
+        # 결과가 튜플인 경우 (generated_image, used_seed)
+        if isinstance(result, tuple) and len(result) >= 1:
+            generated_image = result[0]
+            
+            # generated_image가 dict 형태인 경우
+            if isinstance(generated_image, dict) and 'path' in generated_image:
+                candidate = generated_image['path']
+                logger.info(f"[FLUX] Extracted path from result dict: {candidate}")
+            else:
+                candidate = generated_image
+                logger.info(f"[FLUX] Using first element of tuple: {candidate}")
+        
+        # 결과가 dict인 경우
+        elif isinstance(result, dict):
+            if 'path' in result:
+                candidate = result['path']
+            elif 'url' in result:
+                candidate = _save_temp_image(requests.get(result['url']).content)
+            else:
                 candidate = None
-
-        # 4) BytesIO
-        elif hasattr(result, "read"):
-            candidate = _save_temp_image(result.read())
-
+        
+        # 문자열인 경우
+        elif isinstance(result, str):
+            if os.path.exists(result):
+                candidate = result
+            elif result.startswith(("http://", "https://")):
+                candidate = _save_temp_image(requests.get(result).content)
+            else:
+                candidate = None
+        
         else:
             candidate = None
+            logger.error(f"[FLUX] Unexpected result type: {type(result)}")
 
         # 최종 검증
         if candidate and os.path.exists(candidate):
+            logger.info(f"[FLUX] ✅ Success! Image saved at: {candidate}")
             return candidate
         else:
-            logger.error("FLUX 이미지 반환값을 처리하지 못했습니다")
+            logger.error(f"[FLUX] ❌ Failed to get valid image path. Candidate: {candidate}")
             return None
 
     except Exception as e:
-        logger.error(f"FLUX 이미지 생성 실패: {e}")
+        logger.error(f"[FLUX] ❌ Image generation failed: {str(e)}")
+        import traceback
+        logger.error(f"[FLUX] Traceback: {traceback.format_exc()}")
         return None
 
 ##############################################################################