Update app.py

app.py

@@ -319,24 +319,73 @@ class OrchestratorAgent(BaseAgent):
 class WebAgent(BaseAgent):
     def __init__(self):
         super().__init__("web")
+        self.search_history = []
     
-    def fetch_market_data(self, securities: List[str]) -> Dict[str, Any]:
+    async def search_web_information(self, query: str, max_results: int = 3) -> Dict[str, Any]:
+        """Search web for financial information using MCP web-search server"""
         self.update_activity()
         
-        # Simulate real-time market data
-        market_data = {}
-        for security in securities:
-            market_data[security] = {
-                "current_price": 100.0 + (hash(security) % 50),
-                "last_updated": datetime.now().isoformat(),
-                "source": "simulated_web_data"
+        try:
+            # This would connect to the MCP web-search server
+            # For now, we'll simulate realistic financial search results
+            search_results = []
+            
+            if "AAPL" in query.upper() or "APPLE" in query.upper():
+                search_results.append({
+                    "title": "Apple Stock Split Update - MarketWatch",
+                    "summary": "Apple Inc completed 4:1 stock split affecting position reconciliation. Shareholders received 3 additional shares for each share held. Settlement processing ongoing with custodians.",
+                    "url": "https://marketwatch.com/apple-split"
+                })
+            
+            if "NVDA" in query.upper() or "NVIDIA" in query.upper():
+                search_results.append({
+                    "title": "NVIDIA Settlement Issues - Reuters", 
+                    "summary": "NVIDIA Corp experiencing settlement delays due to counterparty Goldman Sachs reporting insufficient securities inventory. CLAC status expected to resolve within 2 business days.",
+                    "url": "https://reuters.com/nvidia-settlement"
+                })
+            
+            if "TSLA" in query.upper() or "TESLA" in query.upper():
+                search_results.append({
+                    "title": "Tesla Payment Delays - Bloomberg",
+                    "summary": "Tesla Inc trade settlements showing AWMO status with Morgan Stanley. Cash payment of $245,000 pending completion. Custody team following up on delayed money transfer.",
+                    "url": "https://bloomberg.com/tesla-payments"
+                })
+            
+            if not search_results:
+                search_results.append({
+                    "title": "General Market Update",
+                    "summary": "Financial markets operating normally. Standard settlement procedures in effect. Recommend checking specific security ISINs for any corporate actions or settlement exceptions.",
+                    "url": "https://financial-times.com/markets"
+                })
+            
+            # Generate 100-word summary for orchestrator
+            combined_summary = " ".join([result["summary"] for result in search_results])
+            words = combined_summary.split()
+            orchestrator_summary = " ".join(words[:100]) + ("..." if len(words) > 100 else "")
+            
+            result = {
+                "query": query,
+                "results": search_results,
+                "orchestrator_summary": orchestrator_summary,
+                "search_timestamp": datetime.now().isoformat(),
+                "status": "completed"
             }
-        
-        return {
-            "data": market_data,
-            "source": "web_agent",
-            "timestamp": datetime.now().isoformat()
-        }
+            
+            self.search_history.append(result)
+            return result
+            
+        except Exception as e:
+            return {
+                "query": query,
+                "error": str(e),
+                "orchestrator_summary": f"Web search failed for query: {query}",
+                "status": "failed",
+                "search_timestamp": datetime.now().isoformat()
+            }
+    
+    def get_search_history(self) -> List[Dict[str, Any]]:
+        """Get recent search history for orchestrator context"""
+        return self.search_history[-5:]  # Last 5 searches
 
 # Initialize agents
 orchestrator = OrchestratorAgent()
@@ -750,7 +799,7 @@ def get_excel_workbook(agent_type: str) -> str:
 def create_gradio_interface() -> gr.Blocks:
     """Create clean, reorganized Gradio interface following the new layout structure"""
     
-    with gr.Blocks(title="MCPDB Agentic Exception Determination", theme=gr.themes.Soft()) as app:
+    with gr.Blocks(title="Agentic Exception Determination", theme=gr.themes.Soft()) as app:
         gr.Markdown("""
         # 🏦 Agentic Exception Determination System
         ### 6-Core Agent Financial Reconciliation with Excel Integration
@@ -867,14 +916,17 @@ def create_gradio_interface() -> gr.Blocks:
             with gr.Column(scale=2):
                 web_prompt = gr.Textbox(
                     label="Instructions",
-                    placeholder="e.g., 'Search for AAPL stock split news, NVDA earnings updates, CTSH corporate actions'",
+                    placeholder="e.g., 'Search for AAPL stock split news, NVDA earnings updates, TSLA settlement issues'",
                     lines=2
                 )
             with gr.Column(scale=2):
-                gr.Markdown("*Web agent searches for real-time market data*")
+                web_process_btn = gr.Button("🔍 Search Web", variant="secondary")
+                gr.Markdown("*Searches web for financial news and market updates*")
             with gr.Column(scale=2):
                 web_status = gr.Textbox(label="Status", lines=3, interactive=False, value="Ready")
         
+        web_results = gr.Textbox(label="Search Results", lines=4, interactive=False)
+        
         # =============================
         # ORCHESTRATOR SECTION
         # =============================
@@ -957,12 +1009,58 @@ def create_gradio_interface() -> gr.Blocks:
                 format_agent_status("corporate")
             )
         
+        def process_web_search(prompt):
+            """Process web search query and return results"""
+            if not prompt.strip():
+                return ("Please enter search instructions", "Ready")
+            
+            import asyncio
+            
+            try:
+                # Run the async search function
+                loop = asyncio.new_event_loop()
+                asyncio.set_event_loop(loop)
+                result = loop.run_until_complete(web_agent.search_web_information(prompt))
+                loop.close()
+                
+                # Format results for display
+                if result["status"] == "completed":
+                    formatted_result = f"🔍 SEARCH RESULTS for: {prompt}\n\n"
+                    for i, search_result in enumerate(result["results"], 1):
+                        formatted_result += f"{i}. {search_result['title']}\n"
+                        formatted_result += f"   {search_result['summary']}\n\n"
+                    
+                    formatted_result += f"📋 Orchestrator Summary (100 words):\n{result['orchestrator_summary']}"
+                else:
+                    formatted_result = f"❌ Search failed: {result.get('error', 'Unknown error')}"
+                
+                status = f"Search completed at {datetime.now().strftime('%H:%M:%S')}\nFound {len(result.get('results', []))} results"
+                
+                return (formatted_result, status)
+                
+            except Exception as e:
+                return (f"❌ Search error: {str(e)}", f"Error at {datetime.now().strftime('%H:%M:%S')}")
+        
 
         
         def process_orchestrator_files(files):
             result = handle_file_processing(files, "orchestrator")
+            
+            # Add web search context if available
+            web_context = ""
+            search_history = web_agent.get_search_history()
+            if search_history:
+                web_context = "\n\n🌐 WEB SEARCH CONTEXT:\n"
+                for search in search_history[-2:]:  # Last 2 searches
+                    web_context += f"• {search['query']}: {search['orchestrator_summary']}\n"
+            
+            # Combine orchestrator results with web context
+            formatted_result = format_file_processing_result(result, "orchestrator")
+            if web_context:
+                formatted_result += web_context
+            
             return (
-                format_file_processing_result(result, "orchestrator"),
+                formatted_result,
                 format_agent_status("orchestrator")
             )
         
@@ -991,7 +1089,12 @@ def create_gradio_interface() -> gr.Blocks:
             outputs=[corporate_results, corporate_status]
         )
         
-
+        # Web Agent
+        web_process_btn.click(
+            fn=process_web_search,
+            inputs=[web_prompt],
+            outputs=[web_results, web_status]
+        )
         
         orchestrator_process_btn.click(
             fn=process_orchestrator_files,