Update app.py

app.py

@@ -1,4 +1,4 @@
-import os, subprocess, time, yaml, json, inspect
+import os, subprocess, time, yaml, json, inspect, requests
 import gradio as gr
 from aibanking import Jocall3
 
@@ -7,15 +7,31 @@ 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...")
+print(f"🚀 Starting Prism Mock Server...")
+# Start Prism as a background process
 subprocess.Popen(["prism", "mock", OPENAPI_URL, "-p", "4010", "-h", "0.0.0.0"])
-time.sleep(5) 
 
-# 2. Setup Client
-client = Jocall3(base_url="http://127.0.0.1:4010", api_key="boss-mode-key")
+# 2. WAIT for Prism to be healthy (This prevents the 'Restarting' loop)
+max_retries = 20
+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!")
+        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")
 
-# 3. Dynamic SDK Mapping
-# This maps the high-level resources in your SDK
+# --- Dynamic Logic ---
 RESOURCES = {
     "Accounts": client.accounts,
     "AI - Oracle": client.ai.oracle,
@@ -23,87 +39,52 @@ RESOURCES = {
     "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,
+    "Web3 - Wallets": client.web3.wallets,
     "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 get_methods_for_resource(res_name):
+def get_methods(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)
 
-def execute_call(res_name, method_name, params_json):
+def update_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)
+
+def execute(res_name, method_name, params_json):
     try:
         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)"
+        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)"
     except Exception as e:
-        return f"❌ SDK Error: {str(e)}\n\nHint: Check if your JSON parameters match the SDK expectations."
+        return f"❌ Error: {str(e)}"
 
-# 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")
-    
+# 4. UI
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🏦 Jocall3 AI Banking: Enterprise Explorer")
     with gr.Row():
         with gr.Column(scale=1):
-            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")
-            
+            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")
         with gr.Column(scale=2):
-            gr.Markdown("#### Live Mock Response")
-            output_display = gr.Code(label="JSON", language="json", lines=30)
-
-    # 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)
+            output_display = gr.Code(label="SDK Response", language="json", lines=25)
 
-    method_choice.change(update_params_hint, inputs=[res_choice, method_choice], outputs=params_input)
-    
-    run_btn.click(execute_call, inputs=[res_choice, method_choice, params_input], outputs=output_display)
+    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)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)