Update chatbot_tab.py

tabs/chatbot_tab.py

@@ -90,14 +90,52 @@ human_message1=""
 thread_id =""
 virulence = 1
 question = []
+thread_id = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+config = {"configurable": {"thread_id": thread_id}}
+to_init = True
+initialized = False
+messages = [
+    SystemMessage(content=""),
+    HumanMessage(content=""),
+    AIMessage(content=""),
+    HumanMessage(content="")
+    ]
 if 'model' in st.session_state:
+    model = st.session_state.model
     used_model = st.session_state.model
 
-# @st.cache_data
+def init_run():
+    global initialized, to_init, thread_id, config, app, context, human_message1, model, used_model, messages
+
+    initialized = True
+    to_init = False
+    thread_id = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    config = {"configurable": {"thread_id": thread_id}}
+    app.invoke(
+    {"messages": messages, "language": language},
+    config,
+    )  
+    st.session_state.thread_id = thread_id
+    st.session_state.config = config
+    st.session_state.messages_init = messages
+    st.session_state.context = context
+    st.session_state.human_message1 = human_message1
+    st.session_state.messages = []
+    if 'model' in st.session_state and (st.session_state.model[:3]=="gpt") and ("OPENAI_API_KEY" in st.session_state):
+        model = ChatOpenAI(model=st.session_state.model,
+        temperature=0.8,  # Adjust creativity level
+        max_tokens=150   # Define max output token limit
+        )           
+    else: 
+        model = ChatMistralAI(model=st.session_state.model)
+    if 'model' in st.session_state:
+        used_model=st.session_state.model
+    return 
+
 def init():
-    global config,thread_id, context,human_message1,ai_message1,language, app, model_speech,prompt,model,question
+    global config,thread_id, context,human_message1,ai_message1,language, app, model_speech,prompt,model,question, to_init, initialized
     global selected_index1, selected_index2, selected_index3, selected_indices4,selected_indices5,selected_indices6,selected_indices7 
-    global selected_options4,selected_options5,selected_options6,selected_options7, selected_index8, virulence, used_model
+    global selected_options4,selected_options5,selected_options6,selected_options7, selected_index8, virulence, used_model, messages
     
     model_speech = whisper.load_model("base") 
     
@@ -229,7 +267,7 @@ Attention: Si le vendeur aborde des points qui ne concerne pas cette simulation,
         SystemMessage(content=context),
         HumanMessage(content=human_message1),
         AIMessage(content=ai_message1),
-        HumanMessage(content=tr("Commençons la conversation. Attention, je parle le premier"))
+        HumanMessage(content=tr("Commençons la conversation. Attention, je suis le vendeur et je parle le premier. Tu es le propect."))
         ]
 
     st.write("")
@@ -240,31 +278,13 @@ Attention: Si le vendeur aborde des points qui ne concerne pas cette simulation,
             to_init = False
     else:
         to_init = True
-        
+    
     if to_init:
-        thread_id = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        config = {"configurable": {"thread_id": thread_id}}
-        app.invoke(
-        {"messages": messages, "language": language},
-        config,
-        )  
-        st.session_state.thread_id = thread_id
-        st.session_state.config = config
-        st.session_state.messages_init = messages
-        st.session_state.context = context
-        st.session_state.human_message1 = human_message1
-        st.session_state.messages = []
-        if 'model' in st.session_state and (st.session_state.model[:3]=="gpt") and ("OPENAI_API_KEY" in st.session_state):
-            model = ChatOpenAI(model=st.session_state.model,
-            temperature=0.8,  # Adjust creativity level
-            max_tokens=150   # Define max output token limit
-            )           
-        else: 
-            model = ChatMistralAI(model=st.session_state.model)
-        if 'model' in st.session_state:
-            used_model=st.session_state.model
-        
-    return config, thread_id
+        if st.button(label=tr("Validez"), on_click=init_run,type="primary"):
+            initialized=True
+        else: initialized = False
+    st.write("**thread_id:** "+thread_id) 
+    return config, thread_id, messages
 
 # Fonction pour générer et jouer le texte en speech
 def play_audio(custom_sentence, Lang_target, speed=1.0):
@@ -298,7 +318,7 @@ def play_audio(custom_sentence, Lang_target, speed=1.0):
 
 
 def run():
-    global thread_id, config, model_speech, language,prompt,model, model_name, question
+    global thread_id, config, model_speech, language,prompt,model, model_name, question, to_init, initialized, messages
    
     st.write("")
     st.write("")
@@ -325,27 +345,17 @@ def run():
             model = ChatMistralAI(model=st.session_state.model)
             
 
-        config,thread_id = init()
+        config,thread_id, messages = init()
         query = ""
-        st.button(label=tr("Validez"), type="primary")
-        st.write("**thread_id:** "+thread_id)
     elif (chosen_id == "tab2"):
         try:
-            config
-            # On ne fait rien
+            if to_init and not initialized:
+                init_run()
         except NameError:
-            config,thread_id = init()
+            config,thread_id, messages = init()
         with st.container():
             # Diviser l'écran en deux colonnes
             col1, col2 = st.columns(2)
-            # with col1:
-            #     st.markdown(
-            #         """
-            #         <div style="height: 400px;">
-            #         </div>
-            #         """,
-            #         unsafe_allow_html=True,
-            #         )
             with col1:
                 st.write("**thread_id:** "+thread_id)
                 query = ""
@@ -430,19 +440,27 @@ def run():
                             with st.chat_message(message["role"]):
                                 st.markdown(message["content"])
     else:
+        if to_init and not initialized:
+            init_run()
         st.write("**thread_id:** "+thread_id)
         for i in range(8,len(question)):
             st.write("")
 
-            q = st.text_input(label="", value=tr(question[i]),label_visibility="collapsed")
-            output = app.invoke(
-                {"messages": q,"language": language},
-                config,
-            ) 
-            custom_sentence = output["messages"][-1].content  
-            st.write(custom_sentence)
-            st.write("")
-            if (used_model[:3] == 'mis'):
-                time.sleep(2)
-            
-            st.divider()
+            q = st.text_input(label=".", value=tr(question[i]),label_visibility="collapsed")
+            if (q !=""):
+                input_messages = [HumanMessage(q)]
+                output = app.invoke(
+                    {"messages": input_messages, "language": language},
+                    config,
+                    )
+            # output = app.invoke(
+            #     {"messages": q,"language": language},
+            #     config,
+            # ) 
+                custom_sentence = output["messages"][-1].content  
+                st.write(custom_sentence)
+                st.write("")
+                if (used_model[:3] == 'mis'):
+                    time.sleep(2)
+                
+                st.divider()