From ee7350d727c8ef7e9ac472e29a28048cb31ba4da Mon Sep 17 00:00:00 2001
From: Jarvis <jarvis@moltbot.ai>
Date: Fri, 30 Jan 2026 12:55:09 +0000
Subject: [PATCH] Feat: Add real API connector with mock fallback

---
 enrich.py | 82 +++++++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 73 insertions(+), 9 deletions(-)

diff --git a/enrich.py b/enrich.py
index c4378f1..43b1d64 100644
--- a/enrich.py
+++ b/enrich.py
@@ -1,29 +1,90 @@
 import json
+import requests
+import os
+from requests.auth import HTTPBasicAuth
 from db import update_lead_status, get_leads, DB_PATH
 import sqlite3
 
-def enrich_lead(lead):
-    # Mock Enrichment Logic
-    # In production, this would call the Company Explorer API
-    
+# Config
+CE_API_URL = os.getenv("CE_API_URL", "http://192.168.178.6:8090/ce/api")
+CE_API_USER = os.getenv("CE_API_USER", "admin")
+CE_API_PASS = os.getenv("CE_API_PASS", "")
+
+def query_company_explorer(company_name):
+    """
+    Fragt die Company Explorer API nach Details zur Firma.
+    Gibt None zurück, wenn die API nicht erreichbar ist.
+    """
+    if not CE_API_PASS:
+        print("⚠️ Warning: No CE_API_PASS set. Skipping API call.")
+        return None
+
+    try:
+        # Beispiel-Endpoint: Suche nach Firma
+        # Wir nehmen an, es gibt einen /search Endpoint
+        response = requests.get(
+            f"{CE_API_URL}/companies/search", 
+            params={"q": company_name},
+            auth=HTTPBasicAuth(CE_API_USER, CE_API_PASS),
+            timeout=5
+        )
+        
+        if response.status_code == 200:
+            results = response.json()
+            if results and isinstance(results, list) and len(results) > 0:
+                return results[0] # Best Match
+        elif response.status_code == 401:
+            print("❌ API Auth failed (401). Check password.")
+            
+    except Exception as e:
+        print(f"❌ API Error: {e}")
+        
+    return None
+
+def get_mock_enrichment(company_name, raw_body):
+    """Fallback Logic wenn API nicht verfügbar"""
     enrichment = {
         'vertical': 'Unknown',
         'score': 0,
-        'recommendation': 'Manual Review'
+        'recommendation': 'Manual Review (Mock Data)'
     }
     
-    company = lead['company_name'].lower()
-    raw = lead['raw_body']
+    name_lower = company_name.lower()
     
-    if 'küche' in company or 'einbau' in company:
+    if 'küche' in name_lower or 'einbau' in name_lower:
         enrichment['vertical'] = 'Manufacturing / Woodworking'
         enrichment['score'] = 85
         enrichment['recommendation'] = 'High Priority - Pitch Dust Control'
+        enrichment['area_estimate'] = '> 1000m²'
         
-    if 'shop' in company or 'roller' in company:
+    if 'shop' in name_lower or 'roller' in name_lower:
         enrichment['vertical'] = 'Retail / Automotive'
         enrichment['score'] = 60
         enrichment['recommendation'] = 'Medium Priority - Pitch Showroom Cleanliness'
+        enrichment['area_estimate'] = '300-500m²'
+
+    return enrichment
+
+def enrich_lead(lead):
+    print(f"🔍 Enriching: {lead['company_name']}...")
+    
+    # 1. Versuch: Echte API
+    api_data = query_company_explorer(lead['company_name'])
+    
+    enrichment = {}
+    
+    if api_data:
+        print("✅ API Hit!")
+        # Mapping der API-Daten auf unser Format
+        enrichment = {
+            'vertical': api_data.get('industry', 'Unknown'),
+            'score': api_data.get('score', 50), # Annahme: API liefert Score
+            'recommendation': 'Data from Company Explorer',
+            'details': api_data
+        }
+    else:
+        print("⚠️ API Miss/Fail -> Using Mock Fallback")
+        enrichment = get_mock_enrichment(lead['company_name'], lead['raw_body'])
 
     # Update DB
     conn = sqlite3.connect(DB_PATH)
@@ -35,9 +96,12 @@ def enrich_lead(lead):
     
 def run_enrichment():
     leads = get_leads()
+    count = 0
     for lead in leads:
         if lead['status'] == 'new':
             enrich_lead(lead)
+            count += 1
+    print(f"Done. Enriched {count} leads.")
 
 if __name__ == "__main__":
     run_enrichment()