Update app.py

app.py

@@ -12,18 +12,38 @@ def render_mermaid(jinja_text, variables):
     except Exception as e:
         return f"**Template Error:** {str(e)}"
 
-# Add a new variable to the dictionary
+# Add variable
 def add_variable(key, value, current_vars):
     if not key:
         return current_vars, "**Variable name is required**"
-
+    if key in current_vars:
+        return current_vars, f"⚠️ Variable '{key}' already exists. Use update instead."
     try:
         parsed_value = json.loads(value)
     except json.JSONDecodeError:
         parsed_value = value
+    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**"
+    if key not in current_vars:
+        return current_vars, f"⚠️ Variable '{key}' does not exist. Use add instead."
+    try:
+        parsed_value = json.loads(value)
+    except json.JSONDecodeError:
+        parsed_value = value
     current_vars[key] = parsed_value
-    return current_vars, ""
+    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():
@@ -73,25 +93,30 @@ with gr.Blocks(css=custom_css) as demo:
                 elem_id="jinja-code-box"
             )
 
-            gr.Markdown("### ➕ Add Template Variables")
+            gr.Markdown("### ➕ Add / ♻️ Update / 🗑️ Delete Template Variables")
             with gr.Row():
-                var_key = gr.Textbox(label="🆔 Variable Name (key)", placeholder="e.g., user_city")
-                var_value = gr.Textbox(label="📦 Value (JSON)", placeholder='e.g., "London", 42, true')
-
-            with gr.Row():
-                add_btn = gr.Button("➕ Add Variable")
-                reset_btn = gr.Button("🔁 Reset Variables")
-
-            error_box = gr.Markdown("", visible=False)
-            rendered_vars = gr.JSON(label="📋 Current Variables")
-            variables_state = gr.State({})  # internal variable dictionary
+                with gr.Column(scale=2):
+                    var_key = gr.Textbox(label="🆔 Variable Name", placeholder="e.g., user_city")
+                    var_value = gr.Textbox(label="📦 Value (JSON)", placeholder='e.g., "London"')
+                    
+                    with gr.Row():
+                        add_btn = gr.Button("➕ Add Variable")
+                        update_btn = gr.Button("♻️ Update Variable")
+                    with gr.Row():
+                        delete_btn = gr.Button("🗑️ Delete Variable")
+                        reset_btn = gr.Button("🔁 Reset Variables")
+
+                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
 
         # RIGHT COLUMN
         with gr.Column(scale=5):
             gr.Markdown("### 📈 Live Mermaid Preview")
             rendered_md = gr.Markdown(elem_classes="markdown-preview")
 
-    # ADD VARIABLE WORKFLOW
+    # ADD WORKFLOW
     add_btn.click(
         fn=add_variable,
         inputs=[var_key, var_value, variables_state],
@@ -107,6 +132,38 @@ 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],
+        outputs=[variables_state, error_box],
+        show_progress=False
+    ).then(
+        fn=lambda x: x,
+        inputs=variables_state,
+        outputs=rendered_vars
+    ).then(
+        fn=render_mermaid,
+        inputs=[jinja_input, variables_state],
+        outputs=rendered_md
+    )
+
+    # DELETE WORKFLOW
+    delete_btn.click(
+        fn=delete_variable,
+        inputs=[var_key, variables_state],
+        outputs=[variables_state, error_box],
+        show_progress=False
+    ).then(
+        fn=lambda x: x,
+        inputs=variables_state,
+        outputs=rendered_vars
+    ).then(
+        fn=render_mermaid,
+        inputs=[jinja_input, variables_state],
+        outputs=rendered_md
+    )
+
     # RESET BUTTON WORKFLOW
     reset_btn.click(
         fn=reset_variables,
@@ -128,4 +185,4 @@ with gr.Blocks(css=custom_css) as demo:
     # On load
     demo.load(render_mermaid, inputs=[jinja_input, variables_state], outputs=rendered_md)
 
-demo.launch()
+demo.launch()