Update app.py

app.py

@@ -1,17 +1,11 @@
-import datetime
 import pandas as pd
 import gradio as gr
+import plotly.express as px
 
 
-def calculate_budget(
-    income,
-    currency,
-    needs_pct,
-    wants_pct,
-    savings_pct,
-):
+def calculate_budget(income, currency, needs_pct, wants_pct, savings_pct):
     if needs_pct + wants_pct + savings_pct != 100:
-        return "❌ Percentages must total 100%", None
+        return gr.Markdown("❌ **Percentages must total 100%**"), None, None
 
     symbols = {
         "USD": "$",
@@ -23,58 +17,79 @@ def calculate_budget(
     }
     sym = symbols.get(currency, "")
 
-    needs_amt = income * needs_pct / 100
-    wants_amt = income * wants_pct / 100
-    savings_amt = income * savings_pct / 100
+    data = {
+        "Needs": income * needs_pct / 100,
+        "Wants": income * wants_pct / 100,
+        "Savings": income * savings_pct / 100,
+    }
 
-    data = [
-        ("Needs", needs_pct, needs_amt),
-        ("Wants", wants_pct, wants_amt),
-        ("Savings", savings_pct, savings_amt),
-    ]
+    df = pd.DataFrame(
+        {
+            "Category": data.keys(),
+            "Amount": data.values(),
+        }
+    )
 
-    df = pd.DataFrame(data, columns=["Bucket", "Percent", "Amount"])
-    df["Amount"] = df["Amount"].apply(lambda x: f"{sym}{x:,.0f}")
+    fig = px.pie(
+        df,
+        names="Category",
+        values="Amount",
+        title="Budget Allocation",
+        hole=0.4,
+    )
 
-    csv = df.to_csv(index=False)
+    summary = f"""
+### 💸 Monthly Budget Summary
 
-    summary = (
-        f"### Monthly Budget Summary\n\n"
-        f"- **Income:** {sym}{income:,.0f}\n"
-        f"- **Needs:** {sym}{needs_amt:,.0f}\n"
-        f"- **Wants:** {sym}{wants_amt:,.0f}\n"
-        f"- **Savings:** {sym}{savings_amt:,.0f}\n"
-    )
+- **Income:** {sym}{income:,.0f}  
+- **Needs:** {sym}{data['Needs']:,.0f}  
+- **Wants:** {sym}{data['Wants']:,.0f}  
+- **Savings:** {sym}{data['Savings']:,.0f}  
+
+✅ Allocation looks good!
+"""
 
-    return summary, csv
+    csv = df.to_csv(index=False)
 
+    return gr.Markdown(summary), fig, csv
 
-with gr.Blocks(title="Monthly Budget Tracker") as demo:
-    gr.Markdown("## 💸 Monthly Budget Tracker")
-    gr.Markdown("Use the 50/30/20 rule or customize your allocation.")
+
+with gr.Blocks(theme=gr.themes.Soft(), title="Monthly Budget Tracker") as demo:
+    gr.Markdown(
+        """
+## 💸 Monthly Budget Tracker
+Plan your money using the **50 / 30 / 20 rule** or customize it to your life.
+"""
+    )
 
     with gr.Row():
-        income = gr.Number(value=5000, label="Monthly Income")
+        income = gr.Number(label="Monthly Income", value=5000)
         currency = gr.Dropdown(
             ["USD", "INR", "EUR", "GBP", "AUD", "CAD"],
             value="INR",
             label="Currency",
         )
 
+    gr.Markdown("### 🎯 Allocation (%)")
+
     with gr.Row():
-        needs = gr.Slider(0, 100, value=50, label="Needs %")
-        wants = gr.Slider(0, 100, value=30, label="Wants %")
-        savings = gr.Slider(0, 100, value=20, label="Savings %")
+        needs = gr.Slider(0, 100, value=50, label="Needs")
+        wants = gr.Slider(0, 100, value=30, label="Wants")
+        savings = gr.Slider(0, 100, value=20, label="Savings")
 
-    run_btn = gr.Button("Calculate Budget")
+    calculate_btn = gr.Button("✨ Calculate Budget", variant="primary")
 
-    output_md = gr.Markdown()
-    output_csv = gr.File(label="Download CSV")
+    with gr.Row():
+        output_md = gr.Markdown()
+    with gr.Row():
+        output_chart = gr.Plot()
+    with gr.Row():
+        output_csv = gr.File(label="📥 Download CSV")
 
-    run_btn.click(
+    calculate_btn.click(
         calculate_budget,
         inputs=[income, currency, needs, wants, savings],
-        outputs=[output_md, output_csv],
+        outputs=[output_md, output_chart, output_csv],
     )
 
 demo.launch()