Update app.py

app.py

@@ -7,84 +7,116 @@ 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...")
-# Start Prism as a background process
+print(f"🚀 Launching Prism for all endpoints...")
 subprocess.Popen(["prism", "mock", OPENAPI_URL, "-p", "4010", "-h", "0.0.0.0"])
 
-# 2. WAIT for Prism to be healthy (This prevents the 'Restarting' loop)
-max_retries = 20
+# 2. FIX: Wait for a VALID endpoint (/system/status) instead of root (/)
+max_retries = 30
 prism_ready = False
 for i in range(max_retries):
     try:
-        # Try to hit the Prism health check/root
-        response = requests.get("http://127.0.0.1:4010", timeout=1)
-        print("✅ Prism is up and running!")
+        # 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
     except:
         print(f"⏳ Waiting for Prism... ({i+1}/{max_retries})")
         time.sleep(2)
 
-if not prism_ready:
-    print("❌ Prism failed to start. App might crash.")
-
 # 3. Setup Client
-client = Jocall3(base_url="http://127.0.0.1:4010", api_key="demo-key")
+client = Jocall3(base_url="http://127.0.0.1:4010", api_key="boss-mode")
 
-# --- Dynamic Logic ---
+# 4. Map the entire SDK tree
 RESOURCES = {
     "Accounts": client.accounts,
+    "Accounts - Statements": client.accounts.statements,
+    "Accounts - Transactions": client.accounts.transactions,
+    "Accounts - Overdraft": client.accounts.overdraft_settings,
     "AI - Oracle": client.ai.oracle,
+    "AI - Oracle Predictions": client.ai.oracle.predictions,
+    "AI - Oracle Simulations": client.ai.oracle.simulations,
     "AI - Ads": client.ai.ads,
+    "AI - Ads Generate": client.ai.ads.generate,
     "AI - Incubator": client.ai.incubator,
+    "AI - Incubator Pitch": client.ai.incubator.pitch,
     "AI - Advisor": client.ai.advisor,
+    "AI - Advisor Chat": client.ai.advisor.chat,
+    "AI - Advisor Tools": client.ai.advisor.tools,
     "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,
-    "Web3 - Wallets": client.web3.wallets,
+    "Corporate - Fraud": client.corporate.risk.fraud,
+    "Corporate - Governance": client.corporate.governance,
+    "Investments": client.investments,
+    "Investments - Portfolios": client.investments.portfolios,
+    "Lending": client.lending,
+    "Lending - Applications": client.lending.applications,
+    "Marketplace": client.marketplace,
     "Payments": client.payments,
+    "Payments - Domestic": client.payments.domestic,
+    "Payments - International": client.payments.international,
+    "Payments - FX": client.payments.fx,
+    "Sustainability": client.sustainability,
     "System": client.system,
+    "System - Notifications": client.system.notifications,
+    "System - Verification": client.system.verification,
     "Transactions": client.transactions,
+    "Users": client.users,
+    "Users - Me": client.users.me,
+    "Web3": client.web3,
+    "Web3 - Wallets": client.web3.wallets,
 }
 
 def get_methods(res_name):
     res = RESOURCES.get(res_name)
     methods = [m for m, _ in inspect.getmembers(res, predicate=inspect.ismethod) if not m.startswith('_')]
-    return gr.Dropdown(choices=methods)
+    return gr.Dropdown(choices=sorted(methods))
 
-def update_hint(res_name, method_name):
+def generate_payload_template(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)
+    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
+    return json.dumps(payload, indent=2)
 
-def execute(res_name, method_name, params_json):
+def run_api(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 (No Content)"
+        return json.dumps(output, indent=2) if output is not None else "✅ Success: Command accepted by server."
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"❌ SDK Error: {str(e)}"
 
-# 4. UI
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🏦 Jocall3 AI Banking: Enterprise Explorer")
+# 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.")
+    
     with gr.Row():
         with gr.Column(scale=1):
-            res_choice = gr.Dropdown(list(RESOURCES.keys()), label="1. Resource")
-            method_choice = gr.Dropdown([], label="2. Method")
-            params_input = gr.TextArea(label="3. Arguments (JSON)", lines=8)
-            run_btn = gr.Button("🚀 EXECUTE", variant="primary")
+            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")
+        
         with gr.Column(scale=2):
-            output_display = gr.Code(label="SDK Response", language="json", lines=25)
+            output_display = gr.Code(label="Mock Server Response", language="json", lines=30)
 
     res_choice.change(get_methods, inputs=res_choice, outputs=method_choice)
-    method_choice.change(update_hint, inputs=[res_choice, method_choice], outputs=params_input)
-    run_btn.click(execute, inputs=[res_choice, method_choice, params_input], outputs=output_display)
+    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)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)