Update app.py

app.py

@@ -1,73 +1,56 @@
 import gradio as gr
 import subprocess
-import os
-import time
 
-# Directories
-COBOL_DIR = "cobol"
-DATA_DIR = "data"
-BIN_DIR = "cobol/bin"
+# Run setup.sh once at startup
+def run_setup():
+    result = subprocess.run(
+        ["sh", "scripts/setup.sh"],
+        capture_output=True,
+        text=True
+    )
+    return result.stdout + "\n" + result.stderr
 
-# Ensure bin directory exists
-os.makedirs(BIN_DIR, exist_ok=True)
+# Run setup to compile COBOL files
+print("🚀 Running setup...")
+setup_output = run_setup()
+print(setup_output)
 
-def compile_cobol():
-    print("🔍 Checking for existing compiled binaries...")
-    programs = ['account', 'loan', 'interest']
-    all_compiled = True
-    for prog in programs:
-        if not os.path.exists(f"{BIN_DIR}/{prog}"):
-            all_compiled = False
-            break
-
-    if all_compiled:
-        print("✅ All COBOL binaries already compiled.")
-        return
-
-    print("⚙️ Compiling COBOL programs...")
-
-    # Install GnuCOBOL if needed
+# Helper function to run COBOL binaries
+def run_cobol_binary(binary_name, input_data=None):
     try:
-        subprocess.run(["cobc", "--version"], capture_output=True, check=True)
-    except FileNotFoundError:
-        print("📦 Installing GNU COBOL...")
-        subprocess.run(["apt-get", "update"], check=True)
-        subprocess.run(["apt-get", "install", "-y", "gnucobol4"], check=True)
+        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)
 
-    # Compile each program
-    for prog in programs:
-        cbl_file = f"{COBOL_DIR}/{prog}.cbl"
-        print(f"🧾 Compiling {prog} from {cbl_file}")
-        subprocess.run(["cobc", "-x", cbl_file, "-o", f"{BIN_DIR}/{prog}"], check=True)
-        print(f"✅ Compiled {prog}")
-
-    print("🎉 All COBOL programs compiled successfully!")
-
-# Functions to call compiled COBOL binaries
+# --- Account Balance ---
 def get_balance(account_number):
-    result = subprocess.run([f"{BIN_DIR}/account", str(account_number)],
-                            capture_output=True,
-                            text=True)
-    return result.stdout or "Account not found"
+    input_str = f"{account_number}\n"
+    return run_cobol_binary("account", input_str)
 
+# --- Loan Status ---
 def check_loan_status(account_number):
-    result = subprocess.run([f"{BIN_DIR}/loan", str(account_number)],
-                            capture_output=True,
-                            text=True)
-    return result.stdout or "Loan not found"
+    input_str = f"{account_number}\n"
+    return run_cobol_binary("loan", input_str)
 
-def calculate_interest(principal, rate, time):
-    input_data = f"{principal}\n{rate}\n{time}\n"
-    result = subprocess.run([f"{BIN_DIR}/interest"],
-                            input=input_data,
-                            capture_output=True,
-                            text=True)
-    return result.stdout or "Error calculating interest"
+# --- Gradio UI ---
+with gr.Blocks(title="COBOL Bank Demo") as demo:
+    gr.Markdown("# 💰 COBOL Banking System\nRun real COBOL programs in your browser!")
 
-# Start compilation on startup
-with gr.Blocks(title="Banking System") as demo:
-    gr.Markdown("# 💰 COBOL Banking System\nRun in-browser using Gradio + HF Spaces")
-    
     with gr.Tab("Account Balance"):
         acc_input = gr.Number(label="Enter Account Number")
         acc_output = gr.Textbox(label="Balance Info")
@@ -80,14 +63,4 @@ with gr.Blocks(title="Banking System") as demo:
         loan_btn = gr.Button("Check Loan")
         loan_btn.click(fn=check_loan_status, inputs=loan_input, outputs=loan_output)
 
-    with gr.Tab("Interest Calculator"):
-        prin = gr.Number(label="Principal Amount")
-        rt = gr.Number(label="Rate (%)")
-        tm = gr.Number(label="Time (years)")
-        int_output = gr.Textbox(label="Result")
-        int_btn = gr.Button("Calculate Interest")
-        int_btn.click(fn=calculate_interest, inputs=[prin, rt, tm], outputs=int_output)
-
-if __name__ == "__main__":
-    compile_cobol()
-    demo.launch()
+demo.launch()