Update app.py

app.py

@@ -2,32 +2,43 @@ import os, subprocess, time, yaml, json, inspect, requests
 import gradio as gr
 from aibanking import Jocall3
 
-# 1. Start Prism Mock Server
+# 1. Get your Secret/Env Var
+# Your SDK specifically looks for X_API_KEY, so we pull it here
+HF_API_KEY = os.environ.get("X_API_KEY", "mock-default-key")
+
+# 2. 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"🚀 Launching Prism for all endpoints...")
+print(f"🚀 Starting Prism Mock Server...")
 subprocess.Popen(["prism", "mock", OPENAPI_URL, "-p", "4010", "-h", "0.0.0.0"])
 
-# 2. FIX: Wait for a VALID endpoint (/system/status) instead of root (/)
+# 3. AUTHORIZED Health Check
+# Prism was throwing 401 because we didn't send the key. This fixed it:
 max_retries = 30
 prism_ready = False
 for i in range(max_retries):
     try:
-        # Hit a known endpoint to verify readiness
-        requests.get("http://127.0.0.1:4010/system/status", timeout=1)
-        print("✅ Prism is verified healthy!")
-        prism_ready = True
-        break
+        # We send the x-api-key header to satisfy the 'Invalid security scheme' error
+        response = requests.get(
+            "http://127.0.0.1:4010/system/status", 
+            headers={"x-api-key": HF_API_KEY}, 
+            timeout=1
+        )
+        if response.status_code < 500:
+            print("✅ Prism authorized and ready!")
+            prism_ready = True
+            break
     except:
-        print(f"⏳ Waiting for Prism... ({i+1}/{max_retries})")
+        print(f"⏳ Waiting for Prism to authorize... ({i+1}/{max_retries})")
         time.sleep(2)
 
-# 3. Setup Client
-client = Jocall3(base_url="http://127.0.0.1:4010", api_key="boss-mode")
+# 4. Setup SDK Client
+# It will use the local mock server but send your real X_API_KEY
+client = Jocall3(base_url="http://127.0.0.1:4010", api_key=HF_API_KEY)
 
-# 4. Map the entire SDK tree
+# 5. ALL Endpoints Mapping (Complete SDK Tree)
 RESOURCES = {
     "Accounts": client.accounts,
     "Accounts - Statements": client.accounts.statements,
@@ -44,31 +55,52 @@ RESOURCES = {
     "AI - Advisor Chat": client.ai.advisor.chat,
     "AI - Advisor Tools": client.ai.advisor.tools,
     "AI - Agent": client.ai.agent,
+    "AI - Agent Prompts": client.ai.agent.prompts,
     "AI - Models": client.ai.models,
     "Corporate": client.corporate,
     "Corporate - Compliance": client.corporate.compliance,
     "Corporate - Treasury": client.corporate.treasury,
     "Corporate - Risk": client.corporate.risk,
     "Corporate - Fraud": client.corporate.risk.fraud,
+    "Corporate - Fraud Rules": client.corporate.risk.fraud.rules,
     "Corporate - Governance": client.corporate.governance,
+    "Corporate - Anomalies": client.corporate.anomalies,
     "Investments": client.investments,
     "Investments - Portfolios": client.investments.portfolios,
+    "Investments - Assets": client.investments.assets,
+    "Investments - Performance": client.investments.performance,
     "Lending": client.lending,
     "Lending - Applications": client.lending.applications,
+    "Lending - Decisions": client.lending.decisions,
     "Marketplace": client.marketplace,
+    "Marketplace - Offers": client.marketplace.offers,
     "Payments": client.payments,
     "Payments - Domestic": client.payments.domestic,
     "Payments - International": client.payments.international,
     "Payments - FX": client.payments.fx,
     "Sustainability": client.sustainability,
+    "Sustainability - Impact": client.sustainability.impact,
+    "Sustainability - Offsets": client.sustainability.offsets,
     "System": client.system,
     "System - Notifications": client.system.notifications,
     "System - Verification": client.system.verification,
+    "System - Webhooks": client.system.webhooks,
+    "System - Sandbox": client.system.sandbox,
     "Transactions": client.transactions,
+    "Transactions - Recurring": client.transactions.recurring,
+    "Transactions - Insights": client.transactions.insights,
     "Users": client.users,
     "Users - Me": client.users.me,
+    "Users - Me Devices": client.users.me.devices,
+    "Users - Me Security": client.users.me.security,
+    "Users - Me Preferences": client.users.me.preferences,
+    "Users - Me Biometrics": client.users.me.biometrics,
     "Web3": client.web3,
     "Web3 - Wallets": client.web3.wallets,
+    "Web3 - NFTs": client.web3.nfts,
+    "Web3 - Contracts": client.web3.contracts,
+    "Web3 - Transactions": client.web3.transactions,
+    "Web3 - Network": client.web3.network,
 }
 
 def get_methods(res_name):
@@ -76,47 +108,44 @@ def get_methods(res_name):
     methods = [m for m, _ in inspect.getmembers(res, predicate=inspect.ismethod) if not m.startswith('_')]
     return gr.Dropdown(choices=sorted(methods))
 
-def generate_payload_template(res_name, method_name):
+def generate_payload(res_name, method_name):
     res = RESOURCES.get(res_name)
     method = getattr(res, method_name)
     sig = inspect.signature(method)
     payload = {}
     for name, param in sig.parameters.items():
-        if name in ['extra_headers', 'extra_query', 'extra_body', 'timeout', 'self']:
-            continue
-        # provide a smart hint based on type
-        payload[name] = "string" if param.annotation == str else 0.0
+        if name in ['extra_headers', 'extra_query', 'extra_body', 'timeout', 'self']: continue
+        if "amount" in name or "deposit" in name: payload[name] = 100.0
+        elif "id" in name or "token" in name: payload[name] = "string_id"
+        else: payload[name] = "string"
     return json.dumps(payload, indent=2)
 
-def run_api(res_name, method_name, params_json):
+def execute(res_name, method_name, params_json):
     try:
         res = RESOURCES.get(res_name)
         method = getattr(res, method_name)
         kwargs = json.loads(params_json) if params_json.strip() else {}
         output = method(**kwargs)
         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 "✅ Success: Command accepted by server."
+        return json.dumps(output, indent=2) if output is not None else "✅ Command Accepted"
     except Exception as e:
-        return f"❌ SDK Error: {str(e)}"
+        return f"❌ Error: {str(e)}"
 
-# 5. The UI
-with gr.Blocks(theme=gr.themes.Default(primary_hue="cyan")) as demo:
-    gr.Markdown("# 🏛️ Jocall3 AI Banking: The Complete 152-Endpoint Console")
-    gr.Markdown("This interface dynamically inspects the `aibanking` SDK and generates requests for the local Prism Mock.")
-    
+# 6. The UI
+with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
+    gr.Markdown("# 🏛️ Jocall3 SDK: The Complete 152-Endpoint Console")
     with gr.Row():
         with gr.Column(scale=1):
-            res_choice = gr.Dropdown(list(RESOURCES.keys()), label="1. Select API Resource Group")
-            method_choice = gr.Dropdown([], label="2. Select SDK Method")
-            params_input = gr.TextArea(label="3. Arguments (Auto-Generated JSON)", lines=12)
-            run_btn = gr.Button("🚀 EXECUTE REMOTE CALL", variant="primary")
-        
+            res_choice = gr.Dropdown(sorted(list(RESOURCES.keys())), label="Resource Group")
+            method_choice = gr.Dropdown([], label="SDK Method")
+            params_input = gr.TextArea(label="Arguments (JSON)", lines=10)
+            run_btn = gr.Button("🚀 EXECUTE COMMAND", variant="primary")
         with gr.Column(scale=2):
-            output_display = gr.Code(label="Mock Server Response", language="json", lines=30)
+            output_display = gr.Code(label="SDK Response", language="json", lines=25)
 
     res_choice.change(get_methods, inputs=res_choice, outputs=method_choice)
-    method_choice.change(generate_payload_template, inputs=[res_choice, method_choice], outputs=params_input)
-    run_btn.click(run_api, inputs=[res_choice, method_choice, params_input], outputs=output_display)
+    method_choice.change(generate_payload, inputs=[res_choice, method_choice], outputs=params_input)
+    run_btn.click(execute, inputs=[res_choice, method_choice, params_input], outputs=output_display)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)