Update chatbot_tab.py

tabs/chatbot_tab.py

@@ -1,5 +1,6 @@
 import streamlit as st # type: ignore
 import os
+from datetime import datetime
 from sentence_transformers import SentenceTransformer
 from translate_app import tr
 import getpass
@@ -26,6 +27,7 @@ os.environ["LANGCHAIN_PROJECT"] = "Sales Coaching Chatbot"
 os.getenv("LANGCHAIN_API_KEY")
 os.getenv("MISTRAL_API_KEY")
 model = ChatMistralAI(model="mistral-large-latest")
+thread_id = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
 dataPath = st.session_state.DataPath
 
@@ -72,16 +74,10 @@ workflow.add_edge("model", END)
 memory = MemorySaver()
 app = workflow.compile(checkpointer=memory)
 
-config = {"configurable": {"thread_id": "abc1234"}}
+config = {"configurable": {"thread_id": thread_id}}
 
-def run():
-
-    st.write("")
-    st.write("")
-    st.title(tr(title))
-    
-    context = """Tu es un Directeur Commercial d'une entreprise qui commercialise une solution technologique B2B"""
-    human_message1 = """Je souhaites que tu simule une conversation entre un commercial, Marc, de mon entreprise et toi, je prospecte.
+context = """Tu es un Directeur Commercial, mal organisé, d'une entreprise qui commercialise une solution technologique B2B"""
+human_message1 = """Je souhaites que tu simule une conversation entre un commercial, Marc, de mon entreprise et toi, je prospecte.
 Mon entreprise propose une solution logicielle pour gérer la proposition de valeur d’entreprises B2B qui commercialises des solutions technologiques.   
     
 Les problématiques adressées par ma solution sont:  
@@ -109,30 +105,46 @@ Les cas d’usages adressés par ma solution sont  :
 - Identifier et reproduire les messages de vente gagnants
 
 Je suis Marc, le vendeur.
-Répond à mes questions en tant que Directeur commercial.  
+Répond à mes questions en tant que Directeur commercial désorganisé, connaissant mal le concept de proposition de valeur,
+et mon équipe de vente n'est pas performante.  
  
 Attention: Ce n'est pas toi qui m'aide, c'est moi qui t'aide avec ma solution.
 """
-  
-    messages = [
-        SystemMessage(content=context),
-        HumanMessage(content=human_message1),
-        AIMessage(content="J'ai bien compris, je suis un Directeur Commercial prospecté et je réponds à tes questions"),
-        ]
+ai_message1 = "J'ai bien compris, je suis un Directeur Commercial prospecté et je réponds à tes questions"   
+
+messages = [
+    SystemMessage(content=context),
+    HumanMessage(content=human_message1),
+    AIMessage(content=ai_message1),
+    ]
     
-    trimmer.invoke(messages)
     
-    # query = "Hi I'm Todd, please tell me a joke."
-    query = st.text_area(label=tr("Vendeur:"), value="")
+trimmer.invoke(messages)
+
+def run():
+
+    st.write("")
+    st.write("")
+    st.title(tr(title))
+    
     language = "French"
-    st.button(label=tr("Validez"), type="primary")
+    st.write("Contexte: "+context+"\n")
+    st.write("Human Message: "+human_message1+"\n")
+    st.write("AI Message: "+ai_message1+"\n")
+    query = ""
+        
+    while query != "exit":
+        st.write("thread_id: "+thread_id)
+        query = st.text_area(label=tr("Vendeur:"), value="")
+        st.button(label=tr("Validez"), type="primary")
     
-    input_messages = [HumanMessage(query)]
-    output = app.invoke(
-            {"messages": input_messages, "language": language},
-            config,
-        )
-    st.write(output["messages"][-1].content) #.pretty_print())
+        input_messages = [HumanMessage(query)]
+        if query != "":
+            output = app.invoke(
+                {"messages": input_messages, "language": language},
+                config,
+            )
+            st.write(output["messages"][-1].content)
 
     
     '''