Create app.py

app.py

@@ -0,0 +1,85 @@
+import gradio as gr
+from transformers import pipeline
+
+# --- CONFIGURATION ---
+
+MODEL_K2H = "ankitklakra/kurukh-to-hindi"
+MODEL_H2K = "ankitklakra/hindi-to-kurukh"
+
+# --- LOAD MODELS ---
+
+print("Loading Kurukh -> Hindi Model...")
+pipe_k2h = pipeline("text2text-generation", model=MODEL_K2H)
+
+print("Loading Hindi -> Kurukh Model...")
+pipe_h2k = pipeline("text2text-generation", model=MODEL_H2K)
+
+# --- TRANSLATION FUNCTION ---
+def translate_text(text, direction):
+    if not text:
+        return ""
+    
+    # Select the correct brain based on user choice
+    if direction == "Kurukh -> Hindi":
+        target_pipeline = pipe_k2h
+    else:
+        target_pipeline = pipe_h2k
+    
+    # Translate
+    # max_length=128 allow for longer sentences
+    results = target_pipeline(text, max_length=128)
+    return results[0]['generated_text']
+
+# --- THE USER INTERFACE ---
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    
+    # Header
+    gr.Markdown(
+        """
+        # 🇮🇳 AI Kurukh (Kurux) Translator
+        ### Preserving Tribal Languages with Artificial Intelligence
+        *Powered by Custom Fine-Tuned Google mT5 Models*
+        """
+    )
+    
+    # Input Section
+    with gr.Row():
+        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 your sentence here...", 
+                lines=5
+            )
+            translate_btn = gr.Button("Translate 🚀", variant="primary")
+        
+        with gr.Column():
+            output_text = gr.Textbox(
+                label="Translation Result", 
+                lines=5, 
+                show_copy_button=True # Allows users to copy result
+            )
+            
+    # Examples to help new users
+    gr.Examples(
+        examples=[
+            ["निघै नामे इन्द्रा हिकै?", "Kurukh -> Hindi"],
+            ["इन्गे अम्मो चि'आ।", "Kurukh -> Hindi"],
+            ["तुम्हारा नाम क्या है?", "Hindi -> Kurukh"],
+            ["मुझे पानी दो।", "Hindi -> Kurukh"]
+        ],
+        inputs=[input_text, direction],
+        label="Click on an example to test:"
+    )
+
+    # Logic Connection
+    translate_btn.click(fn=translate_text, inputs=[input_text, direction], outputs=output_text)
+
+# Launch
+demo.launch()