Update app.py

app.py

@@ -106,7 +106,10 @@ def format_prompt(message, history, user_id=""):
         if role == "user":
             prompt += f"[USER] {content} [/USER]"
         elif role == "bot":
-            prompt += f" [BOT]{filter_sensitive_info(content)}[/BOT] "
+            if content.startswith("[BOT]"):
+                prompt += f"{filter_sensitive_info(content)}[/BOT] "
+            else:
+                prompt += f" [BOT]{filter_sensitive_info(content)}[/BOT] "
     prompt += f"</s>[INST][USER] {message} [/USER][/INST]"
 
     # Ajouter le contexte statique et dynamique à chaque fois que l'utilisateur pose une question
@@ -115,6 +118,7 @@ def format_prompt(message, history, user_id=""):
     return prompt
 
 
+
 def get_random_string(length):
     # choose from all lowercase letter
     letters = string.ascii_lowercase
@@ -147,6 +151,8 @@ def generate(
     if request:
         # Récupérer l'identifiant unique de l'utilisateur à partir de la requête HTML et son historique de conversation
         user_id = dict(request.query_params).get("user_id")
+        if not user_id:
+            return "Vous devez vous connecter pour accéder au bot."
         history = load_history(user_id)
     else:
         user_id = ""
@@ -182,7 +188,7 @@ def generate(
 
 
 mychatbot = gr.Chatbot(
-    avatar_images=["./berger.jpg", "./pretre.jpg"],
+    avatar_images=["./berger.jpg", "./pretre.jpeg"],
     bubble_full_width=False,
     show_label=False,
 )