Update app.py

app.py

@@ -1,5 +1,8 @@
 # app.py
 import gradio as gr
+import csv
+import os
+from datetime import datetime
 from classifier import classify_toxic_comment
 
 def clear_inputs():
@@ -255,8 +258,36 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
             semantic_similarity_display, empathy_score_display
         )
 
-    def handle_feedback(feedback, comment):
-        return f"Thank you for your feedback: {feedback}\nAdditional comment: {comment}"
+    def handle_feedback(feedback, additional_comment, comment, prediction, confidence):
+        """
+        Handle user feedback and store it in a CSV file.
+        """
+        if not feedback:
+            return "Please select a feedback option before submitting."
+
+        # Define the CSV file path
+        csv_file_path = "/home/user/app/feedback.csv"
+
+        # Check if the CSV file exists; if not, create it with headers
+        file_exists = os.path.isfile(csv_file_path)
+        with open(csv_file_path, mode='a', newline='', encoding='utf-8') as csv_file:
+            fieldnames = ['timestamp', 'comment', 'prediction', 'confidence', 'feedback', 'additional_comment']
+            writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
+
+            if not file_exists:
+                writer.writeheader()  # Write headers if the file is newly created
+
+            # Write the feedback to the CSV
+            writer.writerow({
+                'timestamp': datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+                'comment': comment,
+                'prediction': prediction,
+                'confidence': confidence,
+                'feedback': feedback,
+                'additional_comment': additional_comment if additional_comment else "N/A"
+            })
+
+        return f"Thank you for your feedback: {feedback}\nAdditional comment: {additional_comment if additional_comment else 'None'}\nFeedback has been saved."
 
     submit_btn.click(
         fn=lambda: (
@@ -296,7 +327,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
 
     feedback_submit.click(
         fn=handle_feedback,
-        inputs=[feedback_input, feedback_comment],
+        inputs=[feedback_input, feedback_comment, comment_input, prediction_output, confidence_output],
         outputs=feedback_output
     )