Update app.py

app.py

@@ -39,6 +39,14 @@ def transliterate_to_hindi(text):
         return text
 
 def save_to_sheet(original, translation, correction, direction):
+    # --- VALIDATION CHECK ---
+   
+    if not original or not original.strip():
+        return "⚠️ Error: Original text is missing."
+    
+    if not correction or not correction.strip():
+        return "⚠️ Error: Please enter your correction before submitting."
+
     try:
         json_creds = os.getenv("GOOGLE_CREDENTIALS")
         if not json_creds:
@@ -54,6 +62,7 @@ def save_to_sheet(original, translation, correction, direction):
 
         sheet = client.open(SHEET_NAME).sheet1
 
+        # Check if header exists, if not add it
         if not sheet.get_all_values():
             sheet.append_row(
                 [
@@ -245,5 +254,11 @@ with gr.Blocks(title="Kurukh AI Translator") as demo:
     gr.HTML(
         "<center style='color:#777;'>Built with ❤️ for the Kurukh Community</center>"
     )
+    
+    translate_btn.click(
+        fn=lambda orig, trans, direc: (orig, trans, direc),
+        inputs=[input_text, output_text, direction],
+        outputs=[fb_original, fb_ai_output, fb_direction]
+    )
 
 demo.launch()