Update app.py

app.py

@@ -9,13 +9,11 @@ from gtts import gTTS
 import tempfile
 import requests
 
-
 # --- CONFIGURATION ---
 MODEL_K2H_REPO = "ankitklakra/kurukh-to-hindi"
 MODEL_H2K_REPO = "ankitklakra/hindi-to-kurukh"
 SHEET_NAME = "Kurukh_Feedback_Log"
 
-
 # --- LOAD MODELS ---
 print(f"🚀 Starting App with Gradio Version: {gr.__version__}")
 
@@ -30,20 +28,16 @@ pipe_h2k = pipeline("text2text-generation", model=model_h2k, tokenizer=tokenizer
 print("Loading Voice Model...")
 asr_pipeline = pipeline("automatic-speech-recognition", model="openai/whisper-tiny")
 
-
 # --- HELPER FUNCTIONS ---
 def transliterate_to_hindi(text):
     try:
-        url = (
-            "https://inputtools.google.com/request?text={}&itc=hi-t-i0-und&num=1"
-        )
+        url = "https://inputtools.google.com/request?text={}&itc=hi-t-i0-und&num=1"
         response = requests.get(url.format(text))
         result = response.json()
         return result[1][0][1][0]
     except:
         return text
 
-
 def save_to_sheet(original, translation, correction, direction):
     try:
         json_creds = os.getenv("GOOGLE_CREDENTIALS")
@@ -55,10 +49,9 @@ def save_to_sheet(original, translation, correction, direction):
             "https://spreadsheets.google.com/feeds",
             "https://www.googleapis.com/auth/drive",
         ]
-        creds = ServiceAccountCredentials.from_json_keyfile_dict(
-            creds_dict, scope
-        )
+        creds = ServiceAccountCredentials.from_json_keyfile_dict(creds_dict, scope)
         client = gspread.authorize(creds)
+
         sheet = client.open(SHEET_NAME).sheet1
 
         if not sheet.get_all_values():
@@ -73,56 +66,39 @@ def save_to_sheet(original, translation, correction, direction):
             )
 
         sheet.append_row(
-            [
-                str(datetime.now()),
-                direction,
-                original,
-                translation,
-                correction,
-            ]
+            [str(datetime.now()), direction, original, translation, correction]
         )
+
         return "✅ Saved to Google Sheets."
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-
 def speech_to_text(audio_path):
     if audio_path is None:
         return ""
     return asr_pipeline(audio_path)["text"]
 
-
 def text_to_speech(text, language="hi"):
     if not text:
         return None
     try:
         tts = gTTS(text=text, lang=language)
-        temp_file = tempfile.NamedTemporaryFile(
-            delete=False, suffix=".mp3"
-        )
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
         tts.save(temp_file.name)
         return temp_file.name
     except:
         return None
 
-
 # --- MAIN TRANSLATION LOGIC ---
 def process_translation(text, audio_input, direction, is_hinglish):
-    original_text = (
-        speech_to_text(audio_input) if audio_input else text
-    )
-
+    original_text = speech_to_text(audio_input) if audio_input else text
     if not original_text:
         return "", "", None
 
     if direction == "Hindi -> Kurukh" and is_hinglish:
         original_text = transliterate_to_hindi(original_text)
 
-    target_pipeline = (
-        pipe_k2h
-        if direction == "Kurukh -> Hindi"
-        else pipe_h2k
-    )
+    target_pipeline = pipe_k2h if direction == "Kurukh -> Hindi" else pipe_h2k
 
     try:
         results = target_pipeline(
@@ -144,71 +120,62 @@ def process_translation(text, audio_input, direction, is_hinglish):
     return original_text, translated_text, audio_output
 
 
-# --- CUSTOM CSS ---
+# --- CSS ---
 universal_css = """
 <style>
 @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600&display=swap');
-
-body, .gradio-container {
-    font-family: 'Poppins', sans-serif !important;
+body, button, input, select, textarea, .gradio-container { 
+    font-family: 'Poppins', sans-serif !important; 
 }
-
-.header-div {
-    text-align: center;
-    margin-bottom: 25px;
+.header-div { 
+    text-align: center; 
+    margin-bottom: 25px; 
     padding: 20px;
     background: linear-gradient(to right, #f8f9fa, #e9ecef);
     border-radius: 15px;
 }
-.header-title {
-    font-size: 2.2em;
-    font-weight: 700;
-    color: #2c3e50;
+.header-title { 
+    font-size: 2.2em; 
+    font-weight: 700; 
+    color: #2c3e50; 
 }
-.header-subtitle {
-    font-size: 1.1em;
-    color: #576574;
+.header-subtitle { 
+    font-size: 1.1em; 
+    color: #576574; 
 }
 </style>
 """
 
 
 # --- UI ---
-with gr.Blocks(
-    title="Kurukh AI Translator", theme=gr.themes.Soft()
-) as demo:
+with gr.Blocks(title="Kurukh AI Translator") as demo:
 
     gr.HTML(universal_css)
 
-    gr.HTML(
-        """
+    gr.HTML("""
     <div class="header-div">
         <h1 class="header-title">🇮🇳 AI Kurukh (Oraon) Translator</h1>
         <p class="header-subtitle">
             Bridging Communities with Artificial Intelligence | Voice & Hinglish Supported
         </p>
     </div>
-    """
-    )
+    """)
 
     with gr.Tabs():
 
-        # ---------------- TRANSLATOR TAB ----------------
-        with gr.Tab("🗣️ Translator"):
+        # --- Translator Tab ---
+        with gr.TabItem("🗣️ Translator"):
 
-            with gr.Accordion(
-                "ℹ️ How to use (Click to expand)", open=False
-            ):
-                gr.Markdown(
-                    """
-                1. **Select Mode:** Kurukh → Hindi OR Hindi → Kurukh  
-                2. **Hinglish:** Enable if you type Hindi in English  
-                3. **Voice:** Speak using the microphone (Hindi only)
-            """
-                )
+            with gr.Accordion("ℹ️ How to use (Click to expand)", open=False):
+                gr.Markdown("""
+                1. Select translation mode.
+                2. Enable Hinglish if typing Hindi in English letters.
+                3. Use Voice input if needed.
+                """)
 
             with gr.Row():
 
+                # LEFT
                 with gr.Column():
                     direction = gr.Radio(
                         ["Kurukh -> Hindi", "Hindi -> Kurukh"],
@@ -217,55 +184,42 @@ with gr.Blocks(
                     )
 
                     is_hinglish = gr.Checkbox(
-                        label="🔤 Hinglish Typing", value=False
+                        label="🔤 Hinglish Typing (e.g., 'Tumhara')", value=False
                     )
 
                     input_text = gr.Textbox(
-                        label="Enter Text",
-                        placeholder="Type sentences here...",
-                        lines=4,
+                        label="Enter Text", placeholder="Type sentences here...", lines=4
                     )
 
                     input_audio = gr.Audio(
-                        label="🎙️ Voice Input (Hindi Only)",
                         sources=["microphone"],
                         type="filepath",
+                        label="🎙️ Voice Input (Hindi Only)",
                     )
 
-                    translate_btn = gr.Button(
-                        "Translate 🚀", variant="primary"
-                    )
+                    translate_btn = gr.Button("Translate 🚀")
 
+                # RIGHT
                 with gr.Column():
                     output_text = gr.Textbox(
                         label="Translation",
                         lines=4,
                         interactive=False,
+                        show_copy_button=True,
                     )
-
                     output_audio = gr.Audio(
-                        label="🔊 Listen (Hindi Only)",
-                        interactive=False,
-                        autoplay=False,
+                        label="🔊 Listen (Hindi Only)", interactive=False
                     )
 
             translate_btn.click(
                 fn=process_translation,
-                inputs=[
-                    input_text,
-                    input_audio,
-                    direction,
-                    is_hinglish,
-                ],
+                inputs=[input_text, input_audio, direction, is_hinglish],
                 outputs=[input_text, output_text, output_audio],
             )
 
-        # ---------------- FEEDBACK TAB ----------------
-        with gr.Tab("📝 Improve the AI"):
-
-            gr.Markdown(
-                "### 🛠️ Help us improve accuracy by submitting corrections."
-            )
+        # --- Feedback Tab ---
+        with gr.TabItem("📝 Improve the AI"):
+            gr.Markdown("### 🛠️ Help us improve accuracy")
 
             fb_direction = gr.Radio(
                 ["Kurukh -> Hindi", "Hindi -> Kurukh"],
@@ -274,34 +228,23 @@ with gr.Blocks(
             )
 
             fb_original = gr.Textbox(label="Original Text")
-            fb_ai_output = gr.Textbox(
-                label="AI's Wrong Translation"
-            )
+            fb_ai_output = gr.Textbox(label="AI's Translation")
             fb_user_correct = gr.Textbox(
-                label="✅ Your Correct Translation",
-                lines=2,
-            )
-
-            submit_btn = gr.Button(
-                "Submit Correction to Database", variant="secondary"
+                label="Correct Translation", lines=2
             )
 
+            submit_btn = gr.Button("Submit Correction")
             status_lbl = gr.Label(label="Status")
 
             submit_btn.click(
                 fn=save_to_sheet,
-                inputs=[
-                    fb_original,
-                    fb_ai_output,
-                    fb_user_correct,
-                    fb_direction,
-                ],
+                inputs=[fb_original, fb_ai_output, fb_user_correct, fb_direction],
                 outputs=status_lbl,
             )
 
     gr.Markdown("---")
     gr.HTML(
-        "<center style='color:#888'>Built for the Kurukh Community • Powered by Google mT5 & Whisper</center>"
+        "<center style='color:#777;'>Built with ❤️ for the Kurukh Community</center>"
     )
 
 demo.launch()