Adeptih commited on
Commit
9ba0f04
·
verified ·
1 Parent(s): cfafea0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -19
app.py CHANGED
@@ -3,35 +3,19 @@ import requests
3
  from flask import Flask, request, jsonify
4
 
5
  app = Flask(__name__)
6
-
7
- # Токен Hugging Face API
8
- HF_TOKEN = os.environ.get("HF_TOKEN")
9
 
10
  def generate_response(user_text):
11
- """Генерация ответа через Hugging Face API"""
12
  API_URL = "https://api-inference.huggingface.co/models/tinkoff-ai/ruDialoGPT-medium"
13
  headers = {"Authorization": f"Bearer {HF_TOKEN}"}
14
-
15
  response = requests.post(API_URL, headers=headers, json={"inputs": user_text})
16
- if response.status_code == 200:
17
- return response.json()[0]["generated_text"]
18
- else:
19
- return "Я не знаю, что ответить 😅"
20
 
21
  @app.route("/chat", methods=["POST"])
22
  def chat():
23
- """Обработка входящих сообщений"""
24
  data = request.json
25
  user_message = data.get("message", "")
26
-
27
- if not user_message.strip():
28
- return jsonify({"response": "Вы отправили пустое сообщение 😅"})
29
-
30
- try:
31
- bot_response = generate_response(user_message)
32
- return jsonify({"response": bot_response})
33
- except Exception as e:
34
- return jsonify({"response": "Упс, я сломался... Попробуй еще раз 😅"})
35
 
36
  if __name__ == "__main__":
37
  app.run(debug=True)
 
3
  from flask import Flask, request, jsonify
4
 
5
  app = Flask(__name__)
6
+ HF_TOKEN = os.environ.get("HF_TOKEN") # Токен должен быть добавлен в Variables
 
 
7
 
8
  def generate_response(user_text):
 
9
  API_URL = "https://api-inference.huggingface.co/models/tinkoff-ai/ruDialoGPT-medium"
10
  headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
11
  response = requests.post(API_URL, headers=headers, json={"inputs": user_text})
12
+ return response.json()[0]["generated_text"] if response.status_code == 200 else "Ошибка"
 
 
 
13
 
14
  @app.route("/chat", methods=["POST"])
15
  def chat():
 
16
  data = request.json
17
  user_message = data.get("message", "")
18
+ return jsonify({"response": generate_response(user_message)})
 
 
 
 
 
 
 
 
19
 
20
  if __name__ == "__main__":
21
  app.run(debug=True)