Update app.py

app.py

@@ -188,12 +188,12 @@ def interact_with_lucas(prompt, messages):
         messages[-1] = ChatMessage(role="assistant", content=response)
         yield messages
         
-def vote(data: gr.LikeData):
-    feedback = {
+def vote(feedback: str, message: str, response: str):
+    feedback_data = {
         "timestamp": datetime.now().isoformat(),
-        "user_input": data.value["user_message"],
-        "bot_response": data.value["bot_message"],
-        "liked": data.liked
+        "user_input": message,
+        "bot_response": response,
+        "liked": feedback == "👍"
     }
     
     api = HfApi()
@@ -211,7 +211,7 @@ def vote(data: gr.LikeData):
         print(f"Erreur lors du téléchargement du fichier : {str(e)}")
         current_feedback = []
     
-    current_feedback.append(feedback)
+    current_feedback.append(feedback_data)
     updated_content = json.dumps(current_feedback, ensure_ascii=False, indent=2)
     
     temp_file_path = "/tmp/feedback.json"
@@ -228,6 +228,8 @@ def vote(data: gr.LikeData):
         print(f"Feedback enregistré dans {repo_id}/{file_name}")
     except Exception as e:
         print(f"Erreur lors de l'enregistrement du feedback : {str(e)}")
+    
+    return "Merci pour votre feedback !"
 
 def load_feedback_data():
     try:
@@ -282,6 +284,22 @@ with gr.Blocks() as demo:
             )
             text_input = gr.Textbox(lines=1, label="Votre message")
             text_input.submit(interact_with_lucas, [text_input, chatbot], [chatbot])
+            
+            with gr.Row():
+                like_btn = gr.Button("👍")
+                dislike_btn = gr.Button("👎")
+            
+            feedback_text = gr.Textbox(label="Feedback")
+            
+            def vote_callback(feedback, history):
+                if history:
+                    last_user_message = history[-1][0]
+                    last_bot_response = history[-1][1]
+                    return vote(feedback, last_user_message, last_bot_response)
+                return "Aucun message à évaluer."
+            
+            like_btn.click(lambda: vote_callback("👍", chatbot.value), outputs=feedback_text)
+            dislike_btn.click(lambda: vote_callback("👎", chatbot.value), outputs=feedback_text)
         
         with gr.Tab("Statistiques"):
             gr.Markdown("# Statistiques d'utilisation 📊")
@@ -295,9 +313,8 @@ with gr.Blocks() as demo:
         with gr.Tab("À propos"):
             gr.Markdown(Path(current_dir / "about.md").read_text())
 
-    chatbot.like(vote, None, None)
-
 demo.queue(max_size=20, default_concurrency_limit=2).launch(max_threads=10)
 
 
 
+