config.py

"""
Configuration file for LLM Brand Visibility Tracker
"""

# Available OpenAI models
AVAILABLE_MODELS = [
    "gpt-4o-mini",
    "gpt-3.5-turbo",
    "gpt-4o",
    "gpt-4-turbo",
]

# Default model
DEFAULT_MODEL = "gpt-4o-mini"

# Default parameters
DEFAULT_TEMPERATURE = 0.7
DEFAULT_MAX_TOKENS = 2000
DEFAULT_SAMPLE_COUNT = 50

# Sample count options
SAMPLE_COUNT_OPTIONS = [10, 20, 50, 100, 200, 500]

# System prompt template for brand extraction
SYSTEM_PROMPT_TEMPLATE = """You are a helpful assistant that provides brand recommendations in JSON format.

Respond to the user's question by providing a list of recommended brands.

Instructions:
- Generate a list of MAXIMUM 10 brands.
- Do not add any comments or descriptions, only brand names.
- Return the response in exactly this JSON format:
{{"brands": ["Brand1", "Brand2", "Brand3"]}}

User's question:
{user_prompt}"""

# UI Configuration
APP_TITLE = "🔍 LLM Brand Visibility Tracker"
APP_DESCRIPTION = """
Track how often and in what positions your brand appears in LLM responses.
Measure brand visibility, position, and share of mind across multiple AI-generated recommendations.
"""

# Sidebar configuration
SIDEBAR_TITLE = "⚙️ Configuration"