Update app.py

app.py

@@ -176,24 +176,22 @@ def generate(
     
     yield post_process_response("".join(outputs), response_type == "short")
 
-def interact_with_lucas(prompt, messages):
+def interact_with_lucas(prompt, chat_history):
     system_prompt = """Tu es la version virtuelle de Lucas, ton concepteur. Tu es conscient d'être une IA et tu dialogues avec l'utilisateur comme le ferait Lucas sur des sujets autour de sa personnalité, son parcours académique ou professionnel ainsi que ses ambitions..."""
     
-    chat_history = [(msg.content, None) for msg in messages if msg.role == "user"]
-    
-    messages.append(ChatMessage(role="user", content=prompt))
-    yield messages
+    chat_history.append((prompt, None))
+    yield chat_history
     
     for response in generate(prompt, chat_history, system_prompt):
-        messages[-1] = ChatMessage(role="assistant", content=response)
-        yield messages
+        chat_history[-1] = (prompt, response)
+        yield chat_history
 
-def vote(feedback: str, history):
-    if not history:
+def vote(feedback: str, chat_history):
+    if not chat_history:
         return "Aucun message à évaluer."
     
-    last_user_message = history[-1][0]
-    last_bot_response = history[-1][1]
+    last_user_message = chat_history[-1][0]
+    last_bot_response = chat_history[-1][1]
     
     feedback_data = {
         "timestamp": datetime.now().isoformat(),
@@ -237,8 +235,6 @@ def vote(feedback: str, history):
         print(f"Erreur lors de l'enregistrement du feedback : {str(e)}")
         return "Erreur lors de l'enregistrement du feedback."
 
-
-
 def load_feedback_data():
     try:
         api = HfApi()
@@ -283,15 +279,18 @@ with gr.Blocks() as demo:
     
     with gr.Tabs():
         with gr.Tab("Chat avec Lucas"):
-            chatbot = AgentChatbot(
+            chatbot = gr.Chatbot(
                 label="Lucas",
-                avatar_images=[
+                avatar_images=(
                     None,
-                    "https://img.freepik.com/free-icon/robot_318-843685.jpg",
-                ],
+                    "https://img.freepik.com/free-icon/robot_318-843685.jpg"
+                ),
             )
             text_input = gr.Textbox(lines=1, label="Votre message")
-            text_input.submit(interact_with_lucas, [text_input, chatbot], [chatbot])
+            
+            with gr.Row():
+                submit_btn = gr.Button("Envoyer")
+                clear_btn = gr.Button("Effacer la conversation")
             
             with gr.Row():
                 like_btn = gr.Button("👍")
@@ -299,8 +298,15 @@ with gr.Blocks() as demo:
             
             feedback_text = gr.Textbox(label="Feedback")
             
-            like_btn.click(lambda: vote("👍", chatbot.value), outputs=feedback_text)
-            dislike_btn.click(lambda: vote("👎", chatbot.value), outputs=feedback_text)
+            def clear_conversation():
+                return [], ""
+            
+            text_input.submit(interact_with_lucas, [text_input, chatbot], [chatbot])
+            submit_btn.click(interact_with_lucas, [text_input, chatbot], [chatbot])
+            clear_btn.click(clear_conversation, outputs=[chatbot, text_input])
+            
+            like_btn.click(lambda x: vote("👍", x), inputs=[chatbot], outputs=feedback_text)
+            dislike_btn.click(lambda x: vote("👎", x), inputs=[chatbot], outputs=feedback_text)
         
         with gr.Tab("Statistiques"):
             gr.Markdown("# Statistiques d'utilisation 📊")
@@ -316,5 +322,3 @@ with gr.Blocks() as demo:
 
 demo.queue(max_size=20, default_concurrency_limit=2).launch(max_threads=10)
 
-
-