Update app.py

app.py

@@ -1,104 +1,109 @@
-import os
-import subprocess
-import time
-import json
-
-import yaml
+import os, subprocess, time, yaml, json, inspect
 import gradio as gr
 from aibanking import Jocall3
 
-# 1. Start Prism
+# 1. Start Prism Mock Server
 with open(".stats.yml", 'r') as f:
     stats = yaml.safe_load(f)
     OPENAPI_URL = stats.get('openapi_spec_url')
 
+print(f"Starting Prism Mock Server for 152 endpoints...")
 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")
+# 2. Setup Client
+client = Jocall3(base_url="http://127.0.0.1:4010", api_key="boss-mode-key")
+
+# 3. Dynamic SDK Mapping
+# This maps the high-level resources in your SDK
+RESOURCES = {
+    "Accounts": client.accounts,
+    "AI - Oracle": client.ai.oracle,
+    "AI - Ads": client.ai.ads,
+    "AI - Incubator": client.ai.incubator,
+    "AI - Advisor": client.ai.advisor,
+    "AI - Agent": client.ai.agent,
+    "AI - Models": client.ai.models,
+    "Corporate": client.corporate,
+    "Corporate - Compliance": client.corporate.compliance,
+    "Corporate - Treasury": client.corporate.treasury,
+    "Corporate - Risk": client.corporate.risk,
+    "Corporate - Governance": client.corporate.governance,
+    "Investments": client.investments,
+    "Lending": client.lending,
+    "Marketplace": client.marketplace,
+    "Payments": client.payments,
+    "Payments - FX": client.payments.fx,
+    "Sustainability": client.sustainability,
+    "System": client.system,
+    "Transactions": client.transactions,
+    "Users": client.users,
+    "Users - Me": client.users.me,
+    "Web3": client.web3,
+    "Web3 - Wallets": client.web3.wallets,
+}
 
-def format_res(data):
-    try:
-        if hasattr(data, 'to_dict'): data = data.to_dict()
-        return json.dumps(data, indent=2)
-    except:
-        return str(data)
+def get_methods_for_resource(res_name):
+    res = RESOURCES.get(res_name)
+    # Get all public methods that aren't built-in properties
+    methods = [m for m, _ in inspect.getmembers(res, predicate=inspect.ismethod) if not m.startswith('_')]
+    return gr.Dropdown(choices=methods)
 
-# 3. Master Logic - mapped exactly to the MENU labels
-def master_controller(category, action, p1, p2):
+def execute_call(res_name, method_name, params_json):
     try:
-        # --- SYSTEM ---
-        if category == "System":
-            if action == "Get Status": return format_res(client.system.get_status())
-            if action == "Get Audit Logs": return format_res(client.system.get_audit_logs(limit=5))
-            if action == "Biometric Compare": 
-                client.system.verification.compare_biometric(sample_a=p1 or "s1", sample_b=p2 or "s2")
-                return json.dumps({"status": "Success", "message": "Biometric match verified by AI"}, indent=2)
-            if action == "Notify Templates": return format_res(client.system.notifications.list_templates())
-            if action == "Send Push":
-                client.system.notifications.send_push(body=p1 or "Alert", title="Bank Notification", user_id=p2 or "u1")
-                return json.dumps({"status": "Queued", "recipient": p2}, indent=2)
-
-        # --- USERS ---
-        if category == "Users":
-            if action == "Get Me": return format_res(client.users.me.retrieve())
-            if action == "Get Preferences": return format_res(client.users.me.preferences.retrieve())
-            if action == "List Devices": return format_res(client.users.me.devices.list())
-
-        # --- ACCOUNTS ---
-        if category == "Accounts":
-            if action == "List All": return format_res(client.accounts.retrieve_me())
-            if action == "Get Details": return format_res(client.accounts.retrieve_details(p1 or "acc_123"))
-
-        # --- AI ORACLE ---
-        if category == "AI Oracle":
-            if action == "Market Crash Prediction": return format_res(client.ai.oracle.predictions.retrieve_market_crash_probability())
-            if action == "Run Simulation": return format_res(client.ai.oracle.simulate.create(prompt=p1 or "Global Recession"))
-
-        # --- WEB3 ---
-        if category == "Web3":
-            if action == "List Wallets": return format_res(client.web3.wallets.list())
-            if action == "Network Status": return format_res(client.web3.network.get_status())
-
-        # --- PAYMENTS ---
-        if category == "Payments":
-            if action == "Get FX Rates": return format_res(client.payments.fx.get_rates(pair=p1 or "EURUSD"))
-            if action == "List All Payments": return format_res(client.payments.list())
-
-        return f"Error: Action '{action}' not found in router logic."
+        res = RESOURCES.get(res_name)
+        method = getattr(res, method_name)
+        
+        # Parse parameters from JSON box
+        kwargs = json.loads(params_json) if params_json.strip() else {}
+        
+        # Call the actual SDK
+        output = method(**kwargs)
+        
+        # Format result
+        if hasattr(output, 'to_dict'):
+            return json.dumps(output.to_dict(), indent=2)
+        return json.dumps(output, indent=2) if output is not None else "✅ Command Executed Successfully (No Content)"
     except Exception as e:
-        return f"❌ SDK Error: {str(e)}"
+        return f"❌ SDK Error: {str(e)}\n\nHint: Check if your JSON parameters match the SDK expectations."
 
-# 4. The Menu (Keep these labels consistent with the logic above)
-MENU = {
-    "System": ["Get Status", "Get Audit Logs", "Biometric Compare", "Notify Templates", "Send Push"],
-    "Users": ["Get Me", "Get Preferences", "List Devices"],
-    "Accounts": ["List All", "Get Details"],
-    "AI Oracle": ["Market Crash Prediction", "Run Simulation"],
-    "Web3": ["List Wallets", "Network Status"],
-    "Payments": ["Get FX Rates", "List All Payments"]
-}
-
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🏦 Jocall3 AI Banking: Enterprise Explorer")
+# 4. UI Layout
+with gr.Blocks(theme=gr.themes.Default(primary_hue="orange")) as demo:
+    gr.Markdown("# 🏦 Jocall3 AI Banking: UNLIMITED SDK EXPLORER")
+    gr.Markdown("### Direct Access to all 152 Endpoints via the aibanking SDK")
     
     with gr.Row():
         with gr.Column(scale=1):
-            category_radio = gr.Radio(list(MENU.keys()), label="1. Category")
-            action_dropdown = gr.Dropdown(choices=[], label="2. Action")
-            p1 = gr.Textbox(label="Param A (ID/Prompt/Sample)", placeholder="e.g. acc_123")
-            p2 = gr.Textbox(label="Param B (Sample/User ID)", placeholder="Optional...")
-            run_btn = gr.Button("🚀 EXECUTE", variant="primary")
-        
+            res_choice = gr.Dropdown(list(RESOURCES.keys()), label="1. Select Resource Group")
+            method_choice = gr.Dropdown([], label="2. Select SDK Method")
+            
+            gr.Markdown("#### 3. Parameters (JSON)")
+            params_input = gr.TextArea(
+                label="Arguments", 
+                placeholder='{\n  "account_id": "string",\n  "amount": 100\n}',
+                lines=8
+            )
+            
+            run_btn = gr.Button("🔥 EXECUTE SDK CALL", variant="primary")
+            
         with gr.Column(scale=2):
-            output_code = gr.Code(label="SDK Response", language="json", lines=25)
+            gr.Markdown("#### Live Mock Response")
+            output_display = gr.Code(label="JSON", language="json", lines=30)
 
-    def get_actions(cat):
-        return gr.Dropdown(choices=MENU.get(cat, []))
+    # UI Interactivity
+    res_choice.change(get_methods_for_resource, inputs=res_choice, outputs=method_choice)
+    
+    # Auto-fill parameter hints when method changes
+    def update_params_hint(res_name, method_name):
+        res = RESOURCES.get(res_name)
+        method = getattr(res, method_name)
+        sig = inspect.signature(method)
+        params = {k: "value" for k in sig.parameters.keys() if k not in ['extra_headers', 'extra_query', 'extra_body', 'timeout']}
+        return json.dumps(params, indent=2)
+
+    method_choice.change(update_params_hint, inputs=[res_choice, method_choice], outputs=params_input)
     
-    category_radio.change(get_actions, inputs=category_radio, outputs=action_dropdown)
-    run_btn.click(master_controller, inputs=[category_radio, action_dropdown, p1, p2], outputs=output_code)
+    run_btn.click(execute_call, inputs=[res_choice, method_choice, params_input], outputs=output_display)
 
 if __name__ == "__main__":
     demo.launch()