import os
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# Get Hugging Face API token
HUGGINGFACE_API_TOKEN = os.getenv('HUGGINGFACE_API_TOKEN')

# API configuration
API_CONFIG = {
    'token': HUGGINGFACE_API_TOKEN,
    'models': {
        'summarization': 'meta-llama/Llama-2-70b-chat-hf',
        'question_answering': 'deepseek-ai/DeepSeek-V2-Chat',
        'image_captioning': 'Qwen/Qwen2.5-VL-7B-Instruct',
        'translation': 'facebook/mbart-large-50-many-to-many-mmt'
    }
}