convo2db (#3)

- transcript2db (adae8efe26968a9fad055865c14e78ca6296f30c)

models/ta_models/config.py

@@ -28,9 +28,9 @@ BP_LAB2STR = {
 }
 
 QUESTION2PHASE = {
-    "question_1": ["0_ActiveEngagement","1_Explore"],
+    "question_1": ["0_ActiveEngagement"],
     "question_4": ["1_Explore"],
-    "question_5": ["0_ActiveEngagement", "1_Explore"],
+    "question_5": ["1_Explore"],
     # "question_7": ["1_Explore"],
     # "question_9": ["4_SP&NS"],
     "question_10": ["4_SP&NS"],

pages/convosim.py

@@ -2,7 +2,7 @@ import os
 import streamlit as st
 from streamlit.logger import get_logger
 from langchain.schema.messages import HumanMessage
-from utils.mongo_utils import get_db_client
+from utils.mongo_utils import get_db_client, update_convo
 from utils.app_utils import create_memory_add_initial_message, get_random_name,  DEFAULT_NAMES_DF, are_models_alive
 from utils.memory_utils import clear_memory, push_convo2db
 from utils.chain_utils import get_chain, custom_chain_predict
@@ -99,6 +99,8 @@ def sent_request_llm(llm_chain, prompt):
     responses = custom_chain_predict(llm_chain, prompt, stopper)
     for response in responses:
         st.chat_message("assistant").write(response)
+    transcript = memoryA.load_memory_variables({})[memoryA.memory_key]
+    update_convo(st.session_state["db_client"], st.session_state["convo_id"], transcript)
 
 # @st.dialog("Bad Practice Detected")
 # def confirm_bp(bp_prediction, prompt):

utils/mongo_utils.py

@@ -41,6 +41,17 @@ def new_convo(client, issue, language, username, is_comparison, model_one, model
     logger.debug(f"DBUTILS: new convo id is {convo_id}")
     st.session_state['convo_id'] = convo_id
 
+def update_convo(client, convo_id, transcript=""):
+    from bson.objectid import ObjectId
+    db = client[DB_SCHEMA]
+    convos = db[DB_CONVOS]
+    myquery = { "_id": ObjectId(convo_id) }
+    newvalues = { "$set": { "transcript": transcript } }
+    result = convos.update_one(myquery, newvalues)
+    if result.matched_count == 1:
+        logger.debug(f"Updated conversation {convo_id}")
+
+
 def new_comparison(client, prompt_timestamp, completion_timestamp, 
                    chat_history, prompt, completionA, completionB,
                    source="webapp", subset=None