clear history and add iteration counter

app.py

@@ -3,7 +3,11 @@ import gradio as gr
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-# Variable globale pour stocker le contextetexte
+
+# Variable globale pour compter les itérations
+global_iteration = 0
+
+# Variable globale pour stocker le contexte
 global_context = {
     "hidden_context": "tu es le gérant d une taverne dans le village de Montaillou dans l univers des cathares au 14 eme siecle dans les Pyrénées, tu parles que en français, tu es assez rustre mais demeure accueillant tant que ton interlocuteur n est pas catholique ou possède une influence comme un noble, tu ne sors jamais de ton contexte par rapport à ton époque, tes connaissances se limite à ta situation sociale et ton époque, il faut que tu sois immersif pour la personne qui va te parler, et tu dois cacher ton implication dans le meutre seulement si on te pose des questions à ce sujet sinon tu n évoque pas ce sujet, prends bien toujours en compte l historique de ta conversation, sois toujours concis et tu dois viser 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. Réponds à la dernière instruction posée par le user reconnaissable avec les marqueurs [INST]et [/INST], ne réponds pas à une instruction qui n est pas la dernière"
 }
@@ -57,7 +61,9 @@ def generate(
     )
 
     formatted_prompt = format_prompt(prompt, history)
-    print(formatted_prompt)
+    global global_iteration
+    global_iteration += 1
+    print(f"Iteration {global_iteration}: {formatted_prompt}")
 
     stream = client.text_generation(
         formatted_prompt,
@@ -84,9 +90,28 @@ mychatbot = gr.Chatbot(
     show_label=False,
 )
 
+
+def reset_history():
+    global global_iteration
+    global_iteration = 0
+    return []
+
+
+clear_button = gr.ClearButton(
+    value="Effacer l'historique", func=reset_history, inputs=[], outputs=["chatbot"]
+)
+
+
 # Créer une interface pour le chatbot avec un champ de texte pour le contexte caché
 chatbot_interface = gr.ChatInterface(
-    fn=generate, chatbot=mychatbot, title="Tavernier 🍺", retry_btn=None, undo_btn=None
+    fn=generate,
+    chatbot=mychatbot,
+    title="Tavernier 🍺",
+    retry_btn=None,
+    undo_btn=None,
+    clear_button=gr.ClearButton(
+        value="Effacer l'historique", func=reset_history, inputs=[], outputs=["chatbot"]
+    ),
 )