Update app.py

app.py

@@ -17,14 +17,14 @@ db = firestore.client()
 
 def save_message(user_id, message, role="user"):
     # Enregistrer le message dans Firestore
-    doc_ref = db.collection("conversations").document(user_id)
+    doc_ref = db.collection("conversations-pretre").document(user_id)
     doc_ref.set(
         {
             "messages": firestore.ArrayUnion(
                 [
                     {
                         "role": role,
-                        "message": f"[{role.upper()}] {message} [/{role.upper()}]",
+                        "message": f"{message}",
                         "timestamp": datetime.now(),
                     }
                 ]
@@ -36,9 +36,9 @@ def save_message(user_id, message, role="user"):
 
 def load_history(user_id):
     # Charger l'historique de conversation de l'utilisateur
-    doc_ref = db.collection("conversations").document(user_id)
+    doc_ref = db.collection("conversations-tavernier").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,7 +107,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 == history[-1].get("message"):
+                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
@@ -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 = ""
@@ -180,7 +186,6 @@ def generate(
 
     return output
 
-
 mychatbot = gr.Chatbot(
     avatar_images=["./berger.jpg", "./tavernier.jpg"],
     bubble_full_width=False,
@@ -195,36 +200,17 @@ def chatbot_interface(request: gr.Request):
         title=chat_name,
         retry_btn=None,
         undo_btn=None,
-        examples=[
-            [
-                "Bonjour",
-                "Salutations, voyageur ! Bienvenue dans ma taverne. Que puis-je faire pour toi ?",
-            ]
-        ],
-        cache_examples=True,
+        submit_btn="Parler",
+        css="footer {visibility: hidden !important} .gradio-container {background-color: #2D4059 !important; color: #FFD460 !important;}"
+        
     )
     return chatbot_interface
 
 
 def main(request: gr.Request):
     user_chatbot_interface = chatbot_interface(request)
-    user_chatbot_interface.queue().launch()
+    user_chatbot_interface.queue().launch(show_api=False,inline=True)
 
 
 if __name__ == "__main__":
-    main(None)
-
-
-# def reset_history(chatbot):
-#     chatbot.history = []
-#     return []
-
-
-# def update_context(context):
-#     global global_context
-#     global_context["static_context"] = context
-#     global mychatbot
-#     mychatbot.history = reset_history(mychatbot)
-#     global global_iteration
-#     global_iteration = 0
-#     return None
+    main(None)