bugfix context interface

app.py

@@ -1,5 +1,5 @@
-from huggingface_hub import InferenceClient
 import gradio as gr
+from huggingface_hub import InferenceClient
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
@@ -8,17 +8,11 @@ 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"
 }
 
-# Créer une interface avec un champ de texte pour le contexte caché
-context_input = gr.Textbox(
-    label="Contexte caché", value=global_context["hidden_context"]
-)
-context_submit = gr.Button("Mettre à jour le contexte")
-
 
-def update_context(context):
+def update_context(context, message, history, **kwargs):
     global global_context
     global_context["hidden_context"] = context
-    return None
+    return generate(message, history, **kwargs)
 
 
 def format_prompt(message, history):
@@ -85,23 +79,44 @@ mychatbot = gr.Chatbot(
     show_label=False,
 )
 
-# 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
-)
-
-
-# Ajouter le champ de texte pour le contexte caché à l'interface
-context_interface = gr.Interface(
+chatbot_interface = gr.Interface(
     fn=update_context,
-    inputs=[context_input, context_submit],
-    outputs=None,
-    title="Modifier le contexte caché",
-    description="Modifie le contexte caché du tavernier",
+    inputs=[
+        gr.Textbox(label="Contexte caché", value=global_context["hidden_context"]),
+        gr.Textbox(label="Message"),
+        gr.Chatbot(
+            avatar_images=["./berger.jpg", "./tavernier.jpg"],
+            bubble_full_width=False,
+            show_label=False,
+        ),
+        gr.Number(
+            label="Temperature", value=0.2, interactive=True, minimum=0.01, maximum=1.0
+        ),
+        gr.Number(
+            label="Max new tokens", value=256, interactive=True, minimum=1, maximum=1024
+        ),
+        gr.Number(
+            label="Top p", value=0.95, interactive=True, minimum=0.0, maximum=1.0
+        ),
+        gr.Number(
+            label="Repetition penalty",
+            value=1.0,
+            interactive=True,
+            minimum=0.0,
+            maximum=2.0,
+        ),
+    ],
+    outputs=gr.Chatbot(
+        avatar_images=["./berger.jpg", "./tavernier.jpg"],
+        bubble_full_width=False,
+        show_label=False,
+    ),
+    title="Tavernier 🍺",
+    retry_btn=None,
+    undo_btn=None,
+    description="Discute avec le tavernier en modifiant le contexte caché à tout moment.",
     theme="huggingface",
     article=False,
 )
 
-# Afficher l'interface
-context_interface.launch(inline=True)
-chatbot_interface.launch(inline=True)
+chatbot_interface.launch()