Update config.py

config.py

@@ -1,24 +1,38 @@
-from flask import Flask,request,render_template,request,Response,jsonify
-from application.utils.convs_handler import ConvHandler
-import json
-
-pipeline_path = './pipeline.json'
-
-with open(pipeline_path, 'r') as file:
-    pipeline_dict = json.load(file)
-
-convs_dict = {
-    #format---> ip:{ 
-    #               convId:{
-    #                   messages = [],
-    #               }
-    #               ...
-    #               metadata (a list containing all convId and title in dict format) = [
-    #                       {
-    #                           "convID",
-    #                           "convTitle"
-    #                       }...]
-    #                  
-}
-
-convHandler = ConvHandler(convs_dict=convs_dict)
+# config.py
+from flask import Flask, request, render_template, request, Response, jsonify
+from application.utils.convs_handler import ConvHandler
+import json
+import uuid  # Import uuid
+
+pipeline_path = './pipeline.json'
+
+with open(pipeline_path, 'r') as file:
+    pipeline_dict = json.load(file)
+
+convs_dict = {
+    # format---> ip:{
+    #               convId:{
+    #                   messages = [],
+    #               }
+    #               ...
+    #               metadata (a list containing all convId and title in dict format) = [
+    #                       {
+    #                           "convID",
+    #                           "convTitle"
+    #                       }...]
+    #
+    #               memory: ""  # Add memory field here
+    # }
+}
+
+
+def get_user_id():
+    user_id = request.cookies.get('user_id')
+    if not user_id:
+        user_id = str(uuid.uuid4())
+    if user_id not in convs_dict:
+        convs_dict[user_id] = {"metadata": [], "memory": ""}  # Initialize memory
+    return user_id
+
+
+convHandler = ConvHandler(convs_dict=convs_dict)