Update app.py

app.py

@@ -1,13 +1,44 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
-import csv
 import os
+import json
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
 from datetime import datetime
 
 # --- CONFIGURATION ---
 MODEL_K2H_REPO = "ankitklakra/kurukh-to-hindi"
 MODEL_H2K_REPO = "ankitklakra/hindi-to-kurukh"
-FEEDBACK_FILE = "feedback_log.csv"
+SHEET_NAME = "Kurukh_Feedback_Log" 
+
+# --- GOOGLE SHEETS SETUP ---
+def save_to_sheet(original, translation, correction, direction):
+    try:
+        # 1. Get credentials from Hugging Face Secrets
+        json_creds = os.getenv("GOOGLE_CREDENTIALS")
+        if not json_creds:
+            return "⚠️ Error: Google Credentials not found in Settings."
+        
+        creds_dict = json.loads(json_creds)
+        
+        # 2. Authenticate
+        scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
+        creds = ServiceAccountCredentials.from_json_keyfile_dict(creds_dict, scope)
+        client = gspread.authorize(creds)
+        
+        # 3. Open Sheet
+        sheet = client.open(SHEET_NAME).sheet1
+        
+        # 4. Add Headers if empty
+        if not sheet.get_all_values():
+            sheet.append_row(["Timestamp", "Direction", "Original Text", "AI Translation", "User Correction"])
+            
+        # 5. Append Row
+        sheet.append_row([str(datetime.now()), direction, original, translation, correction])
+        return "✅ Success! Saved to Google Sheets."
+        
+    except Exception as e:
+        return f"❌ Error saving to Sheet: {str(e)}"
 
 # --- LOAD RESOURCES ---
 print("Loading Tokenizer...")
@@ -20,7 +51,7 @@ model_h2k = AutoModelForSeq2SeqLM.from_pretrained(MODEL_H2K_REPO)
 pipe_k2h = pipeline("text2text-generation", model=model_k2h, tokenizer=tokenizer)
 pipe_h2k = pipeline("text2text-generation", model=model_h2k, tokenizer=tokenizer)
 
-# --- HELPER FUNCTIONS ---
+# --- TRANSLATION FUNCTION ---
 def translate_text(text, direction):
     if not text:
         return ""
@@ -31,25 +62,8 @@ def translate_text(text, direction):
     except Exception as e:
         return f"Error: {str(e)}"
 
-def save_feedback(original_text, translation, corrected_text, direction):
-    """Saves user corrections to a CSV file"""
-    if not corrected_text:
-        return "⚠️ Please enter a correction before submitting."
-    
-    file_exists = os.path.isfile(FEEDBACK_FILE)
-    
-    with open(FEEDBACK_FILE, mode='a', newline='', encoding='utf-8') as file:
-        writer = csv.writer(file)
-        if not file_exists:
-            writer.writerow(["Timestamp", "Direction", "Original", "AI_Translation", "User_Correction"])
-        writer.writerow([datetime.now(), direction, original_text, translation, corrected_text])
-    
-    return "✅ Thank you! Your correction has been saved."
-
 # --- THE UI ---
-
 with gr.Blocks() as demo:
-    
     gr.Markdown("# 🇮🇳 AI Kurukh (Kurux) Translator")
     gr.Markdown("### Preserving Tribal Languages with Artificial Intelligence")
     
@@ -57,50 +71,39 @@ with gr.Blocks() as demo:
         # TAB 1: TRANSLATOR
         with gr.TabItem("🗣️ Translator"):
             with gr.Row():
-                direction = gr.Radio(
-                    ["Kurukh -> Hindi", "Hindi -> Kurukh"], 
-                    label="Translation Mode", 
-                    value="Kurukh -> Hindi"
-                )
-            
+                direction = gr.Radio(["Kurukh -> Hindi", "Hindi -> Kurukh"], label="Translation Mode", value="Kurukh -> Hindi")
             with gr.Row():
-                with gr.Column():
-                    input_text = gr.Textbox(label="Input Text", placeholder="Type here...", lines=4)
-                    translate_btn = gr.Button("Translate 🚀", variant="primary")
-                
-                with gr.Column():
-                    output_text = gr.Textbox(label="AI Translation", lines=4, interactive=False)
+                input_text = gr.Textbox(label="Input Text", placeholder="Type here...", lines=4)
+                output_text = gr.Textbox(label="AI Translation", lines=4, interactive=False)
             
+            translate_btn = gr.Button("Translate 🚀", variant="primary")
             translate_btn.click(fn=translate_text, inputs=[input_text, direction], outputs=output_text)
             
             gr.Examples(
                 examples=[
                     ["निघै नामे इन्द्रा हिकै?", "Kurukh -> Hindi"],
-                    ["इन्गे अम्मो चि'आ।", "Kurukh -> Hindi"],
                     ["तुम्हारा नाम क्या है?", "Hindi -> Kurukh"]
                 ],
                 inputs=[input_text, direction]
             )
 
-        # TAB 2: FEEDBACK
+        # TAB 2: GOOGLE SHEETS FEEDBACK
         with gr.TabItem("📝 Improve the AI"):
-            gr.Markdown("### Help us make this model better!")
-            
+            gr.Markdown("### Help us improve!")
             with gr.Row():
                 fb_direction = gr.Radio(["Kurukh -> Hindi", "Hindi -> Kurukh"], label="Direction", value="Kurukh -> Hindi")
-            
             with gr.Row():
                 fb_original = gr.Textbox(label="Original Text")
                 fb_ai_output = gr.Textbox(label="What the AI said (Optional)")
-                fb_user_correct = gr.Textbox(label="What it SHOULD be (Your Correction)", lines=2)
+                fb_user_correct = gr.Textbox(label="Your Correction (Required)", lines=2)
             
-            submit_fb_btn = gr.Button("Submit Correction ✅")
-            fb_status = gr.Label(label="Status")
+            submit_btn = gr.Button("Submit")
+            status_lbl = gr.Label(label="Status")
             
-            submit_fb_btn.click(
-                fn=save_feedback,
+            submit_btn.click(
+                fn=save_to_sheet,
                 inputs=[fb_original, fb_ai_output, fb_user_correct, fb_direction],
-                outputs=fb_status
+                outputs=status_lbl
             )
 
 demo.launch()