config.py

# 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)