|
import os |
|
import requests |
|
from flask import Flask, request, jsonify |
|
|
|
app = Flask(__name__) |
|
HF_TOKEN = os.environ.get("HF_TOKEN") |
|
|
|
def generate_response(user_text): |
|
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}) |
|
return response.json()[0]["generated_text"] if response.status_code == 200 else "Ошибка" |
|
|
|
@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) |
