Update app.py

app.py

@@ -651,7 +651,12 @@ EXAMPLE_PROMPTS: dict[str, str] = {
 }
 STYLE_KEYS = list(EXAMPLE_PROMPTS.keys())
 
-
+def pick_flux_style(slide_idx: int) -> str:
+    """
+    슬라이드 번호(i)를 6가지 FLUX 스타일 키
+    ('Product Design' … 'Flowchart') 중 하나로 매핑한다.
+    """
+    return STYLE_KEYS[slide_idx % len(STYLE_KEYS)]
 
 # ------------------------------------------------------------------
 # 2) generate_flux_prompt 함수 전체  ★ 수정 완료
@@ -1908,39 +1913,57 @@ def create_advanced_ppt_from_content(
                                     pass
                     except Exception as e:
                         logger.error(f"Failed to add conclusion image: {e}")
-            
+
             elif visual_type[0] == 'diverse_images':
-                # 다양한 슬라이드 이미지 생성 (API별 제한 확인)
-                logger.info(f"Generating diverse images for slide {i+1} (3D: {image_count_3d}/{max_images_per_api}, FLUX: {image_count_flux}/{max_images_per_api})")
-                prompt_3d, prompt_photo = generate_diverse_prompt(slide_title, slide_content, i)
-                
+                # --- FLUX 다이어그램(6-style) & 3D 이미지 생성 ---
+                logger.info(
+                    f"Generating FLUX diagram-style image for slide {i+1} "
+                    f"(3D {image_count_3d}/{max_images_per_api}, "
+                    f"FLUX {image_count_flux}/{max_images_per_api})"
+                )
+        
+                # 3D-API용(그림·일러스트) 프롬프트는 그대로
+                prompt_3d, _ = generate_diverse_prompt(slide_title, slide_content, i)
+
+    # ⚡ FLUX-API용 프롬프트: 6가지 다이어그램 구도 중 하나 선택
+                style_key   = pick_flux_style(i)   # ← ① 새 함수 사용
+                prompt_flux = generate_flux_prompt(slide_title, slide_content, style_key)
+
                 selected_image = None
-                
-                # API별 제한에 따라 이미지 생성
-                if i % 2 == 0 and image_count_3d < max_images_per_api:
-                    image_3d = generate_ai_image_via_3d_api(prompt_3d)
-                    if image_3d:
-                        selected_image = image_3d
+
+    # 3D-API 먼저 시도 (짝수 슬라이드 우선이라는 기존 정책 유지)
+                if (i % 2 == 0) and (image_count_3d < max_images_per_api):
+                    img_3d = generate_ai_image_via_3d_api(prompt_3d)
+                    if img_3d:
+                        selected_image = img_3d
                         image_count_3d += 1
-                elif image_count_flux < max_images_per_api:
-                    image_photo = generate_flux_image_via_api(prompt_photo)
-                    if image_photo:
-                        selected_image = image_photo
+
+    # FLUX-API (다이어그램 스타일) 시도
+                if (selected_image is None) and (image_count_flux < max_images_per_api):
+                    img_flux = generate_flux_image_via_api(prompt_flux)
+                    if img_flux:
+                        selected_image = img_flux
                         image_count_flux += 1
-                
+
+    # 슬라이드에 삽입
                 if selected_image and os.path.exists(selected_image):
                     try:
-                        pic = slide.shapes.add_picture(
+                        slide.shapes.add_picture(
                             selected_image,
                             Inches(5.2), Inches(1.5),
                             width=Inches(4.3), height=Inches(3.0)
                         )
                         visual_added = True
-                        
-                        if selected_image:
-                            os.unlink(selected_image)
                     except Exception as e:
                         logger.error(f"Failed to add slide image: {e}")
+                    finally:
+            # 임시 파일 정리
+                        try:
+                            os.unlink(selected_image)
+                        except:
+                            pass
+
+
             
             # 시각적 요소가 추가되지 않은 경우 플레이스홀더 추가
             if not visual_added: