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

app = Flask(__name__)
HF_TOKEN = os.environ.get("HF_TOKEN")  # Токен должен быть добавлен в Variables

def generate_response(user_text):
    API_URL = "https://api-inference.huggingface.co/models/tinkoff-ai/ruDialoGPT-medium"
    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
    try:
        response = requests.post(API_URL, headers=headers, json={"inputs": user_text, "wait_for_model": True})
        if response.status_code == 200:
            return response.json()[0]["generated_text"]
        else:
            return f"Ошибка: {response.status_code} - {response.text}"
    except Exception as e:
        return f"Ошибка при запросе к модели: {str(e)}"

@app.route("/chat", methods=["POST"])
def chat():
    data = request.json
    user_message = data.get("message", "")
    return jsonify({"response": generate_response(user_message)})

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