Update app.py

app.py

@@ -42,6 +42,7 @@ os.makedirs(VECTOR_DB_FOLDER, exist_ok=True)
 
 # Global variables
 CHROMA_PATH = None
+TEMP_PATH = None
 #TABLE_PATH = None
 
 #System prompt
@@ -95,14 +96,12 @@ def chat():
     global CHROMA_PATH
     #global TABLE_PATH
     
-    old_db = session.get('old_db', None) 
-    print(f"Selected DB: {CHROMA_PATH}")
+    #old_db = session.get('old_db', None) 
+    #print(f"Selected DB: {CHROMA_PATH}")
     
-    # if old_db != None:
-    #     if CHROMA_PATH != old_db:
-    #         session['history'] = [] 
-        
-    #print("sessionhist1",session['history'])
+    if TEMP_PATH != None and TEMP_PATH != CHROMA_PATH:
+        session['history'] = []
+    else: TEMP_PATH=CHROMA_PATH
     
     if request.method == 'POST':
         query_text = request.form['query_text']
@@ -159,9 +158,9 @@ def chat():
         session.modified = True
         print("sessionhist2",session['history'])
 
-        return render_template('chat.html', query_text=query_text, answer=data, history=session['history'],old_db=CHROMA_PATH)
+        return render_template('chat.html', query_text=query_text, answer=data, history=session['history'])
 
-    return render_template('chat.html', history=session['history'], old_db=CHROMA_PATH)
+    return render_template('chat.html', history=session['history'])
 
 @app.route('/create-db', methods=['GET', 'POST'])
 def create_db():