Initial MVP Commit: Ingest, Enrich, Dashboard
This commit is contained in:
9
Dockerfile
Normal file
9
Dockerfile
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
FROM python:3.9-slim
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
COPY . .
|
||||||
|
|
||||||
|
RUN pip install streamlit pandas
|
||||||
|
|
||||||
|
CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]
|
||||||
BIN
__pycache__/db.cpython-311.pyc
Normal file
BIN
__pycache__/db.cpython-311.pyc
Normal file
Binary file not shown.
55
app.py
Normal file
55
app.py
Normal file
@@ -0,0 +1,55 @@
|
|||||||
|
import streamlit as st
|
||||||
|
import pandas as pd
|
||||||
|
from db import get_leads, init_db
|
||||||
|
import json
|
||||||
|
|
||||||
|
st.set_page_config(page_title="TradingTwins Lead Engine", layout="wide")
|
||||||
|
|
||||||
|
st.title("🚀 Lead Engine: TradingTwins")
|
||||||
|
|
||||||
|
# Metrics
|
||||||
|
leads = get_leads()
|
||||||
|
df = pd.DataFrame(leads)
|
||||||
|
|
||||||
|
if not df.empty:
|
||||||
|
col1, col2, col3 = st.columns(3)
|
||||||
|
col1.metric("Total Leads", len(df))
|
||||||
|
col2.metric("New Today", len(df[df['status'] == 'new']))
|
||||||
|
col3.metric("Action Required", len(df[df['status'] == 'enriched']))
|
||||||
|
|
||||||
|
st.subheader("Latest Inquiries")
|
||||||
|
|
||||||
|
for index, row in df.iterrows():
|
||||||
|
with st.expander(f"{row['company_name']} - {row['status']}"):
|
||||||
|
c1, c2 = st.columns(2)
|
||||||
|
c1.write(f"**Contact:** {row['contact_name']}")
|
||||||
|
c1.write(f"**Email:** {row['email']}")
|
||||||
|
|
||||||
|
enrichment = json.loads(row['enrichment_data']) if row['enrichment_data'] else {}
|
||||||
|
|
||||||
|
if enrichment:
|
||||||
|
c2.success(f"Score: {enrichment.get('score')}")
|
||||||
|
c2.write(f"**Vertical:** {enrichment.get('vertical')}")
|
||||||
|
c2.info(f"💡 {enrichment.get('recommendation')}")
|
||||||
|
|
||||||
|
if st.button(f"Generate Response for {row['id']}"):
|
||||||
|
st.write("Generating draft... (Simulated)")
|
||||||
|
# Here we would call the LLM
|
||||||
|
draft = f"Hallo {row['contact_name']},\n\ndanke für Ihre Anfrage..."
|
||||||
|
st.text_area("Draft", draft, height=200)
|
||||||
|
|
||||||
|
else:
|
||||||
|
st.info("No leads found. Waiting for ingest...")
|
||||||
|
|
||||||
|
if st.sidebar.button("Run Ingest (Mock)"):
|
||||||
|
from ingest import ingest_mock_leads
|
||||||
|
init_db()
|
||||||
|
count = ingest_mock_leads()
|
||||||
|
st.sidebar.success(f"Ingested {count} leads.")
|
||||||
|
st.rerun()
|
||||||
|
|
||||||
|
if st.sidebar.button("Run Enrichment"):
|
||||||
|
from enrich import run_enrichment
|
||||||
|
run_enrichment()
|
||||||
|
st.sidebar.success("Enrichment complete.")
|
||||||
|
st.rerun()
|
||||||
BIN
data/leads.db
Normal file
BIN
data/leads.db
Normal file
Binary file not shown.
70
db.py
Normal file
70
db.py
Normal file
@@ -0,0 +1,70 @@
|
|||||||
|
import sqlite3
|
||||||
|
import json
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
DB_PATH = 'lead-engine/data/leads.db'
|
||||||
|
|
||||||
|
def init_db():
|
||||||
|
conn = sqlite3.connect(DB_PATH)
|
||||||
|
c = conn.cursor()
|
||||||
|
c.execute('''
|
||||||
|
CREATE TABLE IF NOT EXISTS leads (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
source_id TEXT UNIQUE,
|
||||||
|
received_at TIMESTAMP,
|
||||||
|
company_name TEXT,
|
||||||
|
contact_name TEXT,
|
||||||
|
email TEXT,
|
||||||
|
phone TEXT,
|
||||||
|
raw_body TEXT,
|
||||||
|
enrichment_data TEXT,
|
||||||
|
status TEXT DEFAULT 'new',
|
||||||
|
response_draft TEXT,
|
||||||
|
sent_at TIMESTAMP
|
||||||
|
)
|
||||||
|
''')
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
def insert_lead(lead_data):
|
||||||
|
conn = sqlite3.connect(DB_PATH)
|
||||||
|
c = conn.cursor()
|
||||||
|
try:
|
||||||
|
c.execute('''
|
||||||
|
INSERT INTO leads (source_id, received_at, company_name, contact_name, email, phone, raw_body, status)
|
||||||
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||||
|
''', (
|
||||||
|
lead_data.get('id'),
|
||||||
|
datetime.now(),
|
||||||
|
lead_data.get('company'),
|
||||||
|
lead_data.get('contact'),
|
||||||
|
lead_data.get('email'),
|
||||||
|
lead_data.get('phone'),
|
||||||
|
lead_data.get('raw_body'),
|
||||||
|
'new'
|
||||||
|
))
|
||||||
|
conn.commit()
|
||||||
|
return True
|
||||||
|
except sqlite3.IntegrityError:
|
||||||
|
return False
|
||||||
|
finally:
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
def get_leads():
|
||||||
|
conn = sqlite3.connect(DB_PATH)
|
||||||
|
conn.row_factory = sqlite3.Row
|
||||||
|
c = conn.cursor()
|
||||||
|
c.execute('SELECT * FROM leads ORDER BY received_at DESC')
|
||||||
|
rows = c.fetchall()
|
||||||
|
conn.close()
|
||||||
|
return [dict(row) for row in rows]
|
||||||
|
|
||||||
|
def update_lead_status(lead_id, status, response_draft=None):
|
||||||
|
conn = sqlite3.connect(DB_PATH)
|
||||||
|
c = conn.cursor()
|
||||||
|
if response_draft:
|
||||||
|
c.execute('UPDATE leads SET status = ?, response_draft = ? WHERE id = ?', (status, response_draft, lead_id))
|
||||||
|
else:
|
||||||
|
c.execute('UPDATE leads SET status = ? WHERE id = ?', (status, lead_id))
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
43
enrich.py
Normal file
43
enrich.py
Normal file
@@ -0,0 +1,43 @@
|
|||||||
|
import json
|
||||||
|
from db import update_lead_status, get_leads
|
||||||
|
import sqlite3
|
||||||
|
|
||||||
|
def enrich_lead(lead):
|
||||||
|
# Mock Enrichment Logic
|
||||||
|
# In production, this would call the Company Explorer API
|
||||||
|
|
||||||
|
enrichment = {
|
||||||
|
'vertical': 'Unknown',
|
||||||
|
'score': 0,
|
||||||
|
'recommendation': 'Manual Review'
|
||||||
|
}
|
||||||
|
|
||||||
|
company = lead['company_name'].lower()
|
||||||
|
raw = lead['raw_body']
|
||||||
|
|
||||||
|
if 'küche' in company or 'einbau' in company:
|
||||||
|
enrichment['vertical'] = 'Manufacturing / Woodworking'
|
||||||
|
enrichment['score'] = 85
|
||||||
|
enrichment['recommendation'] = 'High Priority - Pitch Dust Control'
|
||||||
|
|
||||||
|
if 'shop' in company or 'roller' in company:
|
||||||
|
enrichment['vertical'] = 'Retail / Automotive'
|
||||||
|
enrichment['score'] = 60
|
||||||
|
enrichment['recommendation'] = 'Medium Priority - Pitch Showroom Cleanliness'
|
||||||
|
|
||||||
|
# Update DB
|
||||||
|
conn = sqlite3.connect('lead-engine/data/leads.db')
|
||||||
|
c = conn.cursor()
|
||||||
|
c.execute('UPDATE leads SET enrichment_data = ?, status = ? WHERE id = ?',
|
||||||
|
(json.dumps(enrichment), 'enriched', lead['id']))
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
def run_enrichment():
|
||||||
|
leads = get_leads()
|
||||||
|
for lead in leads:
|
||||||
|
if lead['status'] == 'new':
|
||||||
|
enrich_lead(lead)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
run_enrichment()
|
||||||
75
ingest.py
Normal file
75
ingest.py
Normal file
@@ -0,0 +1,75 @@
|
|||||||
|
import re
|
||||||
|
from db import insert_lead
|
||||||
|
|
||||||
|
def parse_tradingtwins_email(body):
|
||||||
|
data = {}
|
||||||
|
|
||||||
|
# Simple regex extraction based on the email format
|
||||||
|
patterns = {
|
||||||
|
'id': r'Lead-ID:\s*(\d+)',
|
||||||
|
'company': r'Firma:\s*(.+)',
|
||||||
|
'contact_first': r'Vorname:\s*(.+)',
|
||||||
|
'contact_last': r'Nachname:\s*(.+)',
|
||||||
|
'email': r'E-Mail:\s*([^\s<]+)',
|
||||||
|
'phone': r'Rufnummer:\s*([^\n]+)',
|
||||||
|
'area': r'Reinigungs-Flche:\s*([^\n]+)',
|
||||||
|
'purpose': r'Einsatzzweck:\s*([^\n]+)'
|
||||||
|
}
|
||||||
|
|
||||||
|
for key, pattern in patterns.items():
|
||||||
|
match = re.search(pattern, body)
|
||||||
|
if match:
|
||||||
|
data[key] = match.group(1).strip()
|
||||||
|
|
||||||
|
# Combine names
|
||||||
|
if 'contact_first' in data and 'contact_last' in data:
|
||||||
|
data['contact'] = f"{data['contact_first']} {data['contact_last']}"
|
||||||
|
|
||||||
|
data['raw_body'] = body
|
||||||
|
return data
|
||||||
|
|
||||||
|
def ingest_mock_leads():
|
||||||
|
# Mock data from the session context
|
||||||
|
leads = [
|
||||||
|
{
|
||||||
|
'id': '2397256',
|
||||||
|
'company': 'pronorm Einbauküchen GmbH',
|
||||||
|
'contact': 'Jakob Funk',
|
||||||
|
'email': 'jakob.funk@pronorm.de',
|
||||||
|
'phone': '+49 5733 979175',
|
||||||
|
'raw_body': """
|
||||||
|
Lead-ID: 2397256
|
||||||
|
Firma: pronorm Einbauküchen GmbH
|
||||||
|
Vorname: Jakob
|
||||||
|
Nachname: Funk
|
||||||
|
Reinigungs-Flche: 1.001 - 10.000 qm
|
||||||
|
Einsatzzweck: Reinigung von Böden
|
||||||
|
"""
|
||||||
|
},
|
||||||
|
{
|
||||||
|
'id': '2414364',
|
||||||
|
'company': 'Quad & Rollershop Schwabmünchen GmbH',
|
||||||
|
'contact': 'Manfred Bihler',
|
||||||
|
'email': 'Rollershop.Schwabmuenchen@web.de',
|
||||||
|
'phone': '+49 8232 905246',
|
||||||
|
'raw_body': """
|
||||||
|
Lead-ID: 2414364
|
||||||
|
Firma: Quad & Rollershop Schwabmünchen GmbH
|
||||||
|
Vorname: Manfred
|
||||||
|
Nachname: Bihler
|
||||||
|
Reinigungs-Flche: 301 - 1.000 qm
|
||||||
|
Einsatzzweck: Reinigung von Böden
|
||||||
|
"""
|
||||||
|
}
|
||||||
|
]
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
for lead in leads:
|
||||||
|
if insert_lead(lead):
|
||||||
|
count += 1
|
||||||
|
return count
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
from db import init_db
|
||||||
|
init_db()
|
||||||
|
print(f"Ingested {ingest_mock_leads()} new leads.")
|
||||||
Reference in New Issue
Block a user