Tim2190 commited on
Commit
17c711c
·
verified ·
1 Parent(s): 73dc5eb

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +18 -6
main.py CHANGED
@@ -11,8 +11,13 @@ DATABASE_URL = os.getenv("DATABASE_URL")
11
  async def handle_feedback(request: Request):
12
  data = await request.json()
13
 
14
- # Извлекаем данные (те самые, что шлет Flowise)
15
  business_name = data.get("business_name", "Unknown")
 
 
 
 
 
16
  risk_level = data.get("risk_level", "Low")
17
  category = data.get("category", "Other")
18
  summary = data.get("summary", "")
@@ -20,16 +25,23 @@ async def handle_feedback(request: Request):
20
  action_required = data.get("action_required", "")
21
 
22
  try:
23
- # Подключаемся к Neon
24
  conn = psycopg2.connect(DATABASE_URL)
25
  cur = conn.cursor()
26
 
27
- # Запрос на вставку (убедись, что имена колонок в базе совпадают)
28
  query = """
29
- INSERT INTO feedback (business_name, risk_level, category, summary, sentiment_score, action_required)
30
- VALUES (%s, %s, %s, %s, %s, %s)
 
 
 
31
  """
32
- cur.execute(query, (business_name, risk_level, category, summary, sentiment_score, action_required))
 
 
 
 
 
33
 
34
  conn.commit()
35
  cur.close()
 
11
  async def handle_feedback(request: Request):
12
  data = await request.json()
13
 
14
+ # 1. ТЕПЕРЬ МЫ ИЗВЛЕКАЕМ ВСЁ
15
  business_name = data.get("business_name", "Unknown")
16
+ branch = data.get("branch") # Добавлено
17
+ contact = data.get("contact") # Добавлено
18
+ raw_message = data.get("raw_message") # Добавлено
19
+ language = data.get("language") # Добавлено
20
+
21
  risk_level = data.get("risk_level", "Low")
22
  category = data.get("category", "Other")
23
  summary = data.get("summary", "")
 
25
  action_required = data.get("action_required", "")
26
 
27
  try:
 
28
  conn = psycopg2.connect(DATABASE_URL)
29
  cur = conn.cursor()
30
 
31
+ # 2. ОБНОВЛЕННЫЙ ЗАПРОС (Добавлены 4 колонки)
32
  query = """
33
+ INSERT INTO feedback (
34
+ business_name, branch, contact, raw_message, language,
35
+ risk_level, category, summary, sentiment_score, action_required
36
+ )
37
+ VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
38
  """
39
+
40
+ # 3. ПЕРЕДАЕМ ВСЕ 10 ПАРАМЕТРОВ
41
+ cur.execute(query, (
42
+ business_name, branch, contact, raw_message, language,
43
+ risk_level, category, summary, sentiment_score, action_required
44
+ ))
45
 
46
  conn.commit()
47
  cur.close()