Update app.py

app.py

@@ -187,38 +187,34 @@ def interact_with_lucas(prompt, messages):
     for response in generate(prompt, chat_history, system_prompt):
         messages[-1] = ChatMessage(role="assistant", content=response)
         yield messages
-        
-def vote(feedback: str, message: str, response: str):
-    feedback_data = {
+
+def vote(data: gr.LikeData, history):
+    user_input = history[-1][0] if history else ""
+    feedback = {
         "timestamp": datetime.now().isoformat(),
-        "user_input": message,
-        "bot_response": response,
-        "liked": feedback == "👍"
+        "user_input": user_input,
+        "bot_response": data.value,
+        "liked": data.liked
     }
-    
     api = HfApi()
     token = os.environ.get("HF_TOKEN")
     repo_id = "Woziii/llama-3-8b-chat-me"
     file_name = "feedback.json"
-    
     try:
-        file_path = hf_hub_download(repo_id=repo_id, filename=file_name, token=token)
-        with open(file_path, "r", encoding="utf-8") as file:
-            current_feedback = json.load(file)
-        if not isinstance(current_feedback, list):
+        try:
+            file_path = hf_hub_download(repo_id=repo_id, filename=file_name, token=token)
+            with open(file_path, "r", encoding="utf-8") as file:
+                current_feedback = json.load(file)
+            if not isinstance(current_feedback, list):
+                current_feedback = []
+        except Exception as e:
+            print(f"Erreur lors du téléchargement du fichier : {str(e)}")
             current_feedback = []
-    except Exception as e:
-        print(f"Erreur lors du téléchargement du fichier : {str(e)}")
-        current_feedback = []
-    
-    current_feedback.append(feedback_data)
-    updated_content = json.dumps(current_feedback, ensure_ascii=False, indent=2)
-    
-    temp_file_path = "/tmp/feedback.json"
-    with open(temp_file_path, "w", encoding="utf-8") as temp_file:
-        temp_file.write(updated_content)
-    
-    try:
+        current_feedback.append(feedback)
+        updated_content = json.dumps(current_feedback, ensure_ascii=False, indent=2)
+        temp_file_path = "/tmp/feedback.json"
+        with open(temp_file_path, "w", encoding="utf-8") as temp_file:
+            temp_file.write(updated_content)
         api.upload_file(
             path_or_fileobj=temp_file_path,
             path_in_repo=file_name,
@@ -228,8 +224,7 @@ def vote(feedback: str, message: str, response: str):
         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:
@@ -285,21 +280,8 @@ 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)
+            # Ajout de l'événement de vote
+            chatbot.like(vote, inputs=[chatbot], outputs=[])
         
         with gr.Tab("Statistiques"):
             gr.Markdown("# Statistiques d'utilisation 📊")