Update app.py

app.py

@@ -114,6 +114,7 @@ def initialize_ai_image_api():
 ##############################################################################
 # AI Image Generation API Configuration - FLUX API
 ##############################################################################
+
 FLUX_API_URL = "http://211.233.58.201:7896"
 FLUX_API_ENABLED = False
 flux_api_client = None
@@ -616,70 +617,104 @@ def generate_diagram_locally(json_data: str, diagram_type: str, output_format: s
         logger.error(f"Failed to generate diagram locally: {e}")
         return None
 
+
 ##############################################################################
-# FLUX Image Generation Functions
+# FLUX Image Prompt Generator – supports 6 diagram styles
 ##############################################################################
-def generate_flux_prompt(title: str, content: str) -> str:
-    """슬라이드 내용을 기반으로 FLUX 이미지 프롬프트 생성"""
-
-# ─────────────── AFTER (강화 버전) ───────────────
-    system_prompt = """You are a senior information-visualisation designer crafting prompts for the FLUX model.  
-    Your task: turn ONE PowerPoint slide (title + exactly five bullet points) into a **diagram-style graphic** that mirrors the slide’s logic, hierarchy and wording.
-
-Strict rules  
-1. Auto-select the ideal diagram type:  
-   • sequential verbs / stages → Flowchart / Timeline  
-   • branching themes → Mind-map / Radial  
-   • numeric comparisons → Flat Infographic with bars, pies or lines  
-   • UI / feature lists → Wire-frame or Product Sketch  
-2. Map **every bullet point** to a labelled node/segment; keep their original order.  
-3. Use explicit connectors: arrows “→”, tree branches “├──”, swim-lane boxes, or chart axes as appropriate.  
-4. Embed short node labels **verbatim** from bullets (≤ 4 words each).  
-5. If a bullet contains numbers, reflect them in the graphic (e.g. “45 %” in a pie slice).  
-6. Optional icon cues in square brackets, e.g. [Gear Icon], only when they aid comprehension.  
-7. Finish with a succinct style cue: colour palette (e.g. “cool blues & greys”), line style (“hand-drawn sketch” or “sleek vector”), whitespace preference.  
-8. ≤ 120 words, English, no markdown, no commentary – **output the prompt text only**.
-
-Mini-examples  
- • “Sleek industrial **product design** sketch: coffee-machine, curved metal body, touchscreen, matte-black, hand-drawn concept lines.”
- • “Hand-drawn colourful **mind-map**: KNOWLEDGE centre → ACQUISITION (Reading, Practice, Observation) … pastel palette.”
- • “Clean hand-drawn **wire-frame mock-up** for mobile banking app: title, login, dashboard (balance|transactions|quick-actions), bottom nav … monochrome sketch lines.”
- • “Flat corporate **infographic** ‘Global Renewable Energy Trends 2025’: stacked bar Solar|Wind|Hydro across 5 regions, pie Solar 45 % … teal-orange palette.”
- • “Detailed hand-drawn **business process diagram**: Market Analysis → Strategy → Design → Implementation → Review, directional arrows, icons (magnifying-glass, lightbulb, gear) … vibrant educational style.”
- • “Vibrant hand-drawn **flowchart** BUSINESS WORKFLOW: START → Collect Requirements → Analyze Data → Implementation (Frontend | Backend) → Test & Integration → DEPLOY → END … minimal icons, colour-coded nodes.”"""
+# ────────────────── Prompt Templates (6 Styles) ──────────────────
+EXAMPLE_PROMPTS: dict[str, str] = {
+    "Product Design": """A sleek industrial design concept for a coffee machine:
+- Curved metallic body with minimal bezel
+- Touchscreen panel for settings
+- Modern matte black finish
+- Hand-drawn concept sketch style""",
+    "Mindmap": """A handrawn colorful mind map diagram, educational style, vibrant colors, clear hierarchy, golden ratio layout.
+KNOWLEDGE
+├── ACQUISITION [Brain with Lightning ~60px]
+│   ├── READING [Open Book with Glow]
+│   ├── PRACTICE [Hands-on Tools]
+│   └── OBSERVATION [Eye with Magnifier]
+├── PROCESSING [Gear Network ~50px]
+│   ├── ANALYSIS [Graph Trending Up]
+│   └── SYNTHESIS [Puzzle Pieces]
+├── RETENTION [Memory Chip ~45px]
+│   ├── SHORT-TERM [Quick Flash]
+│   └── LONG-TERM [Solid Archive]
+└── APPLICATION
+    ├── CREATION [Artist Palette]
+    └── INNOVATION [Lightbulb Constellation]""",
+    "Mockup": """A clean hand-drawn style wireframe for a mobile banking app:
+- Title screen with logo
+- Login screen (username, password, login button)
+- Dashboard with 3 main sections (balance, transactions, quick actions)
+- Bottom navigation bar (home, transfers, profile)""",
+    "Infographic": """A sophisticated flat-style infographic for a multinational corporation's annual report:
+- Title: "Global Renewable Energy Trends 2025"
+- Subtitle: "Market Share and Growth Analysis"
+- Visual Elements:
+    - Multi-segmented bar charts comparing Solar, Wind, and Hydro energy production across regions
+    - Pie chart displaying overall energy distribution: Solar (45%), Wind (30%), Hydro (25%)
+    - Trend lines indicating year-over-year growth
+- Icons: Sleek, minimalist representations of a sun, wind turbine, and water droplet
+- Layout: Clean, grid-based design with ample white space and pastel accents for a modern corporate look
+- Annotations: Brief, impactful data callouts highlighting key performance indicators and future forecasts""",
+    "Diagram": """A detailed hand-drawn diagram illustrating an end-to-end business workflow:
+- Title: "Integrated Business Process Diagram"
+- Components:
+    - Market Analysis → Strategy Development → Product Design → Implementation → Post-Launch Review
+- Visual Elements:
+    - Directional arrows, magnifying glass, lightbulb, gear, checklist icons
+- Style: Vibrant, educational yet professional
+- Layout: Clear hierarchy, color-coded sections""",
+    "Flowchart": """A hand-drawn style flowchart, vibrant colors, minimalistic icons.
+BUSINESS WORKFLOW
+├── START [Green Button ~40px]
+│   ├── COLLECT REQUIREMENTS [Folder Icon]
+│   └── ANALYZE DATA [Chart Icon]
+├── IMPLEMENTATION [Coding Symbol ~50px]
+│   ├── FRONTEND [Browser Icon]
+│   └── BACKEND [Server Icon]
+├── TEST & INTEGRATION [Gear Icon ~45px]
+└── DEPLOY → END [Checkered Flag ~40px]"""
+}
+STYLE_KEYS = list(EXAMPLE_PROMPTS.keys())
 
+def generate_flux_prompt(title: str, content: str, style_key: str) -> str:
+    """
+    Build a FLUX image-generation prompt for one slide, following six
+    pre-defined visual styles (Product Design, Mindmap, Mockup, Infographic,
+    Diagram, Flowchart).  `content` is the raw bullet-point block coming from
+    the slide; `style_key` picks which template to use.
+    """
+    # 1) Extract clean bullet points (max 8)
+    bullets = [
+        re.sub(r'^[\-\•\●]\s*', '', line).strip()
+        for line in content.splitlines()
+        if line.strip().startswith(('-', '•', '●'))
+    ][:8]
+
+    # Build node / list representation for templates that need it
+    if bullets:
+        node_block = '\n'.join([f"- {b}" for b in bullets])
+        tree_block = '\n'.join([f"├── {b}" if i < len(bullets) - 1 else f"└── {b}"
+                                for i, b in enumerate(bullets)])
+    else:
+        node_block = "- No explicit bullet points provided -"
+        tree_block = node_block
+
+    # 2) Select template
+    tpl = EXAMPLE_PROMPTS.get(style_key, EXAMPLE_PROMPTS["Diagram"])
+
+    # 3) Inject dynamic parts
+    prompt_body = tpl.format(title=title, nodes=node_block if "{nodes}" in tpl else tree_block)
+
+    # 4) Final stylistic tail – ensure English, ≤120 words
+    style_tail = "Corporate palette, white background, high-resolution vector, clean composition."
+
+    # 5) Return full prompt (no markdown / commentary)
+    full_prompt = f"{prompt_body}\n\n{style_tail}"
+    return full_prompt.strip()
 
-    
-    messages = [
-        {"role": "system", "content": system_prompt},
-        {"role": "user", "content": f"Create an image prompt for:\nTitle: {title}\nContent: {content[:500]}"}
-    ]
-    
-    headers = {
-        "Authorization": f"Bearer {FRIENDLI_TOKEN}",
-        "Content-Type": "application/json"
-    }
-    
-    payload = {
-        "model": FRIENDLI_MODEL_ID,
-        "messages": messages,
-        "max_tokens": 200,
-        "temperature": 0.8,
-        "stream": False
-    }
-    
-    try:
-        response = requests.post(FRIENDLI_API_URL, headers=headers, json=payload, timeout=30)
-        if response.status_code == 200:
-            response_data = response.json()
-            if 'choices' in response_data and len(response_data['choices']) > 0:
-                prompt = response_data['choices'][0]['message']['content'].strip()
-                return f"Professional business presentation slide illustration: {prompt}, modern clean style, corporate colors, white background"
-    except Exception as e:
-        logger.error(f"Error generating FLUX prompt: {e}")
-    
-    # Fallback prompt
-    return f"Professional business presentation illustration about {title}, modern minimalist style, clean design, corporate colors"
 
 def generate_flux_image_via_api(prompt: str) -> Optional[str]:
     """FLUX API를 통해 이미지 생성"""