Update app.py

app.py

@@ -120,6 +120,17 @@ def save_feedback(egyptian_input, generated_output, correct_incorrect, rating, c
     except Exception as e:
         return f"⚠️ Feedback could not be saved: {str(e)}"
 
+# ==============================================
+#  RESET UI — ONE ROUND TRIP INSTEAD OF FOUR
+# ==============================================
+def reset_feedback_ui():
+    return (
+        gr.update(visible=False),  # feedback_group
+        gr.update(value=""),       # correction_textbox
+        gr.update(value="Yes"),    # correct_radio
+        gr.update(value=None),     # rating_slider
+    )
+
 # ==============================================
 #  GRADIO INTERFACE
 # ==============================================
@@ -174,17 +185,8 @@ with gr.Blocks(title="SlangGPT Translator") as demo:
         inputs=[egyptian_input, latest_translation, correct_radio, rating_slider, correction_textbox],
         outputs=feedback_status
     ).then(
-        lambda: gr.update(visible=False),
-        outputs=feedback_group
-    ).then(
-        lambda: gr.update(value=""),
-        outputs=correction_textbox
-    ).then(
-        lambda: gr.update(value="Yes"),
-        outputs=correct_radio
-    ).then(
-        lambda: gr.update(value=None),
-        outputs=rating_slider
+        fn=reset_feedback_ui,
+        outputs=[feedback_group, correction_textbox, correct_radio, rating_slider]
     )
 
     gr.Markdown("### Examples")