Update chat_application/main.py

chat_application/main.py

@@ -143,6 +143,7 @@ frobot = "projects/700531062565/locations/us-central1/endpoints/2951406418055397
 client = MongoClient("mongodb://127.0.0.1:27017/")
 db = client["huggingFaceData"]
 rooms_collection = db.rooms
+feedback_collection = db.feedback
 
 # List of fruits to choose display names from
 FRUIT_NAMES = ["blueberry", "strawberry", "orange", "cherry"]
@@ -639,6 +640,16 @@ def choose():
         "ended": False,
         "ended_at": None
     })
+    # Create the new feedback in the database
+    feedback_collection.insert_one({
+        "_id": room_id,
+        # creation date/time
+        "created_at": datetime.utcnow(),
+        # user identity
+        "user_id": user_id,
+        # empty feedback history
+        "feedbacks": [],
+    })
 
     session['room'] = room_id
     session['display_name'] = user_name
@@ -807,6 +818,34 @@ def handle_message(payload):
     # Ask each bot for a response
     socketio.start_background_task(ask_bot_round, room)
 
+
+@socketio.on('feedback_given')
+def handle_message(payload):
+    room = session.get('room')
+    name = session.get('display_name')
+    if not room or not name:
+        return
+
+    text = payload.get("feedback", "").strip()
+    if not text:
+        return  # ignore empty text
+
+    # Database-only message (with datetime)
+    db_feedback = {
+        "message": text,
+        "timestamp": datetime.utcnow()
+    }
+
+    print(db_feedback)
+    # Store the full version in the database
+    result = feedback_collection.update_one(
+        {"_id": room},
+        {"$push": {"feedback_responses": db_feedback}}
+    )
+
+    #LR TODO: add success/faillure msg and confirmation
+
+
 @socketio.on('disconnect')
 def handle_disconnect():
     room = session.get("room")