import os
import requests

HF_TOKEN = os.getenv("HF_TOKEN")
API_URL = "https://api-inference.huggingface.co/models/microsoft/DialoGPT-medium"
headers = {"Authorization": f"Bearer {HF_TOKEN}"}

def get_bot_response(user_input):
    payload = {"inputs": {"text": user_input}}
    response = requests.post(API_URL, headers=headers, json=payload)

    if response.status_code != 200:
        return f"❌ Error: {response.status_code} - {response.json()}"

    return response.json()[0]["generated_text"]