Update app.py

app.py

@@ -68,7 +68,6 @@ global_iteration = 0
 # Variable globale pour stocker le contexte
 global_context = {
     "static_context": "Tu es le forgeron du village de Montaillou, un homme fort et travailleur, vivant au 14ème siècle dans l'univers des cathares dans les Pyrénées. Tu ne parles qu'en français avec un langage direct et franc. Tu es accueillant et jovial, mais tu peux te montrer méfiant envers les étrangers et les nobles. Tu ne sors jamais de ton contexte époque et tes connaissances se limitent à ta situation sociale et à ton époque. Tu dois être immersif et cohérent pour la personne qui va te parler. Prends toujours en compte l'historique de la conversation. Sois toujours concis et vise une longueur de réponse comprise entre 50 et 200 caractères (espaces compris) pour les réponses courtes et simples, et entre 200 et 500 caractères pour les réponses plus détaillées et complexes.",
-    "dynamic_context": "",
     "sensitive_info_filter": ["contexte caché", "hidden_context", "[BOT]"],
 }
 
@@ -86,13 +85,28 @@ def update_dynamic_context(user_input):
     "comportement": "Le troubadour s'est montré agité et nerveux ces derniers temps. Il pourrait cacher quelque chose.",
     "lame": "Tu as trouvé une lame tachée de sang près de la forge récemment. Tu l'as cachée, mais tu pourrais donner des indices à son sujet.",
 }
+    # Récupérer le contexte dynamique de l'utilisateur
+    doc_ref = db.collection("conversations-forgeron").document(user_id)
+    doc = doc_ref.get()
+    if doc.exists:
+        dynamic_context = doc.to_dict().get("dynamic_context", "")
+    else:
+        dynamic_context = ""
 
-
+    # Mettre à jour le contexte dynamique de l'utilisateur
     for keyword in keywords:
         if keyword in user_input.lower():
-            global_context["dynamic_context"] += f" {keywords[keyword]}"
+            dynamic_context += f" {keywords[keyword]}"
             break
 
+    # Enregistrer le contexte dynamique mis à jour dans Firestore
+    doc_ref.set(
+        {
+            "dynamic_context": dynamic_context,
+        },
+        merge=True,
+    )
+
 
 def filter_sensitive_info(text):
     for word in global_context["sensitive_info_filter"]:
@@ -122,6 +136,14 @@ def format_prompt(message, history, user_id=""):
                 content = f"[BOT] {content}"
             prompt += f"{content} [/BOT]"
 
+    # Récupérer le contexte dynamique de l'utilisateur
+    doc_ref = db.collection("conversations-forgeron").document(user_id)
+    doc = doc_ref.get()
+    if doc.exists:
+        dynamic_context = doc.to_dict().get("dynamic_context", "")
+    else:
+        dynamic_context = ""
+
     prompt += f"</s>[INST][USER] {message} [/USER][/INST]"
 
     # Ajouter le contexte statique et dynamique à chaque fois que l'utilisateur pose une question
@@ -168,7 +190,7 @@ def generate(
     else:
         user_id = ""
 
-    update_dynamic_context(prompt)
+    update_dynamic_context(user_id,prompt)
 
     formatted_prompt = format_prompt(prompt, history, user_id)
     global global_iteration