Integrate DataEnrichmentManager into app.py with new UI tab

app.py

@@ -5,6 +5,7 @@ import json
 from typing import List, Dict, Any, Optional
 from pathlib import Path
 from linkedin_data import fetch_linkedin_profile, get_cache_stats
+from data_enrichment import DataEnrichmentManager
 
 # --- Data Storage Configuration ---
 DATA_DIR = Path("/data")
@@ -33,6 +34,10 @@ def save_user_data(username: str, data_type: str, data: Any):
     with open(data_file, 'w') as f:
         json.dump(data, f, indent=2)
 
+        def get_enrichment_manager(username: str) -> DataEnrichmentManager:
+    """Get or create DataEnrichmentManager for a user"""
+    return DataEnrichmentManager(username)
+
 # --- Consent/Event Audit Panel ---
 def consent_audit_panel(consent_events: List[Dict[str, Any]], profile: Optional[gr.OAuthProfile]) -> str:
     """
@@ -219,6 +224,11 @@ with gr.Blocks(title="StrategyStack UI") as demo:
             audit_output = gr.Markdown()
             audit_refresh = gr.Button("Refresh Data")
             user_profile_4 = gr.State()
+
+                    with gr.Tab("Data Enrichment"):
+            enrichment_output = gr.Markdown()
+            enrichment_refresh = gr.Button("Refresh Data")
+            user_profile_5 = gr.State()
     
     # Load initial data
     def load_consent_data(profile: Optional[gr.OAuthProfile] = None):
@@ -232,17 +242,67 @@ with gr.Blocks(title="StrategyStack UI") as demo:
     
     def load_audit_data(profile: Optional[gr.OAuthProfile] = None):
         return audit_trail_widget(profile)
+
+            def load_enrichment_data(profile: Optional[gr.OAuthProfile] = None):
+        if profile is None:
+            return "# 🔒 Authentication Required\n\nPlease sign in with your Hugging Face account to access data enrichment features."
+        
+        username = profile.username
+        manager = get_enrichment_manager(username)
+        stats = manager.get_enrichment_stats()
+        
+        enrichment_md = f"# 📊 Data Enrichment Dashboard\n\n**User:** {username}\n\n"
+        enrichment_md += "## Enrichment Statistics\n\n"
+        enrichment_md += f"- **Total API Calls:** {stats.get('total_api_calls', 0)}\n"
+        enrichment_md += f"- **Cache Hits:** {stats.get('cache_hits', 0)}\n"
+        enrichment_md += f"- **Cache Misses:** {stats.get('cache_misses', 0)}\n"
+        enrichment_md += f"- **Total Enrichments:** {stats.get('total_enrichments', 0)}\n"
+        enrichment_md += f"- **Successful Enrichments:** {stats.get('successful_enrichments', 0)}\n"
+        enrichment_md += f"- **Failed Enrichments:** {stats.get('failed_enrichments', 0)}\n\n"
+        
+        # Calculate success rate
+        if stats.get('total_enrichments', 0) > 0:
+            success_rate = (stats.get('successful_enrichments', 0) / stats['total_enrichments']) * 100
+            enrichment_md += f"- **Success Rate:** {success_rate:.1f}%\n\n"
+        
+        enrichment_md += "## Available APIs\n\n"
+        enrichment_md += "- ✅ LinkedIn Profile Enrichment (4 providers with fallback)\n"
+        enrichment_md += "- ✅ Company Profile Enrichment\n"
+        enrichment_md += "- ✅ Apollo.io Person Enrichment\n"
+        enrichment_md += "- ✅ Twitter User Data\n"
+        enrichment_md += "- ✅ Facebook Data\n"
+        enrichment_md += "- ✅ Google Search\n"
+        enrichment_md += "- ✅ Mediafy Enrichment\n"
+        enrichment_md += "- ✅ Local Business Data (LetScrape)\n"
+        enrichment_md += "- ✅ Website Contacts Scraper (LetScrape)\n\n"
+        
+        enrichment_md += "## Features\n\n"
+        enrichment_md += "- 🔒 **Per-User Data Isolation**: Your enrichment data is stored separately\n"
+        enrichment_md += "- ⚡ **Intelligent Caching**: 30-day TTL to minimize API costs\n"
+        enrichment_md += "- 🔄 **Automatic Fallback**: Multiple providers ensure high availability\n"
+        enrichment_md += "- 📈 **Real-time Statistics**: Track API usage and success rates\n\n"
+        
+        enrichment_md += "## Usage\n\n"
+        enrichment_md += "To use data enrichment in your workflows:\n"
+        enrichment_md += "1. Add contact data with fields like `linkedin_url`, `email`, `domain`\n"
+        enrichment_md += "2. The system automatically enriches contacts using all available APIs\n"
+        enrichment_md += "3. Enriched data is cached for 30 days to optimize performance\n"
+        enrichment_md += "4. View statistics and monitor API usage from this dashboard\n\n"
+        
+        return gr.Markdown(enrichment_md)
     
     # Set up event handlers
     demo.load(load_consent_data, inputs=None, outputs=consent_output)
     demo.load(load_branching_data, inputs=None, outputs=branching_output)
     demo.load(load_diagnostics_data, inputs=None, outputs=diagnostics_output)
     demo.load(load_audit_data, inputs=None, outputs=audit_output)
+        demo.load(load_enrichment_data, inputs=None, outputs=enrichment_output)
     
     consent_refresh.click(load_consent_data, inputs=None, outputs=consent_output)
     branching_refresh.click(load_branching_data, inputs=None, outputs=branching_output)
     diagnostics_refresh.click(load_diagnostics_data, inputs=None, outputs=diagnostics_output)
     audit_refresh.click(load_audit_data, inputs=None, outputs=audit_output)
+    enrichment_refresh.click(load_enrichment_data, inputs=None, outputs=enrichment_output)
 
 if __name__ == "__main__":
     demo.launch()