Update app.py

app.py

@@ -41,7 +41,7 @@ def analyze_sentiment(user_input: str):
         return f"Error analyzing sentiment: {e}", None
 
 # Function to save data
-def save_feedback(user_input: str, sentiment: str, confidence: float) -> str:
+def save_feedback(user_input: str, sentiment: str, confidence: float, user_feedback: str) -> str:
     try:
         with scheduler.lock:
             with JSON_DATASET_PATH.open("a") as f:
@@ -50,13 +50,14 @@ def save_feedback(user_input: str, sentiment: str, confidence: float) -> str:
                         "review": user_input,
                         "sentiment": sentiment,
                         "confidence": confidence,
+                        "user_feedback": user_feedback,
                         "datetime": datetime.now().isoformat(),
                     },
                     f,
                 )
                 f.write("\n")
         # Push the changes to Hugging Face Hub
-        scheduler.push_to_hub()  # No `commit_message` argument
+        scheduler.push_to_hub()
         return "Data saved successfully"
     except Exception as e:
         return f"Error saving data: {e}"
@@ -65,18 +66,27 @@ def save_feedback(user_input: str, sentiment: str, confidence: float) -> str:
 def handle_feedback(user_input):
     sentiment, confidence = analyze_sentiment(user_input)
     if confidence is None:  # Check for sentiment analysis errors
-        return sentiment
+        return sentiment, "", ""
     result_message = f"Sentiment: {sentiment}, Confidence: {confidence:.2f}"
-    save_status = save_feedback(user_input, sentiment, confidence)
-    if "Error" in save_status:
-        return save_status
-    return result_message
+    return result_message, sentiment, confidence
+
+def submit_feedback(user_input, sentiment, confidence, user_feedback):
+    if not user_feedback:
+        return "Please provide feedback before submitting."
+    save_status = save_feedback(user_input, sentiment, confidence, user_feedback)
+    return save_status
 
 with gr.Blocks() as demo:
     with gr.Row():
         user_input = gr.Textbox(label="Enter Your Review", placeholder="Type your review here...")
         output = gr.Textbox(label="Analysis Result", interactive=False)
-    submit_button = gr.Button("Analyze and Save")
-    submit_button.click(fn=handle_feedback, inputs=user_input, outputs=output)
+    with gr.Row():
+        user_feedback = gr.Textbox(label="Your Feedback on the Model's Response", placeholder="Type your feedback here...")
+    submit_button = gr.Button("Analyze")
+    feedback_submit_button = gr.Button("Submit Feedback")
+
+    # Connect functions to buttons
+    submit_button.click(fn=handle_feedback, inputs=user_input, outputs=[output, user_feedback, output])
+    feedback_submit_button.click(fn=submit_feedback, inputs=[user_input, user_feedback, output, user_feedback], outputs=output)
 
-demo.launch()
+demo.launch()