import os
import json
from openai import OpenAI
# -------------------------------------------------
# Customer / Willkommensseite
# -------------------------------------------------
WELCOME_DIR = os.path.join(os.path.dirname(__file__), "media", "lobby")
os.makedirs(WELCOME_DIR, exist_ok=True)
WELCOME_FILENAME = "welcome.html"
def get_openai_client():
"""Initialize OpenAI client from environment variable"""
api_key = os.getenv("OPENAI_API_KEY")
if not api_key:
raise ValueError("OPENAI_API_KEY environment variable not set")
return OpenAI(api_key=api_key)
def get_brandfetch_logo(domain):
"""Try to fetch a logo URL from Brandfetch CDN."""
print(f"🔍 Versuche Brandfetch für Domain: {domain}")
# Direct CDN URL construction - Brandfetch provides logos via cdn.brandfetch.io
logo_url = f"https://cdn.brandfetch.io/{domain}?c=1idyd4Tpb2nKaXIIc8T"
# if validate_logo_url(logo_url):
print(f"✅ Brandfetch Logo gefunden: {logo_url}")
return logo_url
# return None
def search_customer_logo(customer_name):
"""Search for customer logo URL using OpenAI and web search"""
try:
client = get_openai_client()
response = client.chat.completions.create(
model="gpt-4",
messages=[
{
"role": "system",
"content": """Du bist ein Experte für die Suche nach Domain Namen. Du findest anhand des Firmennamens die richtige Domain dazu
Gib nur den Domain Namen zurück, keine weiteren Informationen.
FALLBACK: Wenn du kein direkt Aufrufbare Domain findest, antworte mit "FALLBACK"."""
},
{
"role": "user",
"content": f"Finde die Domain für dasUnternehmen: {customer_name}"
}
],
temperature=0.1,
max_tokens=100
)
logo_url = response.choices[0].message.content.strip()
print (f"🔍 OpenAI Logo-URL: {logo_url}")
if logo_url.upper() == "FALLBACK":
logo_url = None
if logo_url:
brandfetch_url = get_brandfetch_logo(logo_url)
if brandfetch_url:
return brandfetch_url
return None
except Exception as e:
print(f"❌ Logo search error: {e}")
return None
def generate_welcome_html(customer_name, logo_url):
"""Generate a welcome page HTML with full-screen design"""
# CANCOM SVG logo inline
cancom_svg = ''''''
html_content = f"""
Willkommen bei CANCOM
{cancom_svg}
meets
{f'' if logo_url else ''}
Herzlich Willkommen in unserer
Niederlassung Stuttgart!
"""
return html_content
def save_welcome_page(customer_name, logo_url):
"""Save welcome page and return filename"""
try:
filepath = os.path.join(os.path.abspath(WELCOME_DIR), WELCOME_FILENAME)
html_content = generate_welcome_html(customer_name, logo_url)
with open(filepath, "w", encoding="utf-8") as f:
f.write(html_content)
if os.path.getsize(filepath) == 0:
raise IOError("File not written")
print(f"✅ Willkommensseite generiert für: {customer_name}")
print(f"✅ Willkommensseite gespeichert in: {filepath}")
print(f"✅ Willkommensseite gespeichert: {WELCOME_FILENAME}")
return WELCOME_FILENAME
except Exception as e:
print(f"❌ Error saving welcome page: {e}")
return None
def add_customer_to_lobby_playlist(html_filename):
"""Add customer welcome page to lobby playlist"""
try:
# Import here to avoid circular imports
from app import load_config, save_config
config = load_config()
screens = config.setdefault("screens", {})
lobby = screens.setdefault("lobby", {})
playlist = lobby.setdefault("playlist", [])
full_filename = f"{html_filename}"
if full_filename in playlist:
playlist.remove(full_filename)
playlist.insert(0, full_filename)
save_config(config)
print(f"✅ Willkommensseite in Lobby-Playliste platziert")
return True
except Exception as e:
print(f"❌ Error adding to playlist: {e}")
return False