Update app.py

app.py

@@ -1,31 +1,44 @@
 import gradio as gr
 import pandas as pd
 
-# Salary Distributor Function
-def salary_distributor(gross, categories_input):
+# Salary Distributor Function with Tree Structure Output
+def salary_distributor_tree(gross, categories_input):
     try:
+        lines = categories_input.strip().split("\n")
         categories = []
-        for line in categories_input.strip().split("\n"):
-            name, mode, value = line.split(",")
+
+        for line in lines:
+            parts = line.split(",")
+            if len(parts) != 3:
+                continue
+            name, mode, value = parts
             value = float(value)
             amount = (gross * value / 100) if mode.strip() == "%" else value
             categories.append({
-                "Category": name.strip(),
-                "Mode": mode.strip(),
-                "Value": value,
-                "Amount": round(amount, 2)
+                "name": name.strip(),
+                "mode": mode.strip(),
+                "value": value,
+                "amount": round(amount, 2)
             })
 
-        df = pd.DataFrame(categories)
-        total_allocated = round(df["Amount"].sum(), 2)
+        total_allocated = sum(c["amount"] for c in categories)
         remaining = round(gross - total_allocated, 2)
 
-        return df, total_allocated, remaining
+        # Create a tree-like representation
+        tree_lines = [f"Gross Salary: {gross}"]
+        for cat in categories:
+            tree_lines.append(f" ├─ {cat['name']} ({cat['mode']} {cat['value']}): {cat['amount']}")
+        tree_lines.append(f" └─ Remaining: {remaining}")
+
+        tree_str = "\n".join(tree_lines)
+
+        df = pd.DataFrame(categories)
+        return tree_str, df, round(total_allocated, 2), remaining
     except Exception as e:
-        return pd.DataFrame(), 0, f"Error: {e}"
+        return f"Error: {e}", pd.DataFrame(), 0, 0
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Salary Distributor Calculator")
+    gr.Markdown("## Salary Distributor Calculator (Tree View)")
 
     with gr.Row():
         gross_salary = gr.Number(label="Gross Salary", value=1000)
@@ -35,11 +48,12 @@ with gr.Blocks() as demo:
             lines=6
         )
 
+    output_tree = gr.Textbox(label="Salary Distribution Tree", lines=10)
     output_table = gr.Dataframe(headers=["Category", "Mode", "Value", "Amount"], datatype=["str", "str", "number", "number"])
     total_output = gr.Number(label="Total Allocated")
     remaining_output = gr.Number(label="Remaining")
 
     run_btn = gr.Button("Calculate")
-    run_btn.click(fn=salary_distributor, inputs=[gross_salary, categories_input], outputs=[output_table, total_output, remaining_output])
+    run_btn.click(fn=salary_distributor_tree, inputs=[gross_salary, categories_input], outputs=[output_tree, output_table, total_output, remaining_output])
 
 demo.launch()