Create app.py

app.py

@@ -0,0 +1,105 @@
+import gradio as gr
+import pandas as pd
+import numpy as np
+
+# Global variables
+expenses = []
+
+# Add Expense
+def add_expense(description, amount, payer, participants):
+    global expenses
+    participants = participants.split(",")  # Split participants by comma
+    amount = float(amount)
+    expense = {
+        "Description": description,
+        "Amount": amount,
+        "Payer": payer,
+        "Participants": participants,
+        "Split Amount": round(amount / len(participants), 2),
+    }
+    expenses.append(expense)
+    return pd.DataFrame(expenses)
+
+# Optimize Balances
+def optimize_balances():
+    global expenses
+    # Create a balance sheet
+    balances = {}
+    for expense in expenses:
+        payer = expense["Payer"]
+        participants = expense["Participants"]
+        split_amount = expense["Split Amount"]
+
+        # Adjust balances for payer
+        balances[payer] = balances.get(payer, 0) + (len(participants) * split_amount - expense["Amount"])
+        
+        # Adjust balances for participants
+        for participant in participants:
+            if participant != payer:
+                balances[participant] = balances.get(participant, 0) - split_amount
+
+    # Simplify debts
+    def simplify_debts(balances):
+        transactions = []
+        debtors = {person: bal for person, bal in balances.items() if bal < 0}
+        creditors = {person: bal for person, bal in balances.items() if bal > 0}
+
+        while debtors and creditors:
+            # Pick a debtor and creditor
+            debtor, debt_amount = min(debtors.items(), key=lambda x: x[1])
+            creditor, credit_amount = max(creditors.items(), key=lambda x: x[1])
+
+            # Settle the minimum of the two
+            settled_amount = min(-debt_amount, credit_amount)
+            transactions.append(f"{debtor} pays {creditor} ${settled_amount:.2f}")
+
+            # Update balances
+            balances[debtor] += settled_amount
+            balances[creditor] -= settled_amount
+
+            # Remove settled accounts
+            if balances[debtor] == 0:
+                del debtors[debtor]
+            else:
+                debtors[debtor] = balances[debtor]
+
+            if balances[creditor] == 0:
+                del creditors[creditor]
+            else:
+                creditors[creditor] = balances[creditor]
+
+        return transactions
+
+    return simplify_debts(balances)
+
+# Reset App
+def reset():
+    global expenses
+    expenses = []
+    return pd.DataFrame(expenses), []
+
+# Gradio Interface
+with gr.Blocks() as app:
+    gr.Markdown("# Expense Splitter App")
+
+    with gr.Row():
+        with gr.Column():
+            description = gr.Textbox(label="Description", placeholder="e.g., Dinner")
+            amount = gr.Number(label="Amount", placeholder="e.g., 100")
+            payer = gr.Textbox(label="Payer", placeholder="e.g., Alice")
+            participants = gr.Textbox(label="Participants", placeholder="e.g., Alice,Bob,Charlie")
+            add_btn = gr.Button("Add Expense")
+        with gr.Column():
+            expense_table = gr.Dataframe(headers=["Description", "Amount", "Payer", "Participants", "Split Amount"], datatype=["str", "number", "str", "list", "number"])
+
+    add_btn.click(add_expense, inputs=[description, amount, payer, participants], outputs=expense_table)
+
+    with gr.Row():
+        optimize_btn = gr.Button("Optimize Balances")
+        result = gr.Textbox(label="Transactions", lines=5)
+        reset_btn = gr.Button("Reset")
+
+    optimize_btn.click(optimize_balances, inputs=[], outputs=result)
+    reset_btn.click(reset, inputs=[], outputs=[expense_table, result])
+
+app.launch()