Update app.py

app.py

@@ -31,6 +31,45 @@ def generate_diagram(prompt, width=1024, height=1024):
     except Exception as e:
         raise gr.Error(f"Error generating diagram: {str(e)}")
 
+# Convert example format for Gradio
+GRADIO_EXAMPLES = [
+    [
+        """A handrawn colorful mind map diagram, educational style, vibrant colors, clear hierarchy.
+        KNOWLEDGE
+        ├── ACQUISITION [Brain with Lightning ~60px]
+        │   ├── READING [Open Book with Glow]
+        │   └── PRACTICE [Hands-on Tools]
+        └── APPLICATION
+            ├── CREATION [Artist Palette]
+            └── INNOVATION [Lightbulb]""",
+        1024,
+        1024
+    ],
+    [
+        """A handrawn colorful mind map diagram, tech-focused style, neon accents.
+        DIGITAL TRANSFORM
+        ├── CLOUD [Cloud with Data ~55px]
+        │   ├── STORAGE [Database]
+        │   └── COMPUTING [Server]
+        └── SECURITY
+            ├── ENCRYPTION [Lock]
+            └── MONITORING [Radar]""",
+        1024,
+        1024
+    ],
+    [
+        """A handrawn colorful mind map diagram, creative style, flowing design.
+        INNOVATION
+        ├── IDEATION [Lightbulb ~60px]
+        │   ├── RESEARCH [Magnifier]
+        │   └── BRAINSTORM [Brain]
+        └── EXECUTION
+            ├── PROTOTYPE [Tools]
+            └── TEST [Checklist]""",
+        1024,
+        1024
+    ]
+]
 
 
 # Enhanced examples with more detailed prompts and specific styling
@@ -236,6 +275,7 @@ GRADIO_EXAMPLES = [
     for example in EXAMPLES
 ]
 
+# Create Gradio interface
 demo = gr.Interface(
     fn=generate_diagram,
     inputs=[
@@ -259,7 +299,7 @@ demo = gr.Interface(
             value=1024
         )
     ],
-    outputs=gr.Image(type="filepath", label="Generated Diagram"),
+    outputs=gr.Image(label="Generated Diagram"),
     title="🎨 FLUX Diagram Generator",
     description="Generate beautiful hand-drawn style diagrams using FLUX AI",
     article="""
@@ -268,33 +308,18 @@ demo = gr.Interface(
     - Include icon descriptions in brackets
     - Keep text concise and meaningful
     - Use consistent formatting
-    
-    ### Template Structure
-    ```
-    MAIN TOPIC
-    ├── SUBTOPIC 1 [Icon]
-    ├── SUBTOPIC 2 [Icon]
-    └── SUBTOPIC 3
-        ├── DETAIL 1 [Icon]
-        └── DETAIL 2 [Icon]
-    ```
     """,
     examples=GRADIO_EXAMPLES,
-    cache_examples=True,
-    theme=gr.themes.Soft()
+    cache_examples=True
 )
 
 # Launch app
 if __name__ == "__main__":
-    try:
-        demo.queue(max_size=10)
-        demo.launch(
-            server_name="0.0.0.0",
-            server_port=7860,
-            share=False,
-            show_error=True,
-            debug=True
-        )
-    except Exception as e:
-        print(f"Error launching app: {str(e)}")
-        raise
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        show_error=True,
+        debug=True,
+        ssr=False  # Disable SSR
+    )