Update chatbot_tab.py

tabs/chatbot_tab.py

@@ -73,31 +73,51 @@ workflow.add_edge("model", END)
 memory = MemorySaver()
 app = workflow.compile(checkpointer=memory)
 
+selected_index1 = 0
+selected_index2 = 0
+selected_index3 = 0
+selected_options4 = []
+selected_options5 = []
+selected_options6 = []
+selected_options7 = []
+virulence = 1
+
 # @st.cache_data
 def init():
-    global config,thread_id, context,human_message1,ai_message1,language, app, model_speech,language,prompt,model
+    global config,thread_id, context,human_message1,ai_message1,language, app, model_speech,prompt,model
+    global selected_index1, selected_index2, selected_index3, selected_options4,selected_options5,selected_options6,selected_options7, virulence
     
     model_speech = whisper.load_model("base") 
-       
-    options = ["Directeur Commercial", "Directeur Général", "Directeur Marketing"]
-    translated_options = [tr(o) for o in options]
-    selected_option = st.selectbox(tr("Interlocuteur"),translated_options)
-    selected_index = translated_options.index(selected_option)
+    
+    if st.button(label=tr("Nouvelle conversation"), type="primary"):
+        selected_index1 = 0
+        selected_index2 = 0
+        selected_index3 = 0
+        selected_options4 = []
+        selected_options5 = []
+        selected_options6 = []
+        selected_options7 = []
+        virulence = 1
+      
+    options1 = ["Directeur Commercial", "Directeur Général", "Directeur Marketing"]
+    translated_options1 = [tr(o) for o in options1]
+    selected_option1 = st.selectbox(tr("Interlocuteur"),translated_options1, index = selected_index1) # index=int(var1_init))
+    selected_index1 = translated_options1.index(selected_option1)
     
     options2 = ["Entreprise qui commercialise des solutions (produits et ou services) B2B innovantes avec une équipe commerciale de plus de 15 personnes",
                 "Entreprise qui commercialise des solutions (produits et ou services) B2B innovantes avec une équipe commerciale de plus de 100 personnes"]
     translated_options2 = [tr(o) for o in options2]
-    selected_option2 = st.selectbox(tr("Activité"),translated_options2)
+    selected_option2 = st.selectbox(tr("Activité"),translated_options2, index = selected_index2) # index=int(var2_init))
     selected_index2 = translated_options2.index(selected_option2)
     
     options3 = ["Logiciels informatiques et d’application SaaS",
                 "Équipements et solutions industrielles",
                 "Services et conseil spécialisés"]
     translated_options3 = [tr(o) for o in options3]
-    selected_option3 = st.selectbox(tr("Domaine d'activité"),translated_options3)
+    selected_option3 = st.selectbox(tr("Domaine d'activité"),translated_options3, index=selected_index3) #index=int(var3_init))
     selected_index3 = translated_options3.index(selected_option3)
 
-    context = tr(f"""Tu es un {options[selected_index]}, mal organisé, d'une {options2[selected_index2]}.
+    context = tr(f"""Tu es un {options1[selected_index1]}, mal organisé, d'une {options2[selected_index2]}.
 Cette entreprise propose des {options3[selected_index3]}.
     """)
     context = st.text_area(label=tr("Résumé du Contexte (modifiable):"), value=context)
@@ -109,8 +129,8 @@ Cette entreprise propose des {options3[selected_index3]}.
                 "Il est difficile d’affiner une proposition de valeur unique et pertinente de l'offre face à la concurrence qui évolue rapidement",
                 "Il est chronophage de former les forces de ventes sur la proposition de valeur et ses évolutions"
     ]
-    selected_options4 = st.multiselect(tr("Problématiques"),[tr(o) for o in options4])
-    problematique = [tr(o) for o in selected_options4]
+    selected_options4 = st.multiselect(tr("Problématiques"),[tr(o) for o in options4], default=[tr(o) for o in selected_options4])
+    problematique = selected_options4
     markdown_text4 = tr("""
                         Les problématiques rencontrés par notre prospect (problèmes à résoudre) sont:""")
     markdown_text4 = markdown_text4+"".join(f"\n- {o}" for o in problematique)
@@ -121,8 +141,8 @@ Cette entreprise propose des {options3[selected_index3]}.
                 "Affiner et modéliser la proposition de valeur",
                 "Mettre en oeuvre des meilleures pratiques commerciales"
     ]
-    selected_options5 = st.multiselect(tr("Processus"),[tr(o) for o in options5])
-    processus = [tr(o) for o in selected_options5]
+    selected_options5 = st.multiselect(tr("Processus"),[tr(o) for o in options5],default=[tr(o) for o in selected_options5])
+    processus = selected_options5
     markdown_text5 = tr("""
                         \nLes processus adressés par le prospect (cas d’usages) sont:""")
     markdown_text5 = markdown_text5+"".join(f"\n- {o}" for o in processus)
@@ -136,8 +156,8 @@ Cette entreprise propose des {options3[selected_index3]}.
                 "Réduire temps de monté en compétence des nouvelles embauches",
                 "Fidéliser les clients"
     ]
-    selected_options6 = st.multiselect(tr("Objectifs d'amélioration"),[tr(o) for o in options6])
-    objectifs = [tr(o) for o in selected_options6]
+    selected_options6 = st.multiselect(tr("Objectifs d'amélioration"),[tr(o) for o in options6],default=[tr(o) for o in selected_options6])
+    objectifs = selected_options6
     markdown_text6 = tr("""
                         \nLes objectifs d’amélioration opérationnelle du prospect (Valeur ajoutée) sont:""")
     markdown_text6 = markdown_text6+"".join(f"\n- {o}" for o in objectifs)
@@ -150,8 +170,8 @@ Cette entreprise propose des {options3[selected_index3]}.
                 "Conseil externe en positionnement marché & produit",
                 "Services externes de formation des équipes commerciales"
     ]
-    selected_options7 = st.multiselect(tr("Solutions utilisées"),[tr(o) for o in options7])
-    solutions_utilisees = [tr(o) for o in selected_options7]
+    selected_options7 = st.multiselect(tr("Solutions utilisées"),[tr(o) for o in options7],default=[tr(o) for o in selected_options7])
+    solutions_utilisees = selected_options7
     markdown_text7 = tr("""
                         \nLes principales Solutions utilisées par nos prospects pour traiter les cas d’usages (Catégories de solutions du marché) sont:""")
     markdown_text7 = markdown_text7+"".join(f"\n- {o}" for o in solutions_utilisees)
@@ -159,7 +179,7 @@ Cette entreprise propose des {options3[selected_index3]}.
     st.write("")
     col1, col2, col3 = st.columns(3)
     with col1:
-        virulence = st.slider(tr("Virulence (choisissez une valeur entre 1 et 5)"), min_value=1, max_value=5, step=1)
+        virulence = st.slider(tr("Virulence (choisissez une valeur entre 1 et 5)"), min_value=1, max_value=5, step=1,value=virulence)
     markdown_text8 = tr(f"""\nTu vas utiliser une échelle de 1 à 5 de virulence. Pour cette simulation utilise le niveaux {virulence}""")
     
     human_message1 = tr("""Je souhaites que nous ayons une conversation verbale entre un commercial de mon entreprise,  et toi que je prospecte.
@@ -179,7 +199,7 @@ Attention: Ce n'est pas toi qui m'aide, c'est moi qui t'aide avec ma solution.
                 ------------------------------------------------------------------------------------
                 ''')
     
-    ai_message1 = tr(f"J'ai bien compris, je suis un {options[selected_index]} prospecté et je réponds seulement à tes questions. Je réponds à une seule question à la fois, sans commencer mes réponses par 'En tant que {options[selected_index]}'")   
+    ai_message1 = tr(f"J'ai bien compris, je suis un {options1[selected_index1]} prospecté et je réponds seulement à tes questions. Je réponds à une seule question à la fois, sans commencer mes réponses par 'En tant que {options1[selected_index1]}'")   
 
 
     
@@ -375,13 +395,12 @@ def run():
         # Ajouter un espace pour séparer les zones
         # st.divider()
             with col2:
-                # with st.container():
-                if query:
-                # Display chat messages from history on app rerun
-                
-                    for message in st.session_state.messages:
-                        with st.chat_message(message["role"]):
-                            st.markdown(message["content"])
+                if ("messages" in st.session_state) :
+                    if (st.session_state.messages != []):
+                    # Display chat messages from history on app rerun
+                        for message in st.session_state.messages:
+                            with st.chat_message(message["role"]):
+                                st.markdown(message["content"])
     else:
         st.write("")
         st.write("")