Update app.py

app.py

@@ -110,6 +110,35 @@ def test_agent_availability(agent_url):
     except requests.exceptions.RequestException as e:
         return f"❌ Agent not available: {str(e)}", {"error": str(e)}
 
+# Enhanced test function that also shows the envelope
+def test_agent_with_envelope(agent_url):
+    try:
+        # Send a simple manifest request to test
+        envelope = build_manifest_request_envelope(agent_url)
+        json_str = envelope.to_json()
+        envelope_dict = json.loads(json_str)
+        json_payload_pretty = json.dumps({"openFloor": envelope_dict}, indent=2)
+        payload = {"openFloor": envelope_dict}
+        
+        headers = {'Content-Type': 'application/json'}
+        response = requests.post(agent_url, json=payload, headers=headers, timeout=5)
+        
+        if response.status_code == 200:
+            try:
+                response_json = response.json()
+                status_msg = f"✅ Agent available at {agent_url}"
+                return status_msg, response_json, json_payload_pretty, response.status_code
+            except:
+                status_msg = f"✅ Agent responded but with non-JSON"
+                return status_msg, {"raw_response": response.text}, json_payload_pretty, response.status_code
+        else:
+            status_msg = f"⚠️ Agent responded with status {response.status_code}"
+            return status_msg, {"status": response.status_code, "response": response.text}, json_payload_pretty, response.status_code
+            
+    except requests.exceptions.RequestException as e:
+        status_msg = f"❌ Agent not available: {str(e)}"
+        return status_msg, {"error": str(e)}, "", 0
+
 # Gradio UI with improved layout
 with gr.Blocks(title="OpenFloor Protocol Tester", theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
@@ -155,27 +184,13 @@ with gr.Blocks(title="OpenFloor Protocol Tester", theme=gr.themes.Soft()) as dem
         
         # Right Column - Results  
         with gr.Column(scale=3):
-            gr.Markdown("### 🔗 Assistant Status")
-            
-            with gr.Row():
-                availability_status = gr.Textbox(
-                    label="Connection Status",
-                    interactive=False,
-                    container=False
-                )
-            
-            availability_response = gr.JSON(
-                label="Availability Response", 
-                container=False
-            )
-            
             gr.Markdown("### 📡 Request & Response")
             
-            # Status message for the main request
+            # Status message for all requests
             request_status = gr.Textbox(
-                label="Request Status",
+                label="Status",
                 interactive=False,
-                visible=False
+                container=False
             )
             
             # Collapsible request details
@@ -188,7 +203,7 @@ with gr.Blocks(title="OpenFloor Protocol Tester", theme=gr.themes.Soft()) as dem
             
             # Response section
             with gr.Group():
-                gr.Markdown("#### 📥 Assistant Response")
+                gr.Markdown("#### 📥 Response")
                 
                 with gr.Row():
                     status_code = gr.Number(
@@ -205,19 +220,19 @@ with gr.Blocks(title="OpenFloor Protocol Tester", theme=gr.themes.Soft()) as dem
     
     # Event handlers
     test_btn.click(
-        fn=test_agent_availability,
+        fn=test_agent_with_envelope,
         inputs=[agent_url],
-        outputs=[availability_status, availability_response]
+        outputs=[request_status, response_output, envelope_output, status_code]
     )
     
     def handle_send_request(*args):
         envelope, status, response, status_msg = send_request(*args)
-        return envelope, status, response, status_msg, gr.update(visible=True)
+        return envelope, status, response, status_msg
     
     send_btn.click(
         fn=handle_send_request,
         inputs=[agent_url, request_type, message_text],
-        outputs=[envelope_output, status_code, response_output, request_status, request_status]
+        outputs=[envelope_output, status_code, response_output, request_status]
     )
     
     gr.Markdown("""