[31388f42] Fix NameError in reply generator and implement free-mail quality detection

lead-engine/db.py

@@ -55,9 +55,15 @@ def insert_lead(lead_data):
         'area': lead_data.get('area'),
         'purpose': lead_data.get('purpose'),
         'zip': lead_data.get('zip'),
-        'city': lead_data.get('city')
+        'city': lead_data.get('city'),
+        'role': lead_data.get('role'),
+        'is_free_mail': lead_data.get('is_free_mail', False),
+        'is_low_quality': lead_data.get('is_low_quality', False)
     }
     
+    # Use provided received_at or default to now
+    received_at = lead_data.get('received_at') or datetime.now()
+    
     conn = sqlite3.connect(DB_PATH)
     c = conn.cursor()
     try:
@@ -66,7 +72,7 @@ def insert_lead(lead_data):
             VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
         ''', (
             lead_data.get('id'),
-            datetime.now(),
+            received_at,
             lead_data.get('company'),
             lead_data.get('contact'),
             lead_data.get('email'),
@@ -111,3 +117,11 @@ def update_lead_status(lead_id, status, response_draft=None):
         c.execute('UPDATE leads SET status = ? WHERE id = ?', (status, lead_id))
     conn.commit()
     conn.close()
+
+def reset_lead(lead_id):
+    """Resets a lead to 'new' status and clears enrichment data."""
+    conn = sqlite3.connect(DB_PATH)
+    c = conn.cursor()
+    c.execute('UPDATE leads SET status = "new", enrichment_data = NULL WHERE id = ?', (lead_id,))
+    conn.commit()
+    conn.close()