request html

app.py

@@ -54,7 +54,7 @@ def remove_length_info(response_text):
     return re.sub(pattern, "", response_text)
 
 
-def format_prompt(message, history):
+def format_prompt(message, history, user_id=""):
     max_history_length = 10  # Limiter la longueur de l'historique
     history = history[-max_history_length:]
 
@@ -79,6 +79,7 @@ def format_prompt(message, history):
 def generate(
     prompt,
     history,
+    user_id,
     temperature=0.2,
     max_new_tokens=256,
     top_p=0.95,
@@ -100,10 +101,10 @@ def generate(
 
     update_dynamic_context(prompt)
 
-    formatted_prompt = format_prompt(prompt, history)
+    formatted_prompt = format_prompt(prompt, history, user_id)
     global global_iteration
     global_iteration += 1
-    print(f"\n\nIteration {global_iteration}: {formatted_prompt}")
+    print(f"\n\nIteration {global_iteration}/{user_id}: {formatted_prompt}")
 
     stream = client.text_generation(
         formatted_prompt,
@@ -129,44 +130,66 @@ mychatbot = gr.Chatbot(
 )
 
 
+def chatbot_interface(request):
+    # Récupérer l'identifiant unique de l'utilisateur à partir de la requête HTML
+    user_id = request.query_params.get("user_id", None)
+
+    # Créer l'interface du chatbot avec l'identifiant unique de l'utilisateur en tant qu'entrée
+    chatbot = gr.Chatbot(
+        avatar_images=["./berger.jpg", "./tavernier.jpg"],
+        bubble_full_width=False,
+        show_label=False,
+    )
+    chatbot_interface = gr.Interface(
+        fn=generate,
+        inputs=[user_id],
+        chatbot=chatbot,
+        title="Mon chatbot",
+        retry_btn=None,
+        undo_btn=None,
+    )
+    return chatbot_interface
+
+
 # Créer une interface pour le chatbot avec un champ de texte pour le contexte caché
 
 
-def reset_history(chatbot):
-    chatbot.history = []
-    return []
+# 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
+# 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
 
 
-chatbot_interface = gr.ChatInterface(
-    fn=generate,
-    chatbot=mychatbot,
-    title=chat_name,
-    retry_btn=None,
-    undo_btn=None,
-)
+# chatbot_interface = gr.ChatInterface(
+#     fn=generate,
+#     chatbot=mychatbot,
+#     title=chat_name,
+#     retry_btn=None,
+#     undo_btn=None,
+# )
 
 
 # Ajouter le champ de texte pour le contexte caché à l'interface
-context_interface = gr.Interface(
-    fn=update_context,
-    inputs=[context_input],
-    outputs=None,
-    title="Modifier le contexte caché",
-    description="Modifie le contexte caché du tavernier",
-    article=False,
-)
+# context_interface = gr.Interface(
+#     fn=update_context,
+#     inputs=[context_input],
+#     outputs=None,
+#     title="Modifier le contexte caché",
+#     description="Modifie le contexte caché du tavernier",
+#     article=False,
+# )
 
 # demo = gr.TabbedInterface([chatbot_interface, context_interface], title="Tavernier 🍺")
 
-# Afficher l'interface
-chatbot_interface.queue().launch()
+# Lancer et afficher l'interface
+if __name__ == "__main__":
+    chatbot_interface.queue().launch()