Update app.py

app.py

@@ -38,7 +38,7 @@ def load_history(user_id):
     # Charger l'historique de conversation de l'utilisateur
     doc_ref = db.collection("conversations-pretre").document(user_id)
     doc = doc_ref.get()
-    if doc:
+    if doc.exists:
         messages = doc.to_dict().get("messages", [])
         return messages
     else:
@@ -60,6 +60,7 @@ global_context = {
     "sensitive_info_filter": [
         "contexte caché",
         "hidden_context",
+        "[BOT]"
     ],
 }
 
@@ -106,10 +107,10 @@ def format_prompt(message, history, user_id=""):
         if role == "user":
             prompt += f"[USER] {content} [/USER]"
         elif role == "bot":
-            if content.startswith("[BOT]"):
-                prompt += f"{filter_sensitive_info(content)}[/BOT] "
-            else:
-                prompt += f" [BOT]{filter_sensitive_info(content)}[/BOT] "
+            if not content.startswith("[BOT]"):
+                content = f"[BOT] {content}"
+            prompt += f"{content} [/BOT]"
+
     prompt += f"</s>[INST][USER] {message} [/USER][/INST]"
 
     # Ajouter le contexte statique et dynamique à chaque fois que l'utilisateur pose une question
@@ -118,7 +119,6 @@ def format_prompt(message, history, user_id=""):
     return prompt
 
 
-
 def get_random_string(length):
     # choose from all lowercase letter
     letters = string.ascii_lowercase