Update app.py

app.py

@@ -1,87 +1,65 @@
 import os
-import sys
 import subprocess
 import time
-import urllib.request
-import stat
-
-# --- 1. CLONE YOUR GITHUB REPO ---
-REPO_URL = "https://github.com/diplomat-bit/aibank.git"
-if not os.path.exists("aibank"):
-    print("Cloning repository...")
-    subprocess.run(["git", "clone", REPO_URL])
-
-# Add your SDK source code to the Python path
-sys.path.append(os.path.abspath("aibank/src"))
-
-# --- 2. DOWNLOAD PRISM MOCK SERVER (Standalone Binary) ---
-# Since we might not have NPM, we download the binary directly
-PRISM_BIN = "./prism"
-if not os.path.exists(PRISM_BIN):
-    print("Downloading Prism Mock Server binary...")
-    url = "https://github.com/stoplightio/prism/releases/download/v5.15.0/prism-linux-x64"
-    urllib.request.urlretrieve(url, PRISM_BIN)
-    # Make it executable
-    st = os.stat(PRISM_BIN)
-    os.chmod(PRISM_BIN, st.st_mode | stat.S_IEXEC)
+import yaml
+import gradio as gr
+from aibanking import Jocall3
 
-# --- 3. START THE MOCK SERVER ---
-# We use the OpenAPI spec URL found in your repo's .stats.yml
-SPEC_URL = "https://storage.googleapis.com/stainless-sdk-openapi-specs/citibank-demo-business-inc-diplomat-bit%2Fjocall3-d15e79b0c3e75e676b0109861fa8142f5121cc917e95d056c3ee9f41f9d803dc.yml"
+# 1. Dynamically get the OpenAPI spec from the cloned .stats.yml
+with open(".stats.yml", 'r') as f:
+    stats = yaml.safe_load(f)
+    OPENAPI_URL = stats.get('openapi_spec_url')
 
-print("Starting Mock Server on port 4010...")
-subprocess.Popen([PRISM_BIN, "mock", SPEC_URL, "-p", "4010", "-h", "0.0.0.0"])
-time.sleep(5) # Wait for server to start
+# 2. Start Prism in the background on port 4010
+print(f"Starting Prism Mock Server...")
+subprocess.Popen([
+    "prism", "mock", OPENAPI_URL, 
+    "-p", "4010", 
+    "-h", "0.0.0.0"
+])
 
-# --- 4. IMPORT YOUR SDK AND SETUP GRADIO ---
-# Now that the path is set, we can import from your 'aibanking' package
-from aibanking import Jocall3
-import gradio as gr
+# Give Prism time to boot
+time.sleep(5)
 
-# Initialize client to talk to the local Prism mock
-# Your SDK is already hardcoded to send 'x-api-key' automatically
+# 3. Setup the SDK Client
+# It handles the 'x-api-key' header automatically
 client = Jocall3(
     base_url="http://127.0.0.1:4010",
-    api_key="hf_mock_token_123"
+    api_key="mock-key-123"
 )
 
-def check_api():
+# 4. App Functions
+def check_status():
     try:
         status = client.system.get_status()
-        return {
-            "API Status": status.api_status,
-            "Mock Server": "Online",
-            "Gemini Uptime": status.gemini_uptime,
-            "Header Sent": "x-api-key"
-        }
+        return f"✅ API Status: {status.api_status}\nGemini Uptime: {status.gemini_uptime}"
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"❌ SDK Error: {str(e)}"
 
-def open_account_demo():
+def create_demo_account():
     try:
         acc = client.accounts.open(
             currency="USD",
-            initial_deposit=5000.0,
+            initial_deposit=1000.0,
             product_type="high_yield_vault"
         )
-        return f"Account Created!\nID: {acc.id}\nBalance: {acc.current_balance}"
+        return f"Account ID: {acc.id}\nBalance: {acc.current_balance}\nBank: {acc.institution_name}"
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"❌ Error: {str(e)}"
 
-# --- 5. THE UI ---
-with gr.Blocks(title="Jocall3 AI Banking Demo") as demo:
+# 5. UI Layout
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🏦 Jocall3 AI Banking SDK Demo")
-    gr.Markdown(f"Automatically pulling code from: `{REPO_URL}`")
+    gr.Markdown("Running a local **Prism Mock** server against the latest OpenAPI spec.")
     
     with gr.Row():
-        btn_status = gr.Button("Check System Health")
-        btn_acc = gr.Button("Test Account Opening")
+        btn_status = gr.Button("Check System Health", variant="primary")
+        btn_acc = gr.Button("Simulate Open Account")
     
-    output = gr.JSON(label="Response Data")
+    output = gr.Textbox(label="Response from SDK", lines=5)
 
-    btn_status.click(check_api, outputs=output)
-    btn_acc.click(open_account_demo, outputs=output)
+    btn_status.click(check_status, outputs=output)
+    btn_acc.click(create_demo_account, outputs=output)
 
 if __name__ == "__main__":
-    # Hugging Face uses port 7860
     demo.launch(server_name="0.0.0.0", server_port=7860)