Save feedback

app.py

@@ -2,15 +2,34 @@ import spaces
 import gradio as gr
 import torch
 from transformers import AutoTokenizer
+import json
 
 tokenizer = AutoTokenizer.from_pretrained("gpt2")
 
+# Global variable to store the latest obfuscation result
+latest_obfuscation = {}
+
 @spaces.GPU
 def temp(text):
     response = tokenizer(input_text, return_tensors="pt")
     return response
 
+def save_data(data):
+    with open("user_data.json", "a") as f:
+        json.dump(data, f)
+        f.write("\n")
+
+def save_feedback(feedback_rating, feedback_text):
+    global latest_obfuscation
+    if latest_obfuscation:
+        latest_obfuscation['feedback'] = {
+            "rating": feedback_rating,
+            "text": feedback_text
+        }
+        save_data(latest_obfuscation)
+
 def greet(input_text, length, function_words, grade_level, sarcasm, formality, voice, persuasive, descriptive, narrative, expository):
+    global latest_obfuscation
     response = (
         f"Hello!\n"
         f"Input Text: {input_text}\n"
@@ -25,6 +44,32 @@ def greet(input_text, length, function_words, grade_level, sarcasm, formality, v
         f"Narrative: {narrative}\n"
         f"Expository: {expository}"
     )
+    
+    # Save the new obfuscation result and reset feedback
+    latest_obfuscation = {
+        "input_text": input_text,
+        "sliders": {
+            "length": length,
+            "function_words": function_words,
+            "grade_level": grade_level,
+            "sarcasm": sarcasm,
+            "formality": formality,
+            "voice": voice,
+            "persuasive": persuasive,
+            "descriptive": descriptive,
+            "narrative": narrative,
+            "expository": expository
+        },
+        "output": response,
+        "feedback": {
+            "rating": "No Feedback Selected",
+            "text": ""
+        }
+    }
+    
+    # Save the obfuscation result
+    save_data(latest_obfuscation)
+    
     return response
 
 def reset_sliders():
@@ -119,4 +164,17 @@ with demo:
             gr.Markdown("### Provide any feedback on the obfuscation")
             feedback_text = gr.Textbox(label="Feedback", lines=3)
 
+            # Trigger save_feedback on feedback change
+            feedback_rating.change(
+                fn=save_feedback, 
+                inputs=[feedback_rating, feedback_text],
+                outputs=[]
+            )
+
+            feedback_text.change(
+                fn=save_feedback, 
+                inputs=[feedback_rating, feedback_text],
+                outputs=[]
+            )
+
 demo.launch()