Update app.py

app.py

@@ -1,66 +1,79 @@
 import gradio as gr
 import subprocess
+import pandas as pd
+from pathlib import Path
 
-# Run setup.sh once at startup
-def run_setup():
+def run_cobol_binary(action, data):
     result = subprocess.run(
-        ["sh", "scripts/setup.sh"],
+        ["./cobol/account", action, data],
         capture_output=True,
         text=True
     )
-    return result.stdout + "\n" + result.stderr
+    return result.stdout or result.stderr
 
-# Run setup to compile COBOL files
-print("🚀 Running setup...")
-setup_output = run_setup()
-print(setup_output)
+def create_account(number, name, balance):
+    input_str = f"{number},{name},{balance}"
+    result = subprocess.run(["./cobol/account", "CREATE", input_str], capture_output=True, text=True)
+    return "✅ Account created"
 
-# Helper function to run COBOL binaries
-def run_cobol_binary(binary_name, input_data=None):
-    try:
-        if input_data:
-            result = subprocess.run(
-                [f"./cobol/{binary_name}"],
-                input=input_data,
-                capture_output=True,
-                text=True,
-                timeout=10
-            )
-        else:
-            result = subprocess.run(
-                [f"./cobol/{binary_name}"],
-                capture_output=True,
-                text=True,
-                timeout=10
-            )
-        return result.stdout or result.stderr
-    except Exception as e:
-        return str(e)
+def create_loan(lid, acc_id, amount, rate, status):
+    input_str = f"{lid},{acc_id},{amount},{rate},{status}"
+    result = subprocess.run(["./cobol/loan", "CREATE", input_str], capture_output=True, text=True)
+    return "✅ Loan created"
 
-# --- Account Balance ---
-def get_balance(account_number):
-    input_str = f"{account_number}\n"
-    return run_cobol_binary("account", input_str)
+def export_accounts():
+    df = pd.read_csv("data/accounts.csv")
+    path = "export_accounts.csv"
+    df.to_csv(path, index=False)
+    return path
 
-# --- Loan Status ---
-def check_loan_status(account_number):
-    input_str = f"{account_number}\n"
-    return run_cobol_binary("loan", input_str)
+def export_loans():
+    df = pd.read_csv("data/loans.csv")
+    path = "export_loans.csv"
+    df.to_csv(path, index=False)
+    return path
 
-# --- Gradio UI ---
-with gr.Blocks(title="COBOL Bank Demo") as demo:
-    gr.Markdown("# 💰 COBOL Banking System\nRun real COBOL programs in your browser!")
+def get_log():
+    with open("data/transactions.log", "r") as f:
+        content = f.read()
+    return f"""
+    <div class="transaction-log">
+    {content.replace('\n', '<br>')}
+    </div>
+    """
 
-    with gr.Tab("Account Balance"):
-        acc_input = gr.Number(label="Enter Account Number")
-        acc_output = gr.Textbox(label="Balance Info")
-        acc_btn = gr.Button("Check Balance")
-        acc_btn.click(fn=get_balance, inputs=acc_input, outputs=acc_output)
+with gr.Blocks(title="COBOL Bank Demo", css="file=static/style.css") as demo:
+    gr.Markdown("# 💰 COBOL Banking System\nCreate, read, and export accounts and loans")
 
-    with gr.Tab("Loan Status"):
-        loan_input = gr.Number(label="Enter Account Number")
-        loan_output = gr.Textbox(label="Loan Info")
-        loan_btn = gr.Button("Check Loan")
-        loan_btn.click(fn=check_loan_status, inputs=loan_input, outputs=loan_output)
+    with gr.Tab("Accounts"):
+        with gr.Row():
+            acc_num = gr.Number(label="Account Number")
+            acc_name = gr.Textbox(label="Name")
+            acc_bal = gr.Number(label="Balance")
+        acc_btn = gr.Button("Create Account")
+        acc_btn.click(fn=create_account, inputs=[acc_num, acc_name, acc_bal], outputs=[])
+
+        exp_acc_btn = gr.Button("Export Accounts")
+        exp_acc_file = gr.File(label="Download Accounts CSV")
+        exp_acc_btn.click(fn=export_accounts, outputs=exp_acc_file)
+
+    with gr.Tab("Loans"):
+        with gr.Row():
+            loan_id = gr.Textbox(label="Loan ID")
+            loan_acc = gr.Number(label="Account Number")
+            loan_amt = gr.Number(label="Amount")
+            loan_rate = gr.Number(label="Rate (%)")
+            loan_status = gr.Textbox(label="Status")
+        loan_btn = gr.Button("Create Loan")
+        loan_btn.click(fn=create_loan, inputs=[loan_id, loan_acc, loan_amt, loan_rate, loan_status], outputs=[])
+
+        exp_loan_btn = gr.Button("Export Loans")
+        exp_loan_file = gr.File(label="Download Loans CSV")
+        exp_loan_btn.click(fn=export_loans, outputs=exp_loan_file)
+
+    with gr.Tab("Transaction Log"):
+        log_box = gr.HTML(label="Logs", value=get_log)
+        refresh_btn = gr.Button("Refresh Log")
+        refresh_btn.click(fn=get_log, outputs=log_box)
 
 demo.launch()