Update app.py

app.py

@@ -25,6 +25,8 @@ import matplotlib as mpl
 import matplotlib.font_manager as fm
 import subprocess
 
+import base64, io, requests
+
 NANUM_PATH = "/usr/share/fonts/truetype/nanum/NanumGothic.ttf"  # 시스템 설치 경로
 
 # matplotlib용 글꼴 등록
@@ -703,42 +705,59 @@ def generate_flux_prompt(title: str, content: str, style_key: str) -> str:
 
 
 
+def _save_temp_image(img_bytes: bytes, suffix=".png") -> str:
+    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
+    tmp.write(img_bytes)
+    tmp.close()
+    return tmp.name
+
 def generate_flux_image_via_api(prompt: str) -> Optional[str]:
-    """FLUX API를 통해 이미지 생성"""
     if not FLUX_API_ENABLED or not flux_api_client:
         return None
-    
+
     try:
-        logger.info(f"Generating FLUX image with prompt: {prompt[:100]}...")
-        
         result = flux_api_client.predict(
             prompt=prompt,
             width=768,
             height=768,
             guidance=3.5,
             inference_steps=8,
-            seed=random.randint(1, 1000000),
+            seed=random.randint(1, 1_000_000),
             do_img2img=False,
-            init_image=None,
-            image2image_strength=0.8,
-            resize_img=True,
-            api_name="/generate_image"
+            api_name="/generate_image",   # ← 엔드포인트 확인 필수!
         )
-        
-        if isinstance(result, tuple) and len(result) > 0:
-            image_path = result[0]
-            if image_path and os.path.exists(image_path):
-                # PNG로 변환
-                with Image.open(image_path) as img:
-                    png_tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
-                    img.save(png_tmp.name, format="PNG")
-                    logger.info(f"FLUX image generated and saved to {png_tmp.name}")
-                    return png_tmp.name
-        
-        return None
-        
+
+        # 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)
+
+        # 3) base64 문자열
+        elif isinstance(result, str) and len(result) > 100:
+            try:
+                candidate = _save_temp_image(base64.b64decode(result.split(",")[-1]))
+            except Exception:
+                candidate = None
+
+        # 4) BytesIO
+        elif hasattr(result, "read"):
+            candidate = _save_temp_image(result.read())
+
+        else:
+            candidate = None
+
+        # 최종 검증
+        if candidate and os.path.exists(candidate):
+            return candidate
+        else:
+            logger.error("FLUX 이미지 반환값을 처리하지 못했습니다")
+            return None
+
     except Exception as e:
-        logger.error(f"Failed to generate FLUX image: {e}")
+        logger.error(f"FLUX 이미지 생성 실패: {e}")
         return None
 
 ##############################################################################
@@ -1951,8 +1970,9 @@ def create_advanced_ppt_from_content(
 
     # 슬라이드에 삽입
                 if selected_image and os.path.exists(selected_image):
+
                     try:
-                        slide.shapes.add_picture(
+                        pic = slide.shapes.add_picture(
                             selected_image,
                             Inches(5.2), Inches(1.5),
                             width=Inches(4.3), height=Inches(3.0)
@@ -1960,13 +1980,13 @@ def create_advanced_ppt_from_content(
                         visual_added = True
                     except Exception as e:
                         logger.error(f"Failed to add slide image: {e}")
-                    finally:
-            # 임시 파일 정리
+        
+# 그림을 실제로 넣었을 때만 삭제
+                    if visual_added and selected_image and os.path.exists(selected_image):
                         try:
                             os.unlink(selected_image)
-                        except:
-                            pass
-
+                        except Exception as e:
+                            logger.warning(f"Temp image delete failed: {e}")
 
             
             # 시각적 요소가 추가되지 않은 경우 플레이스홀더 추가