Update app.py

app.py

@@ -447,20 +447,35 @@ def list_exceptions() -> str:
     exceptions = list(store.exceptions.values())
     return json.dumps(exceptions, indent=2, default=str)
 
-def resolve_exception_manually(
-    exception_id: str,
-    resolution_action: str,
-    justification: str
-) -> str:
-    """Resolve a specific exception"""
-    resolution = {
-        "action": resolution_action,
-        "justification": justification,
-        "resolved_by": "manual"
+def process_exception_decision(exception_id: str, action: str) -> str:
+    """Process user decision on identified exception"""
+    valid_exceptions = ["EXC_001", "EXC_002", "EXC_003", "EXC_004", "EXC_005"]
+    
+    if exception_id not in valid_exceptions:
+        return f"❌ Invalid Exception ID. Please use one of: {', '.join(valid_exceptions)}"
+    
+    exception_details = {
+        "EXC_001": "Apple stock split position discrepancy",
+        "EXC_002": "NVIDIA settlement failure - counterparty issue", 
+        "EXC_003": "Tesla pending cash settlement",
+        "EXC_004": "Microsoft tender offer completed",
+        "EXC_005": "Alphabet bond maturity completed"
     }
     
-    result = orchestrator.resolve_exception(exception_id, resolution)
-    return json.dumps(result, indent=2, default=str)
+    if action == "APPROVE":
+        result = f"✅ {exception_id} APPROVED\n"
+        result += f"Issue: {exception_details[exception_id]}\n"
+        result += f"Action: Resolution approved and forwarded to appropriate team\n"
+        result += f"Status: Processing initiated"
+    elif action == "REJECT":
+        result = f"❌ {exception_id} REJECTED\n" 
+        result += f"Issue: {exception_details[exception_id]}\n"
+        result += f"Action: Resolution rejected - requires manual review\n"
+        result += f"Status: Escalated for further analysis"
+    else:
+        result = f"⚠️ Invalid action: {action}"
+    
+    return result
 
 def get_dashboard() -> str:
     """Get system dashboard data"""
@@ -596,29 +611,48 @@ def format_file_processing_result(result_dict: dict, agent_type: str) -> str:
     
     return "\n".join(result)
 
-def format_exceptions_list(exceptions_json: str) -> str:
-    """Format exceptions list in user-friendly way"""
-    try:
-        exceptions = json.loads(exceptions_json)
-    except:
-        return "No exceptions found"
-    
-    if not exceptions:
-        return "✅ No open exceptions - system running smoothly!"
+def get_identified_exceptions() -> str:
+    """Get the 5 identified exceptions requiring user approval/rejection"""
+    exceptions = [
+        {
+            "id": "EXC_001",
+            "issue": "Apple stock split position discrepancy (-9,000 shares)",
+            "suggested_resolution": "Adjust position to reflect 4:1 split completion",
+            "status": "Pending Approval"
+        },
+        {
+            "id": "EXC_002", 
+            "issue": "NVIDIA settlement failure - counterparty insufficient securities",
+            "suggested_resolution": "Contact Goldman Sachs to resolve CLAC status",
+            "status": "Pending Approval"
+        },
+        {
+            "id": "EXC_003",
+            "issue": "Tesla pending cash settlement ($245,000)",
+            "suggested_resolution": "Follow up with Morgan Stanley for AWMO completion",
+            "status": "Pending Approval"
+        },
+        {
+            "id": "EXC_004",
+            "issue": "Microsoft tender offer processed successfully",
+            "suggested_resolution": "No action required - mark as resolved",
+            "status": "Pending Approval"
+        },
+        {
+            "id": "EXC_005",
+            "issue": "Alphabet bond maturity completed correctly",
+            "suggested_resolution": "No action required - mark as resolved",
+            "status": "Pending Approval"
+        }
+    ]
     
-    result = []
-    result.append("⚠️ OPEN EXCEPTIONS")
-    result.append("=" * 30)
-    
-    for i, exc in enumerate(exceptions, 1):
-        severity_emoji = {"high": "🔴", "medium": "🟡", "low": "🟢"}.get(exc.get("severity", "low"), "⚪")
-        result.append(f"{severity_emoji} Exception #{i}")
-        result.append(f"   ID: {exc.get('id', 'N/A')}")
-        result.append(f"   Type: {exc.get('type', 'Unknown')}")
-        result.append(f"   Severity: {exc.get('severity', 'Unknown').upper()}")
-        result.append("")
+    result = "📋 IDENTIFIED EXCEPTIONS REQUIRING REVIEW:\n\n"
+    for exc in exceptions:
+        result += f"🔶 {exc['id']}: {exc['issue']}\n"
+        result += f"   💡 Suggested: {exc['suggested_resolution']}\n"
+        result += f"   📊 Status: {exc['status']}\n\n"
     
-    return "\n".join(result)
+    return result
 
 def format_demo_result(demo_json: str) -> str:
     """Format demo results in user-friendly way"""
@@ -869,33 +903,27 @@ def create_gradio_interface() -> gr.Blocks:
         gr.Markdown("### 📊 System Control & Exception Management")
         
         with gr.Row():
-            with gr.Column(scale=2):
+            with gr.Column(scale=1):
                 gr.Markdown("#### 📊 System Summary")
                 summary_btn = gr.Button("📋 Get All Agent Status", variant="primary")
                 processing_summary = gr.Textbox(label="System Overview", lines=4, interactive=False)
             
             with gr.Column(scale=2):
                 gr.Markdown("#### ⚠️ Exception Management")
-                exception_id_input = gr.Textbox(label="Exception ID", placeholder="EXC_001")
-                resolution_action = gr.Dropdown(
-                    choices=["APPROVE", "REJECT", "REVIEW", "ESCALATE"],
-                    value="APPROVE",
-                    label="Action"
-                )
-                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")
-        
-        # Results Row
-        with gr.Row():
-            resolution_output = gr.Textbox(label="Resolution Result", lines=2, interactive=False)
-            exceptions_output = gr.Textbox(label="Current Exceptions", lines=2, interactive=False)
-            demo_result = gr.Textbox(label="Demo Result", lines=2, interactive=False)
+                exceptions_btn = gr.Button("📄 Show Identified Issues", variant="secondary")
+                exceptions_output = gr.Textbox(label="Identified Issues", lines=8, interactive=False)
+                
+                with gr.Row():
+                    exception_id_input = gr.Textbox(label="Exception ID", placeholder="EXC_001", scale=1)
+                    resolution_action = gr.Dropdown(
+                        choices=["APPROVE", "REJECT"],
+                        value="APPROVE",
+                        label="Action",
+                        scale=1
+                    )
+                    resolve_btn = gr.Button("✅ Process", variant="stop", scale=1)
+                
+                resolution_output = gr.Textbox(label="Resolution Result", lines=3, interactive=False)
         
         # =============================
         # WIRE UP ALL CALLBACKS
@@ -975,18 +1003,14 @@ def create_gradio_interface() -> gr.Blocks:
         # System controls
         summary_btn.click(fn=get_all_agents_summary, outputs=processing_summary)
         
-        exceptions_btn.click(fn=lambda: format_exceptions_list(list_exceptions()), outputs=exceptions_output)
+        exceptions_btn.click(fn=get_identified_exceptions, outputs=exceptions_output)
         
         resolve_btn.click(
-            fn=lambda e_id, act: format_file_processing_result(resolve_exception_manually(e_id, act, "Manual resolution"), "resolution"),
+            fn=process_exception_decision,
             inputs=[exception_id_input, resolution_action],
             outputs=resolution_output
         )
         
-        # Demo buttons
-        demo_normal_btn.click(fn=lambda: format_demo_result(demo_normal_settlement()), outputs=demo_result)
-        demo_failed_btn.click(fn=lambda: format_demo_result(demo_failed_trade()), outputs=demo_result)
-        demo_corp_btn.click(fn=lambda: format_demo_result(demo_corporate_action()), outputs=demo_result)
         
     return app