Update chatbot_tab.py

tabs/chatbot_tab.py

@@ -74,10 +74,14 @@ workflow.add_edge("model", END)
 memory = MemorySaver()
 app = workflow.compile(checkpointer=memory)
 
-config = {"configurable": {"thread_id": thread_id}}
-
-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.
+@st.cache_data
+def init():
+    global config,context,human_message1,ai_message1,trimmer,language
+    
+    config = {"configurable": {"thread_id": thread_id}}
+    
+    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:  
@@ -110,20 +114,22 @@ 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.
 """
-ai_message1 = "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),
-    ]
+    messages = [
+        SystemMessage(content=context),
+        HumanMessage(content=human_message1),
+        AIMessage(content=ai_message1),
+        ]
     
     
-trimmer.invoke(messages)
-language = "French"
-st.write("Contexte: "+context+"\n")
-st.write("Human Message: "+human_message1+"\n")
-st.write("AI Message: "+ai_message1+"\n")
+    trimmer.invoke(messages)
+    language = "French"
+    st.write("Contexte: "+context+"\n")
+    st.write("Human Message: "+human_message1+"\n")
+    st.write("AI Message: "+ai_message1+"\n")
+
+init()
 
 def run():