Update app.py

app.py

@@ -9,6 +9,7 @@ from datasets import Dataset
 from huggingface_hub import HfApi, login
 import time
 
+# Install required packages if not present
 from gradio_modal import Modal
 import huggingface_hub
 import datasets
@@ -131,7 +132,7 @@ def submit_research_feedback(conversation_history, satisfaction, feedback_text):
     # Get token from environment variable
     env_token = os.environ.get("HF_TOKEN")
     
-    # Use environment token, ignoring any passed token
+    # Use environment token
     push_success = push_feedback_to_hub(env_token)
     
     if push_success:
@@ -141,9 +142,8 @@ def submit_research_feedback(conversation_history, satisfaction, feedback_text):
     
     return status_msg
 
-# Remove the conversation_state as we'll access the chatbot history directly
+# Create the Gradio interface
 with gr.Blocks() as demo:
-    
     with gr.Row():
         with gr.Column(scale=3):
             chatbot = gr.ChatInterface(
@@ -155,13 +155,6 @@ with gr.Blocks() as demo:
                 type="messages"
             )
         
-            # Update conversation_state with each new message
-            chatbot.submit_btn.click(
-                lambda history: history,
-                inputs=[chatbot.chat_history],
-                outputs=[conversation_state]
-            )
-        
         with gr.Column(scale=1):
             report_button = gr.Button("Share Feedback", variant="primary")
     
@@ -171,7 +164,6 @@ with gr.Blocks() as demo:
             gr.Markdown("## Research Preview Feedback")
             gr.Markdown("Thank you for testing our research model. Your feedback (positive or negative) helps us improve!")
             
-            # Remove HF token input as we're using environment variables
             satisfaction = gr.Radio(
                 ["Very satisfied", "Satisfied", "Neutral", "Unsatisfied", "Very unsatisfied"],
                 label="How would you rate your experience with this research model?",
@@ -187,17 +179,17 @@ with gr.Blocks() as demo:
             submit_button = gr.Button("Submit Research Feedback", variant="primary")
             response_text = gr.Textbox(label="Status", interactive=False)
     
-    # Connect the "File a Report" button to show the modal
+    # Connect the "Share Feedback" button to show the modal
     report_button.click(
         lambda: Modal(visible=True),
         None,
         feedback_modal
     )
     
-    # Connect the submit button to the submit_research_feedback function
+    # Connect the submit button to the submit_research_feedback function with the current chat history
     submit_button.click(
-        submit_research_feedback,
-        inputs=[conversation_state, satisfaction, feedback_text],
+        lambda satisfaction, feedback_text: submit_research_feedback(chatbot.value, satisfaction, feedback_text),
+        inputs=[satisfaction, feedback_text],
         outputs=response_text
     )