54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
import sqlite3
|
|
import os
|
|
import json
|
|
import requests
|
|
import argparse
|
|
from dotenv import load_dotenv
|
|
|
|
# --- Configuration & Setup ---
|
|
load_dotenv(dotenv_path="/home/node/clawd/.env", override=True)
|
|
DB_PATH = "/home/node/clawd/repos/brancheneinstufung2/company_explorer_local.db"
|
|
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|
|
|
|
class SniperGenerator:
|
|
def __init__(self, db_path=DB_PATH):
|
|
self.db_path = db_path
|
|
|
|
def get_lead_data(self, company_id):
|
|
conn = sqlite3.connect(self.db_path)
|
|
conn.row_factory = sqlite3.Row
|
|
cursor = conn.cursor()
|
|
cursor.execute("SELECT * FROM companies WHERE id = ?", (company_id,))
|
|
company_data = cursor.fetchone()
|
|
conn.close()
|
|
return dict(company_data) if company_data else None
|
|
|
|
def generate_copy(self, company_id, target_role="Wirtschaftl. Entscheider"):
|
|
lead_data = self.get_lead_data(company_id)
|
|
if not lead_data: return "Error: Company data not found."
|
|
|
|
prompt = f"""
|
|
Du bist ein Weltklasse B2B-Stratege. Erstelle eine 3-Satz-E-Mail-Einleitung.
|
|
Firma: {lead_data.get('name')}
|
|
Branche: {lead_data.get('industry_ai')}
|
|
Rolle: {target_role}
|
|
"""
|
|
|
|
headers = {"Content-Type": "application/json"}
|
|
payload = {"contents": [{"parts": [{"text": prompt}]}]}
|
|
url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={GEMINI_API_KEY}"
|
|
|
|
try:
|
|
resp = requests.post(url, headers=headers, json=payload, timeout=20)
|
|
resp.raise_for_status()
|
|
return resp.json()['candidates'][0]['content']['parts'][0]['text'].strip()
|
|
except Exception as e:
|
|
return f"Error: {e}"
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("company_id", type=int)
|
|
args = parser.parse_args()
|
|
sniper = SniperGenerator()
|
|
print(sniper.generate_copy(args.company_id))
|