Update app.py

app.py

@@ -723,136 +723,156 @@ def create_gradio_interface() -> gr.Blocks:
         **Deployed on Hugging Face Spaces with built-in Excel MCP Server support**
         """)
         
+        # =============================
+        # CUSTODIAN AGENT SECTION
+        # =============================
+        gr.Markdown("---")
+        gr.Markdown("### 🏛️ Custodian Agent")
+        
         with gr.Row():
-            # =============================
-            # LEFT COLUMN: Agent Instructions
-            # =============================
             with gr.Column(scale=2):
-                gr.Markdown("### 💬 Agent Instructions")
-                
                 custodian_prompt = gr.Textbox(
-                    label="🏛️ Custodian Instructions",
+                    label="Instructions",
                     placeholder="e.g., 'Look for AAPL positions affected by stock split'",
                     lines=2
                 )
-                
+            with gr.Column(scale=2):
+                custodian_upload = gr.File(
+                    label="Upload Files",
+                    file_count="multiple",
+                    file_types=[".xlsx", ".xls", ".csv"]
+                )
+                custodian_process_btn = gr.Button("🔄 Process Files", variant="secondary")
+            with gr.Column(scale=2):
+                custodian_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
+        
+        custodian_results = gr.Textbox(label="Results", lines=2, interactive=False)
+        
+        # =============================
+        # CASH AGENT SECTION
+        # =============================
+        gr.Markdown("---")
+        gr.Markdown("### 💰 Cash Agent")
+        
+        with gr.Row():
+            with gr.Column(scale=2):
                 cash_prompt = gr.Textbox(
-                    label="💰 Cash Instructions", 
+                    label="Instructions", 
                     placeholder="e.g., 'Flag any trades over $1M requiring additional liquidity'",
                     lines=2
                 )
-                
+            with gr.Column(scale=2):
+                cash_upload = gr.File(
+                    label="Upload Files",
+                    file_count="multiple",
+                    file_types=[".xlsx", ".xls", ".csv"]
+                )
+                cash_process_btn = gr.Button("🔄 Process Files", variant="secondary")
+            with gr.Column(scale=2):
+                cash_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
+        
+        cash_results = gr.Textbox(label="Results", lines=2, interactive=False)
+        
+        # =============================
+        # TRADE AGENT SECTION
+        # =============================
+        gr.Markdown("---")
+        gr.Markdown("### 📈 Trade Agent")
+        
+        with gr.Row():
+            with gr.Column(scale=2):
                 trade_prompt = gr.Textbox(
-                    label="📈 Trade Instructions",
+                    label="Instructions",
                     placeholder="e.g., 'Combine related trades from same counterparty'",
                     lines=2
                 )
-                
+            with gr.Column(scale=2):
+                trade_upload = gr.File(
+                    label="Upload Files",
+                    file_count="multiple",
+                    file_types=[".xlsx", ".xls", ".csv"]
+                )
+                trade_process_btn = gr.Button("🔄 Process Files", variant="secondary")
+            with gr.Column(scale=2):
+                trade_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
+        
+        trade_results = gr.Textbox(label="Results", lines=2, interactive=False)
+        
+        # =============================
+        # CORPORATE AGENT SECTION
+        # =============================
+        gr.Markdown("---")
+        gr.Markdown("### 🏢 Corporate Agent")
+        
+        with gr.Row():
+            with gr.Column(scale=2):
                 corporate_prompt = gr.Textbox(
-                    label="🏢 Corporate Instructions",
+                    label="Instructions",
                     placeholder="e.g., 'Check for dividend payments affecting settlement dates'",
                     lines=2
                 )
-                
+            with gr.Column(scale=2):
+                corporate_upload = gr.File(
+                    label="Upload Files",
+                    file_count="multiple",
+                    file_types=[".xlsx", ".xls", ".csv"]
+                )
+                corporate_process_btn = gr.Button("🔄 Process Files", variant="secondary")
+            with gr.Column(scale=2):
+                corporate_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
+        
+        corporate_results = gr.Textbox(label="Results", lines=2, interactive=False)
+        
+        # =============================
+        # WEB AGENT SECTION
+        # =============================
+        gr.Markdown("---")
+        gr.Markdown("### 🌐 Web Agent")
+        
+        with gr.Row():
+            with gr.Column(scale=2):
                 web_prompt = gr.Textbox(
-                    label="🌐 Web Agent Instructions",
+                    label="Instructions",
                     placeholder="e.g., 'Search for AAPL stock split news, NVDA earnings updates, CTSH corporate actions'",
                     lines=2
                 )
-            
-            # =============================
-            # CENTER COLUMN: Agent File Uploads & Status
-            # =============================
-            with gr.Column(scale=4):
-                gr.Markdown("### 📁 Agent File Processing & Status")
-                
-                # Custodian Agent Row
-                with gr.Row():
-                    gr.Markdown("#### 🏛️ Custodian Agent")
-                with gr.Row():
-                    with gr.Column(scale=2):
-                        custodian_upload = gr.File(
-                            label="Upload Files",
-                            file_count="multiple",
-                            file_types=[".xlsx", ".xls", ".csv"]
-                        )
-                        custodian_process_btn = gr.Button("🔄 Process Files", variant="secondary")
-                    with gr.Column(scale=2):
-                        custodian_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
-                custodian_results = gr.Textbox(label="Results", lines=2, interactive=False)
-                
-                # Cash Agent Row
-                with gr.Row():
-                    gr.Markdown("#### 💰 Cash Agent")
-                with gr.Row():
-                    with gr.Column(scale=2):
-                        cash_upload = gr.File(
-                            label="Upload Files",
-                            file_count="multiple",
-                            file_types=[".xlsx", ".xls", ".csv"]
-                        )
-                        cash_process_btn = gr.Button("🔄 Process Files", variant="secondary")
-                    with gr.Column(scale=2):
-                        cash_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
-                cash_results = gr.Textbox(label="Results", lines=2, interactive=False)
-                
-                # Trade Agent Row
-                with gr.Row():
-                    gr.Markdown("#### 📈 Trade Agent")
-                with gr.Row():
-                    with gr.Column(scale=2):
-                        trade_upload = gr.File(
-                            label="Upload Files",
-                            file_count="multiple",
-                            file_types=[".xlsx", ".xls", ".csv"]
-                        )
-                        trade_process_btn = gr.Button("🔄 Process Files", variant="secondary")
-                    with gr.Column(scale=2):
-                        trade_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
-                trade_results = gr.Textbox(label="Results", lines=2, interactive=False)
-                
-                # Corporate Agent Row
-                with gr.Row():
-                    gr.Markdown("#### 🏢 Corporate Agent")
-                with gr.Row():
-                    with gr.Column(scale=2):
-                        corporate_upload = gr.File(
-                            label="Upload Files",
-                            file_count="multiple",
-                            file_types=[".xlsx", ".xls", ".csv"]
-                        )
-                        corporate_process_btn = gr.Button("🔄 Process Files", variant="secondary")
-                    with gr.Column(scale=2):
-                        corporate_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
-                corporate_results = gr.Textbox(label="Results", lines=2, interactive=False)
-                
-                # Web Agent Row (Status only - web search focused)
-                with gr.Row():
-                    gr.Markdown("#### 🌐 Web Agent")
+            with gr.Column(scale=2):
+                gr.Markdown("*Web agent searches for real-time market data*")
+            with gr.Column(scale=2):
                 web_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
         
         # =============================
-        # BOTTOM SECTION: Orchestrator & System Summary
+        # ORCHESTRATOR SECTION
         # =============================
         gr.Markdown("---")
-        gr.Markdown("### 🎯 Orchestrator & System Control")
+        gr.Markdown("### 🎯 Orchestrator")
         
         with gr.Row():
             with gr.Column(scale=2):
-                gr.Markdown("#### 🎯 Orchestrator")
+                gr.Markdown("*Orchestrator analyzes all agent results*")
+            with gr.Column(scale=2):
                 orchestrator_upload = gr.File(
                     label="Upload Exception/Reconciliation Files",
                     file_count="multiple",
                     file_types=[".xlsx", ".xls", ".csv"]
                 )
                 orchestrator_process_btn = gr.Button("🔄 Process Files", variant="primary")
+            with gr.Column(scale=2):
                 orchestrator_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
-                orchestrator_results = gr.Textbox(label="Results", lines=3, interactive=False)
-            
+        
+        orchestrator_results = gr.Textbox(label="Results", lines=2, interactive=False)
+        
+        # =============================
+        # SYSTEM CONTROL SECTION
+        # =============================
+        gr.Markdown("---")
+        gr.Markdown("### 📊 System Control & Exception Management")
+        
+        with gr.Row():
             with gr.Column(scale=2):
                 gr.Markdown("#### 📊 System Summary")
                 summary_btn = gr.Button("📋 Get All Agent Status", variant="primary")
-                processing_summary = gr.Textbox(label="System Overview", lines=6, interactive=False)
+                processing_summary = gr.Textbox(label="System Overview", lines=4, interactive=False)
             
             with gr.Column(scale=2):
                 gr.Markdown("#### ⚠️ Exception Management")
@@ -864,20 +884,18 @@ def create_gradio_interface() -> gr.Blocks:
                 )
                 resolve_btn = gr.Button("✅ Resolve", variant="stop")
                 exceptions_btn = gr.Button("📄 List Exceptions")
+            
+            with gr.Column(scale=2):
+                gr.Markdown("#### 🧪 Quick Demo")
+                demo_normal_btn = gr.Button("🎯 Normal Trade", size="sm")
+                demo_failed_btn = gr.Button("💥 Failed Trade", size="sm") 
+                demo_corp_btn = gr.Button("🍎 Corporate Action", size="sm")
         
-        # Exception Results
+        # Results Row
         with gr.Row():
             resolution_output = gr.Textbox(label="Resolution Result", lines=2, interactive=False)
-            exceptions_output = gr.Textbox(label="Current Exceptions", lines=4, interactive=False)
-        
-        # Demo Section (simplified)
-        gr.Markdown("---")
-        gr.Markdown("### 🧪 Quick Demo")
-        with gr.Row():
-            demo_normal_btn = gr.Button("🎯 Normal Trade", size="sm")
-            demo_failed_btn = gr.Button("💥 Failed Trade", size="sm") 
-            demo_corp_btn = gr.Button("🍎 Corporate Action", size="sm")
-        demo_result = gr.Textbox(label="Demo Result", lines=3, interactive=False)
+            exceptions_output = gr.Textbox(label="Current Exceptions", lines=2, interactive=False)
+            demo_result = gr.Textbox(label="Demo Result", lines=2, interactive=False)
         
         # =============================
         # WIRE UP ALL CALLBACKS