Update app.py

app.py

@@ -1,8 +1,6 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 import re
-import random
-import string
 import firebase_admin
 from firebase_admin import credentials, firestore
 from datetime import datetime
@@ -71,11 +69,6 @@ global_context = {
     "sensitive_info_filter": ["contexte caché", "hidden_context", "[BOT]"],
 }
 
-# Créer une interface avec un champ de texte pour le contexte caché
-context_input = gr.Textbox(
-    label="Contexte caché", value=global_context["static_context"]
-)
-
 
 def update_dynamic_context(user_id,user_input):
     global global_context
@@ -151,15 +144,24 @@ def format_prompt(message, history, user_id=""):
 
     return prompt
 
+# Liste des noms des collections Firestore pour chaque espace
+collections = [
+    "conversations-tavernier",
+    "conversations-guerisseuse",
+    "conversations-troubadour",
+    "conversations-forgeron",
+    "conversations-pretre"
+]
 
-def get_random_string(length):
-    # choose from all lowercase letter
-    letters = string.ascii_lowercase
-    return "".join(random.choice(letters) for i in range(length))
+def delete_user_data(user_id):
+    for collection in collections:
+        doc_ref = db.collection(collection).document(user_id)
+        doc_ref.delete()
 
 
 def generate(
     prompt,
+    accuse_btn,
     history,
     request: gr.Request,
     temperature=0.2,
@@ -190,6 +192,22 @@ def generate(
     else:
         user_id = ""
 
+    if accuse_btn:  # Vérifier si le bouton "Accuser" a été cliqué
+        accused_character = "troubadour"  # Le personnage accusé (le troubadour dans ce cas)
+
+        if accused_character == "troubadour":  # Vérifier si le personnage accusé est le coupable
+            output = "Félicitations, tu as trouvé le coupable !"
+            win = True
+        else:
+            output = "Désolé, tu as accusé le mauvais personnage. Tu as perdu."
+            win = False
+
+        # Supprimer l'historique et le contexte dynamique de tous les personnages dans tous les espaces
+        delete_user_data(user_id)
+
+        # Réinitialiser le chatbot et l'interface utilisateur
+        return (output, gr.update(visible=False), gr.update(visible=False))
+
     update_dynamic_context(user_id,prompt)
 
     formatted_prompt = format_prompt(prompt, history, user_id)
@@ -234,10 +252,23 @@ def chatbot_interface(request: gr.Request):
         title=chat_name,
         retry_btn=None,
         undo_btn=None,
+        clear_btn=None,
         submit_btn="Parler",
+        inputs=[gr.Textbox(label="Message"), gr.Button("Accuser")],
         css="footer {visibility: hidden !important} .gradio-container {background-color: #2D4059 !important; color: #FFD460 !important;}",
         js=js_func
     )
+
+    def handle_output(output, chatbot, textbox, accuse_btn):
+        if isinstance(output, tuple):
+            output, chatbot_visible, accuse_btn_visible = output
+            chatbot.visible = chatbot_visible
+            accuse_btn.visible = accuse_btn_visible
+
+        return output, chatbot, textbox, accuse_btn
+
+    chatbot_interface.process(fn=handle_output, inputs=[chatbot_interface, chatbot_interface.inputs[0], chatbot_interface.inputs[1]])
+
     return chatbot_interface