Make message dissapear after 5 seconds

app.py

@@ -115,7 +115,7 @@ def save_feedback(feedback_rating, feedback_text, latest_obfuscation):
     latest_obfuscation["feedback_rating"] = feedback_rating
     latest_obfuscation["feedback_text"] = feedback_text
     save_data(latest_obfuscation)
-    return "No Feedback Selected", ""
+    return "No Feedback Selected", "", gr.update(visible=True)
 
 @spaces.GPU
 def greet(input_text, length, function_words, grade_level, formality, sarcasm, voice, persuasive, descriptive, narrative, expository):
@@ -169,8 +169,10 @@ def greet(input_text, length, function_words, grade_level, formality, sarcasm, v
         with torch.no_grad(): 
             outputs = model.generate(**inputs, max_new_tokens=MAX_NEW_TOKENS, top_p = 0.95)
         response = tokenizer.decode(outputs[0, input_length:], skip_special_tokens=True).strip()
+        full_output = tokenizer.decode(outputs[0], skip_special_tokens=False)
     else:
         response = input_text # If no sliders passed, do not do anything
+        full_output = response
 
     # print_nvidia_smi() # Print GPU usage
 
@@ -193,6 +195,7 @@ def greet(input_text, length, function_words, grade_level, formality, sarcasm, v
         },
         "input": input_text,
         "output": response,
+        "full_output": full_output,
         "feedback_rating": "No Feedback Selected",
         "feedback_text": ""
     }
@@ -202,9 +205,12 @@ def greet(input_text, length, function_words, grade_level, formality, sarcasm, v
     
     return response, gr.update(interactive=True), gr.update(interactive=True), latest_obfuscation
 
-def reset_sliders():
+def auto_sliders():
     return [0.5] * 7 + [0] * 3
 
+def reset_sliders():
+    return [0] * 7 + [0] * 3
+
 def toggle_slider(checked, value):
     if checked:
         return gr.update(value=value, interactive=True)
@@ -247,7 +253,9 @@ with demo:
             )
             gr.Markdown("# 2) Style Element Sliders\n### Adjust the style element sliders to the desired levels to steer the obfuscation.")
             
-            reset_button = gr.Button("Choose slider values automatically (based on input text)")
+            with gr.Row():
+                auto_button = gr.Button("Choose slider values automatically (based on input text)")
+                reset_button = gr.Button("Reset slider values")
             
             sliders = []
             slider_values = [
@@ -297,7 +305,9 @@ with demo:
                 "<div style='text-align: center; color: red;'>⚠️ Please enter text before obfuscating. ⚠️</div>", visible=True
             )
             
+            auto_button.click(fn=auto_sliders, inputs=[], outputs=sliders)
             reset_button.click(fn=reset_sliders, inputs=[], outputs=sliders)
+
             input_text.change(fn=update_obfuscate_button, inputs=input_text, outputs=[obfuscate_button, warning_message])
             # Initialize the button and warning message state on page load
             demo.load(fn=update_obfuscate_button, inputs=input_text, outputs=[obfuscate_button, warning_message])
@@ -326,10 +336,14 @@ with demo:
                 outputs=[output, feedback_rating, feedback_text, latest_obfuscation])
 
             save_feedback_button = gr.Button("Save Feedback", interactive=False)
+
             feedback_warning_message = gr.Markdown(
                 "<div style='text-align: center; color: red;'>⚠️ Please provide feedback or a rating before submitting. ⚠️</div>", visible=True
             )
-            
+            confirmation_message = gr.Markdown(
+                "<div id='confirmation-message' style='text-align: center; color: green;'>🥳 Feedback has been submitted successfully! 🎊</div>", visible=False
+            )
+
             # Update the interactivity of the save_feedback_button based on feedback_rating and feedback_text
             feedback_rating.change(fn=update_save_feedback_button, inputs=[feedback_rating, feedback_text], outputs=[save_feedback_button, feedback_warning_message])
             feedback_text.change(fn=update_save_feedback_button, inputs=[feedback_rating, feedback_text], outputs=[save_feedback_button, feedback_warning_message])
@@ -337,8 +351,24 @@ with demo:
             save_feedback_button.click(
                 fn=save_feedback,
                 inputs=[feedback_rating, feedback_text, latest_obfuscation],
-                outputs=[feedback_rating, feedback_text]
+                outputs=[feedback_rating, feedback_text, confirmation_message]
+            )
+
+            hide_code = """
+            () => {
+                document.getElementById('confirmation-message').style.display = 'block';
+                setTimeout(() => {
+                    document.getElementById('confirmation-message').style.display = 'none';
+                }, 5000);
+            }
+            """ # Hides the message after 5 seconds
+            save_feedback_button.click(
+                fn=None,
+                inputs=[],
+                outputs=None,
+                js=hide_code
             )
+            
             # Initialize the save feedback button and warning message state on page load
             demo.load(fn=check_initial_feedback_state, inputs=[feedback_rating, feedback_text], outputs=[save_feedback_button, feedback_warning_message])