Update app.py

app.py

@@ -623,32 +623,36 @@ def generate_diagram_locally(json_data: str, diagram_type: str, output_format: s
 ##############################################################################
 # ────────────────── Prompt Templates (6 Styles) ──────────────────
 EXAMPLE_PROMPTS: dict[str, str] = {
-
     "Product Design": (
-        "A sleek industrial product-design sketch.\n"
-        "{nodes}"
+        "A sleek industrial product-design sketch showing the hierarchy below.\n"
+        "{tree}"
     ),
     "Mindmap": (
-        "A hand-drawn colorful mind-map, educational style, clear hierarchy.\n"
-        "{nodes}"
+        "A hand-drawn, colorful MIND-MAP of the following tree (central node on the left, radial branches).\n"
+        "{tree}"
     ),
     "Mockup": (
-        "A clean hand-drawn wire-frame for a mobile banking app.\n"
-        "{nodes}"
+        "A clean hand-drawn wire-frame mock-up – lay out the following screens/pages as boxes and arrows.\n"
+        "{tree}"
     ),
     "Infographic": (
-        "A flat corporate infographic – “Global Renewable Energy Trends 2025”.\n"
-        "{nodes}"
+        "A flat CORPORATE INFOGRAPHIC that arranges each item below as its own icon + label.\n"
+        "{tree}"
     ),
     "Diagram": (
-        "A hand-drawn business process diagram.\n"
-        "{nodes}"
+        "A hand-drawn BUSINESS-PROCESS DIAGRAM. Use swim-lanes/boxes connected by arrows following the tree below.\n"
+        "{tree}"
     ),
     "Flowchart": (
-        "A vibrant hand-drawn flow-chart.\n"
-        "{nodes}"
+        "A vibrant hand-drawn FLOW-CHART. Use decision diamonds / process boxes exactly in this order:\n"
+        "{tree}"
     ),
 }
+
+
+
+
+
 STYLE_KEYS = list(EXAMPLE_PROMPTS.keys())
 
 def pick_flux_style(slide_idx: int) -> str:
@@ -691,9 +695,9 @@ def generate_flux_prompt(title: str, content: str, style_key: str) -> str:
 
     # 3) stylistic tail (≤120 words total)
     tail = (
-        "Corporate palette, white background, hand-drawn line style, "
-        "clean composition, high-resolution vector."
-    )
+        "corporate colour palette, white background, hand-drawn line style; "
+        "boxes, arrows, connectors clearly visible; high-resolution vector."
+    )    
 
     return f"{prompt_body}\n\n{tail}".strip()