Update app.py

app.py

@@ -1,65 +1,109 @@
-import os
-import subprocess
-import time
-import yaml
+import os, subprocess, time, yaml, json
 import gradio as gr
 from aibanking import Jocall3
 
-# 1. Dynamically get the OpenAPI spec from the cloned .stats.yml
+# 1. Start Prism (using the spec URL in your .stats.yml)
 with open(".stats.yml", 'r') as f:
     stats = yaml.safe_load(f)
     OPENAPI_URL = stats.get('openapi_spec_url')
 
-# 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"
-])
-
-# Give Prism time to boot
-time.sleep(5)
-
-# 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="mock-key-123"
-)
-
-# 4. App Functions
-def check_status():
+subprocess.Popen(["prism", "mock", OPENAPI_URL, "-p", "4010", "-h", "0.0.0.0"])
+time.sleep(5) 
+
+# 2. Setup SDK Client
+client = Jocall3(base_url="http://127.0.0.1:4010", api_key="demo-key")
+
+# --- Helper for pretty printing ---
+def format_res(data):
     try:
-        status = client.system.get_status()
-        return f"✅ API Status: {status.api_status}\nGemini Uptime: {status.gemini_uptime}"
-    except Exception as e:
-        return f"❌ SDK Error: {str(e)}"
+        # If it's a Pydantic model (SDK response), convert to dict
+        if hasattr(data, 'to_dict'):
+            return json.dumps(data.to_dict(), indent=2)
+        return json.dumps(data, indent=2)
+    except:
+        return str(data)
 
-def create_demo_account():
+# --- Resource Handlers ---
+def api_router(category, action, param1=None, param2=None):
     try:
-        acc = client.accounts.open(
-            currency="USD",
-            initial_deposit=1000.0,
-            product_type="high_yield_vault"
-        )
-        return f"Account ID: {acc.id}\nBalance: {acc.current_balance}\nBank: {acc.institution_name}"
+        # SYSTEM
+        if category == "System":
+            if action == "Status": return format_res(client.system.get_status())
+            if action == "Audit Logs": return format_res(client.system.get_audit_logs(limit=5))
+            if action == "Webhooks": return format_res(client.system.webhooks.list())
+        
+        # USERS
+        if category == "Users":
+            if action == "Me": return format_res(client.users.me.retrieve())
+            if action == "Preferences": return format_res(client.users.me.preferences.retrieve())
+            if action == "Devices": return format_res(client.users.me.devices.list())
+            if action == "Security Log": return format_res(client.users.me.security.retrieve_log())
+
+        # ACCOUNTS
+        if category == "Accounts":
+            if action == "List All": return format_res(client.accounts.retrieve_me())
+            if action == "Details": return format_res(client.accounts.retrieve_details(param1 or "acc_123"))
+            if action == "History": return format_res(client.accounts.retrieve_balance_history(param1 or "acc_123"))
+
+        # AI ORACLE
+        if category == "AI Oracle":
+            if action == "Market Crash Prob": return format_res(client.ai.oracle.predictions.retrieve_market_crash_probability())
+            if action == "Inflation": return format_res(client.ai.oracle.predictions.retrieve_inflation(region=param1 or "GLOBAL"))
+            if action == "Simulate": return format_res(client.ai.oracle.simulate.create(prompt=param1 or "Recession scenario"))
+
+        # CORPORATE
+        if category == "Corporate":
+            if action == "Risk Exposure": return format_res(client.corporate.risk.get_risk_exposure())
+            if action == "List Cards": return format_res(client.corporate.cards.list_all())
+            if action == "Anomalies": return format_res(client.corporate.anomalies.list_detected())
+            if action == "Liquidity": return format_res(client.corporate.treasury.get_liquidity_positions())
+
+        # WEB3
+        if category == "Web3":
+            if action == "Wallets": return format_res(client.web3.wallets.list())
+            if action == "Network": return format_res(client.web3.network.get_status())
+            if action == "NFTs": return format_res(client.web3.nfts.list())
+
+        # PAYMENTS / FX
+        if category == "Payments":
+            if action == "FX Rates": return format_res(client.payments.fx.get_rates(pair=param1 or "EURUSD"))
+            if action == "List Payments": return format_res(client.payments.list())
+
+        return "Action not implemented in UI yet, but endpoint is live in Prism."
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"❌ SDK Error: {str(e)}"
 
-# 5. UI Layout
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🏦 Jocall3 AI Banking SDK Demo")
-    gr.Markdown("Running a local **Prism Mock** server against the latest OpenAPI spec.")
+# --- UI Layout ---
+with gr.Blocks(title="Jocall3 Full API Explorer") as demo:
+    gr.Markdown("# 🏛️ Jocall3 AI Banking: 152 Endpoints Live")
+    gr.Markdown("Select a category to test the SDK against the Prism Mock Server.")
     
     with gr.Row():
-        btn_status = gr.Button("Check System Health", variant="primary")
-        btn_acc = gr.Button("Simulate Open Account")
-    
-    output = gr.Textbox(label="Response from SDK", lines=5)
+        with gr.Column(scale=1):
+            cat = gr.Radio(["System", "Users", "Accounts", "AI Oracle", "Corporate", "Web3", "Payments"], label="Resource Category")
+            act = gr.Dropdown(choices=[], label="Action / Endpoint")
+            p1 = gr.Textbox(label="Parameter 1 (ID/Prompt/Region)", visible=True)
+            run_btn = gr.Button("Execute Request", variant="primary")
+
+        with gr.Column(scale=2):
+            output = gr.Code(label="SDK JSON Response", language="json", lines=25)
+
+    # Dynamic Dropdown Logic
+    def update_actions(category):
+        actions = {
+            "System": ["Status", "Audit Logs", "Webhooks"],
+            "Users": ["Me", "Preferences", "Devices", "Security Log"],
+            "Accounts": ["List All", "Details", "History"],
+            "AI Oracle": ["Market Crash Prob", "Inflation", "Simulate"],
+            "Corporate": ["Risk Exposure", "List Cards", "Anomalies", "Liquidity"],
+            "Web3": ["Wallets", "Network", "NFTs"],
+            "Payments": ["FX Rates", "List Payments"]
+        }
+        return gr.Dropdown(choices=actions.get(category, []))
 
-    btn_status.click(check_status, outputs=output)
-    btn_acc.click(create_demo_account, outputs=output)
+    cat.change(update_actions, inputs=cat, outputs=act)
+    run_btn.click(api_router, inputs=[cat, act, p1], outputs=output)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)