Update app.py

app.py

@@ -44,6 +44,23 @@ example_prompts = {
 - 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:
@@ -141,7 +158,7 @@ with gr.Blocks(css=css, title="Workflow Canvas") as demo:
         <div style="text-align:center;">
             <h1>Workflow Canvas</h1>
             <p>Generate design concepts and workflow diagrams for your business needs using our multi-tab interface.</p>
-            <p><strong>AI Spaces:</strong> Product Design, Mockup, Infographic, Diagram, Flowchart</p>
+            <p><strong>AI Spaces:</strong> Product Design, Mindmap, Mockup, Infographic, Diagram, Flowchart</p>
         </div>
         """
     )
@@ -157,8 +174,8 @@ with gr.Blocks(css=css, title="Workflow Canvas") as demo:
             steps_slider = gr.Slider(label="Inference Steps", minimum=1, maximum=50, step=1, value=20)
             guidance_slider = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=15.0, step=0.5, value=7.5)
         
-        # Main area: Tabbed interface for different design types in specified order:
-        # Order: Product Design, Mockup, Infographic, Diagram, Flowchart
+        # Main area: Tabbed interface with updated tab order:
+        # 1: Product Design, 2: Mindmap, 3: Mockup, 4: Infographic, 5: Diagram, 6: Flowchart
         with gr.Column(scale=8):
             with gr.Tabs():
                 # Tab 1: Product Design
@@ -175,7 +192,21 @@ with gr.Blocks(css=css, title="Workflow Canvas") as demo:
                         for ex in example_prompts["Product Design"]:
                             gr.Button(ex, variant="secondary").click(fn=lambda ex=ex: set_prompt(ex), outputs=pd_prompt)
                 
-                # Tab 2: Mockup
+                # Tab 2: Mindmap
+                with gr.Tab("Mindmap"):
+                    mindmap_prompt = gr.Textbox(
+                        label="Mindmap Prompt",
+                        placeholder="Enter a mind map description...",
+                        lines=5,
+                        value=example_prompts["Mindmap"][0]
+                    )
+                    mindmap_generate = gr.Button("Generate Mindmap")
+                    mindmap_image = gr.Image(label="Generated Mindmap", value="w6.webp")
+                    with gr.Accordion("Example Prompts", open=True, elem_classes="example-accordion"):
+                        for ex in example_prompts["Mindmap"]:
+                            gr.Button(ex, variant="secondary").click(fn=lambda ex=ex: set_prompt(ex), outputs=mindmap_prompt)
+                
+                # Tab 3: Mockup
                 with gr.Tab("Mockup"):
                     mock_prompt = gr.Textbox(
                         label="Mockup Prompt",
@@ -189,7 +220,7 @@ with gr.Blocks(css=css, title="Workflow Canvas") as demo:
                         for ex in example_prompts["Mockup"]:
                             gr.Button(ex, variant="secondary").click(fn=lambda ex=ex: set_prompt(ex), outputs=mock_prompt)
                 
-                # Tab 3: Infographic
+                # Tab 4: Infographic
                 with gr.Tab("Infographic"):
                     info_prompt = gr.Textbox(
                         label="Infographic Prompt",
@@ -203,7 +234,7 @@ with gr.Blocks(css=css, title="Workflow Canvas") as demo:
                         for ex in example_prompts["Infographic"]:
                             gr.Button(ex, variant="secondary").click(fn=lambda ex=ex: set_prompt(ex), outputs=info_prompt)
                 
-                # Tab 4: Diagram
+                # Tab 5: Diagram
                 with gr.Tab("Diagram"):
                     diag_prompt = gr.Textbox(
                         label="Diagram Prompt",
@@ -217,7 +248,7 @@ with gr.Blocks(css=css, title="Workflow Canvas") as demo:
                         for ex in example_prompts["Diagram"]:
                             gr.Button(ex, variant="secondary").click(fn=lambda ex=ex: set_prompt(ex), outputs=diag_prompt)
                 
-                # Tab 5: Flowchart
+                # Tab 6: Flowchart
                 with gr.Tab("Flowchart"):
                     flow_prompt = gr.Textbox(
                         label="Flowchart Prompt",
@@ -238,6 +269,12 @@ with gr.Blocks(css=css, title="Workflow Canvas") as demo:
         outputs=[pd_image, seed_slider]
     )
     
+    mindmap_generate.click(
+        fn=generate_image,
+        inputs=[mindmap_prompt, seed_slider, randomize_seed, width_slider, height_slider, steps_slider, guidance_slider],
+        outputs=[mindmap_image, seed_slider]
+    )
+    
     mock_generate.click(
         fn=generate_image,
         inputs=[mock_prompt, seed_slider, randomize_seed, width_slider, height_slider, steps_slider, guidance_slider],