100 lines
4.2 KiB
Python
100 lines
4.2 KiB
Python
import json
|
|
import time
|
|
import os
|
|
import sys
|
|
|
|
# Ensure we can import from lead-engine
|
|
sys.path.append(os.path.join(os.path.dirname(__file__), 'lead-engine'))
|
|
try:
|
|
from trading_twins_ingest import process_leads
|
|
except ImportError:
|
|
print("Warning: Could not import trading_twins_ingest from lead-engine. Email ingestion disabled.")
|
|
process_leads = None
|
|
|
|
from company_explorer_connector import handle_company_workflow
|
|
|
|
def run_trading_twins_process(target_company_name: str):
|
|
"""
|
|
Startet den Trading Twins Prozess für ein Zielunternehmen.
|
|
Ruft den Company Explorer Workflow auf, um das Unternehmen zu finden,
|
|
zu erstellen oder anzureichern.
|
|
"""
|
|
print(f"\n{'='*50}")
|
|
print(f"Starte Trading Twins Analyse für: {target_company_name}")
|
|
print(f"{'='*50}\n")
|
|
|
|
# Aufruf des Company Explorer Workflows
|
|
# Diese Funktion prüft, ob die Firma existiert.
|
|
# Wenn nicht, erstellt sie die Firma und startet die Anreicherung.
|
|
# Sie gibt am Ende die Daten aus dem Company Explorer zurück.
|
|
company_data_result = handle_company_workflow(target_company_name)
|
|
|
|
# Verarbeitung der Rückgabe (für den POC genügt eine Ausgabe)
|
|
print("\n--- Ergebnis vom Company Explorer Connector (für Trading Twins) ---")
|
|
|
|
status = company_data_result.get("status")
|
|
data = company_data_result.get("data")
|
|
|
|
if status == "error":
|
|
print(f"Ein Fehler ist aufgetreten: {company_data_result.get('message')}")
|
|
elif status == "found":
|
|
print(f"Unternehmen gefunden. ID: {data.get('id')}, Name: {data.get('name')}")
|
|
print(json.dumps(data, indent=2, ensure_ascii=False))
|
|
elif status == "created_and_enriched":
|
|
print(f"Unternehmen erstellt und Enrichment angestoßen. ID: {data.get('id')}, Name: {data.get('name')}")
|
|
print("Hinweis: Enrichment-Prozesse laufen im Hintergrund und können einige Zeit dauern, bis alle Daten verfügbar sind.")
|
|
print(json.dumps(data, indent=2, ensure_ascii=False))
|
|
elif status == "created_discovery_timeout":
|
|
print(f"Unternehmen erstellt, aber Discovery konnte keine Website finden (ID: {data.get('id')}, Name: {data.get('name')}).")
|
|
print("Der Analyse-Prozess wurde daher nicht gestartet.")
|
|
print(json.dumps(data, indent=2, ensure_ascii=False))
|
|
else:
|
|
print("Ein unerwarteter Status ist aufgetreten.")
|
|
print(json.dumps(company_data_result, indent=2, ensure_ascii=False))
|
|
|
|
print(f"\n{'='*50}")
|
|
print(f"Trading Twins Analyse für {target_company_name} abgeschlossen.")
|
|
print(f"{'='*50}\n")
|
|
|
|
def run_email_ingest():
|
|
"""Starts the automated email ingestion process for Tradingtwins leads."""
|
|
if process_leads:
|
|
print("\nStarting automated email ingestion via Microsoft Graph...")
|
|
process_leads()
|
|
print("Email ingestion completed.")
|
|
else:
|
|
print("Error: Email ingestion module not available.")
|
|
|
|
if __name__ == "__main__":
|
|
# Simulieren der Umgebungsvariablen für diesen Testlauf, falls nicht gesetzt
|
|
if "COMPANY_EXPLORER_API_USER" not in os.environ:
|
|
os.environ["COMPANY_EXPLORER_API_USER"] = "admin"
|
|
if "COMPANY_EXPLORER_API_PASSWORD" not in os.environ:
|
|
os.environ["COMPANY_EXPLORER_API_PASSWORD"] = "gemini"
|
|
|
|
print("Trading Twins Tool - Main Menu")
|
|
print("1. Process specific company name")
|
|
print("2. Ingest leads from Email (info@robo-planet.de)")
|
|
print("3. Run demo sequence (Robo-Planet, Erding, etc.)")
|
|
|
|
choice = input("\nSelect option (1-3): ").strip()
|
|
|
|
if choice == "1":
|
|
name = input("Enter company name: ").strip()
|
|
if name:
|
|
run_trading_twins_process(name)
|
|
elif choice == "2":
|
|
run_email_ingest()
|
|
elif choice == "3":
|
|
# Testfall 1: Ein Unternehmen, das wahrscheinlich bereits existiert
|
|
run_trading_twins_process("Robo-Planet GmbH")
|
|
time.sleep(2)
|
|
# Testfall 1b: Ein bekanntes, real existierendes Unternehmen
|
|
run_trading_twins_process("Klinikum Landkreis Erding")
|
|
time.sleep(2)
|
|
# Testfall 2: Ein neues, eindeutiges Unternehmen
|
|
new_unique_company_name = f"Trading Twins New Target {int(time.time())}"
|
|
run_trading_twins_process(new_unique_company_name)
|
|
else:
|
|
print("Invalid choice.")
|