import os
import requests
from flask import Flask, request, jsonify

app = Flask(__name__)

# Токен Hugging Face API
HF_TOKEN = os.environ.get("HF_TOKEN")

def generate_response(user_text):
    """Генерация ответа через Hugging Face API"""
    API_URL = "https://api-inference.huggingface.co/models/tinkoff-ai/ruDialoGPT-medium"
    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
    
    response = requests.post(API_URL, headers=headers, json={"inputs": user_text})
    if response.status_code == 200:
        return response.json()[0]["generated_text"]
    else:
        return "Я не знаю, что ответить 😅"

@app.route("/chat", methods=["POST"])
def chat():
    """Обработка входящих сообщений"""
    data = request.json
    user_message = data.get("message", "")
    
    if not user_message.strip():
        return jsonify({"response": "Вы отправили пустое сообщение 😅"})
    
    try:
        bot_response = generate_response(user_message)
        return jsonify({"response": bot_response})
    except Exception as e:
        return jsonify({"response": "Упс, я сломался... Попробуй еще раз 😅"})

if __name__ == "__main__":
    app.run(debug=True)