Tim2190 commited on
Commit
73dc5eb
·
verified ·
1 Parent(s): 82d293d

Create main.py

Browse files
Files changed (1) hide show
  1. main.py +44 -0
main.py ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import psycopg2
3
+ from fastapi import FastAPI, Request
4
+
5
+ app = FastAPI()
6
+
7
+ # Берем данные из переменных окружения
8
+ DATABASE_URL = os.getenv("DATABASE_URL")
9
+
10
+ @app.post("/webhook")
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", "")
19
+ sentiment_score = data.get("sentiment_score", 0)
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()
36
+ conn.close()
37
+
38
+ return {"status": "success", "message": "Feedback saved to Neon"}
39
+ except Exception as e:
40
+ return {"status": "error", "message": str(e)}
41
+
42
+ @app.get("/")
43
+ def read_root():
44
+ return {"message": "Python Bridge is Running!"}