Update app.py

app.py

@@ -3,66 +3,107 @@ import pandas as pd
 from tinydb import TinyDB, Query
 from datetime import datetime
 
-# Initialize DB
+# Initialize database
 db = TinyDB("db.json")
 buyers_table = db.table("buyers")
 payments_table = db.table("payments")
 
-# Functions
-def add_buyer(name, total_due):
-    buyers_table.insert({"name": name, "total_due": total_due, "paid": 0.0})
-    return "Buyer added!"
+# --- Functions ---
+def add_buyer(name, total_due, guarantor1, guarantor2):
+    buyers_table.insert({
+        "name": name,
+        "total_due": total_due,
+        "paid": 0.0,
+        "guarantor1": guarantor1,
+        "guarantor2": guarantor2,
+        "last_payment": None
+    })
+    return f"✅ Buyer '{name}' added successfully!"
 
 def record_payment(buyer_name, amount):
     buyer = Query()
     old_paid = buyers_table.get(buyer.name == buyer_name)["paid"]
-    buyers_table.update({"paid": old_paid + amount}, buyer.name == buyer_name)
+    buyers_table.update({
+        "paid": old_paid + amount,
+        "last_payment": str(datetime.now())
+    }, buyer.name == buyer_name)
     payments_table.insert({"buyer": buyer_name, "amount": amount, "date": str(datetime.now())})
-    return "Payment recorded!"
+    return f"💰 Payment of {amount} recorded for {buyer_name}!"
 
 def dashboard():
     buyers = pd.DataFrame(buyers_table.all())
     if buyers.empty:
-        return "No buyers yet."
+        return "No buyers yet.", None
     buyers["remaining"] = buyers["total_due"] - buyers["paid"]
-    total_pending = buyers["remaining"].sum()
-    return buyers[["name","total_due","paid","remaining"]], f"Total Pending Money: {total_pending}"
+    
+    # Alert for monthly pending payments
+    alerts = []
+    now_month = datetime.now().month
+    for _, row in buyers.iterrows():
+        if row["last_payment"] is None:
+            alerts.append(f"{row['name']} has not paid yet this month!")
+        else:
+            last_month = datetime.fromisoformat(row["last_payment"]).month
+            if last_month < now_month:
+                alerts.append(f"{row['name']} missed payment this month!")
+    
+    # Build interactive table with Plotly
+    import plotly.graph_objects as go
+    fig = go.Figure(data=[go.Table(
+        header=dict(values=["Buyer","Guarantor1","Guarantor2","Total Due","Paid","Remaining"],
+                    fill_color='paleturquoise',
+                    align='center'),
+        cells=dict(values=[
+            buyers["name"], buyers["guarantor1"], buyers["guarantor2"],
+            buyers["total_due"], buyers["paid"], buyers["remaining"]
+        ],
+        fill_color='lavender',
+        align='center'))
+    ])
+    
+    return "\n".join(alerts) if alerts else "✅ All payments up-to-date", fig
 
 def generate_receipt(buyer_name, amount):
-    return f"Receipt\nBuyer: {buyer_name}\nAmount Paid: {amount}\nDate: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+    return f"""
+    🧾 Receipt
+    Buyer: {buyer_name}
+    Amount Paid: {amount}
+    Date: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
+    """
 
-# Gradio UI
+# --- Gradio Interface ---
 with gr.Blocks() as demo:
-    gr.Markdown("## 💰 Installment Management Dashboard")
-    
+    gr.Markdown("## 💳 Installment Management Dashboard (Enhanced)")
+
     with gr.Tab("Add Buyer"):
         name_input = gr.Textbox(label="Buyer Name")
         total_input = gr.Number(label="Total Due")
-        add_btn = gr.Button("Add Buyer")
-        output_add = gr.Textbox()
-        add_btn.click(add_buyer, inputs=[name_input, total_input], outputs=output_add)
-    
+        guarantor1_input = gr.Textbox(label="Guarantor 1 Name")
+        guarantor2_input = gr.Textbox(label="Guarantor 2 Name")
+        add_btn = gr.Button("➕ Add Buyer", elem_id="animated-btn")
+        add_output = gr.Textbox()
+        add_btn.click(add_buyer, inputs=[name_input, total_input, guarantor1_input, guarantor2_input], outputs=add_output)
+
     with gr.Tab("Record Payment"):
-        buyers_list = [b["name"] for b in buyers_table.all()]
-        if not buyers_list:
-            buyers_list = ["No buyers yet"]
+        buyers_list = [b["name"] for b in buyers_table.all()] or ["No buyers yet"]
         buyer_select = gr.Dropdown(choices=buyers_list, label="Select Buyer")
         amount_input = gr.Number(label="Payment Amount")
-        pay_btn = gr.Button("Submit Payment")
-        output_pay = gr.Textbox()
-        pay_btn.click(record_payment, inputs=[buyer_select, amount_input], outputs=output_pay)
-    
+        pay_btn = gr.Button("💵 Submit Payment", elem_id="animated-btn")
+        pay_output = gr.Textbox()
+        pay_btn.click(record_payment, inputs=[buyer_select, amount_input], outputs=pay_output)
+
     with gr.Tab("Dashboard"):
-        dash_btn = gr.Button("Refresh Dashboard")
-        dash_table = gr.Dataframe()
+        dash_btn = gr.Button("📊 Refresh Dashboard", elem_id="animated-btn")
         dash_text = gr.Textbox()
-        dash_btn.click(dashboard, outputs=[dash_table, dash_text])
-    
+        dash_table = gr.Plot()
+        dash_btn.click(dashboard, outputs=[dash_text, dash_table])
+
     with gr.Tab("Receipt"):
         rec_buyer = gr.Textbox(label="Buyer Name")
         rec_amount = gr.Number(label="Amount Paid")
-        rec_btn = gr.Button("Generate Receipt")
+        rec_btn = gr.Button("🧾 Generate Receipt", elem_id="animated-btn")
         rec_output = gr.Textbox()
         rec_btn.click(generate_receipt, inputs=[rec_buyer, rec_amount], outputs=rec_output)
 
-demo.launch()
+# --- CSS Animation for Buttons ---
+demo.launch(share=True, server_name="0.0.0.0", server_port=7860, inbrowser=False)