Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 from jinja2 import Template
 import json
 
-# Render the Mermaid chart with Jinja variables
 def render_mermaid(jinja_text, variables):
     try:
         template = Template(jinja_text)
@@ -12,7 +11,6 @@ def render_mermaid(jinja_text, variables):
     except Exception as e:
         return f"**Template Error:** {str(e)}"
 
-# Add variable
 def add_variable(key, value, current_vars):
     if not key:
         return current_vars, "**Variable name is required**"
@@ -25,7 +23,6 @@ def add_variable(key, value, current_vars):
     current_vars[key] = parsed_value
     return current_vars, f"✅ Variable '{key}' added"
 
-# Update variable
 def update_variable(key, value, current_vars):
     if not key:
         return current_vars, "**Variable name is required**"
@@ -38,24 +35,20 @@ def update_variable(key, value, current_vars):
     current_vars[key] = parsed_value
     return current_vars, f"♻️ Variable '{key}' updated"
 
-# Delete variable
 def delete_variable(key, current_vars):
     if key in current_vars:
         del current_vars[key]
         return current_vars, f"🗑️ Variable '{key}' deleted"
     return current_vars, f"⚠️ Variable '{key}' not found"
 
-# Reset all variables
 def reset_variables():
     return {}
 
-# Default Jinja + Mermaid template
 default_template = """
 flowchart TD
-    Start(["Start"])
-    Start --> Decision{"Is city known?"}
-    Decision -->|Yes| City["You are in {{ user_city }}"]
-    Decision -->|No| AskCity["Please choose a city"]
+    Start(["Start"]) --> A{"Is city known?"}
+    A -->|Yes| B["You are in {{ user_city }}"]
+    A -->|No| C["Please choose a city"]
 """
 
 custom_css = """
@@ -83,7 +76,6 @@ with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("## 🧠 Jinja2 → Mermaid Renderer (Dynamic Variables - Interactive)")
 
     with gr.Row():
-        # LEFT COLUMN
         with gr.Column(scale=7):
             jinja_input = gr.Code(
                 label="✍️ Jinja2 + Mermaid Template",
@@ -109,14 +101,12 @@ with gr.Blocks(css=custom_css) as demo:
                 with gr.Column(scale=2):
                     error_box = gr.Markdown("", visible=False)
                     rendered_vars = gr.JSON(label="📋 Current Variables")
-                    variables_state = gr.State({})  # internal variable dictionary
+                    variables_state = gr.State({})
 
-        # RIGHT COLUMN
         with gr.Column(scale=5):
             gr.Markdown("### 📈 Live Mermaid Preview")
             rendered_md = gr.Markdown(elem_classes="markdown-preview")
 
-    # ADD WORKFLOW
     add_btn.click(
         fn=add_variable,
         inputs=[var_key, var_value, variables_state],
@@ -132,7 +122,6 @@ with gr.Blocks(css=custom_css) as demo:
         outputs=rendered_md
     )
 
-    # UPDATE WORKFLOW
     update_btn.click(
         fn=update_variable,
         inputs=[var_key, var_value, variables_state],
@@ -148,7 +137,6 @@ with gr.Blocks(css=custom_css) as demo:
         outputs=rendered_md
     )
 
-    # DELETE WORKFLOW
     delete_btn.click(
         fn=delete_variable,
         inputs=[var_key, variables_state],
@@ -164,7 +152,6 @@ with gr.Blocks(css=custom_css) as demo:
         outputs=rendered_md
     )
 
-    # RESET BUTTON WORKFLOW
     reset_btn.click(
         fn=reset_variables,
         inputs=[],
@@ -179,10 +166,8 @@ with gr.Blocks(css=custom_css) as demo:
         outputs=rendered_md
     )
 
-    # On template change
     jinja_input.change(render_mermaid, inputs=[jinja_input, variables_state], outputs=rendered_md)
 
-    # On load
     demo.load(render_mermaid, inputs=[jinja_input, variables_state], outputs=rendered_md)
 
 demo.launch()