Update app.py

app.py

@@ -188,18 +188,25 @@ def interact_with_lucas(prompt, messages):
         messages[-1] = ChatMessage(role="assistant", content=response)
         yield messages
 
-def vote(data: gr.LikeData, history):
-    user_input = history[-1][0] if history else ""
-    feedback = {
+def vote(feedback: str, history):
+    if not history:
+        return "Aucun message à évaluer."
+    
+    last_user_message = history[-1][0]
+    last_bot_response = history[-1][1]
+    
+    feedback_data = {
         "timestamp": datetime.now().isoformat(),
-        "user_input": user_input,
-        "bot_response": data.value,
-        "liked": data.liked
+        "user_input": last_user_message,
+        "bot_response": last_bot_response,
+        "liked": feedback == "👍"
     }
+    
     api = HfApi()
     token = os.environ.get("HF_TOKEN")
     repo_id = "Woziii/llama-3-8b-chat-me"
     file_name = "feedback.json"
+    
     try:
         try:
             file_path = hf_hub_download(repo_id=repo_id, filename=file_name, token=token)
@@ -210,11 +217,14 @@ def vote(data: gr.LikeData, history):
         except Exception as e:
             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"
         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,
@@ -222,8 +232,11 @@ def vote(data: gr.LikeData, history):
             token=token
         )
         print(f"Feedback enregistré dans {repo_id}/{file_name}")
+        return "Merci pour votre feedback !"
     except Exception as e:
         print(f"Erreur lors de l'enregistrement du feedback : {str(e)}")
+        return "Erreur lors de l'enregistrement du feedback."
+
 
 
 def load_feedback_data():
@@ -280,8 +293,14 @@ with gr.Blocks() as demo:
             text_input = gr.Textbox(lines=1, label="Votre message")
             text_input.submit(interact_with_lucas, [text_input, chatbot], [chatbot])
             
-            # Ajout de l'événement de vote
-            chatbot.like(vote, inputs=[chatbot], outputs=[])
+            with gr.Row():
+                like_btn = gr.Button("👍")
+                dislike_btn = gr.Button("👎")
+            
+            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)
         
         with gr.Tab("Statistiques"):
             gr.Markdown("# Statistiques d'utilisation 📊")
@@ -299,4 +318,3 @@ demo.queue(max_size=20, default_concurrency_limit=2).launch(max_threads=10)
 
 
 
-