Update app.py

app.py

@@ -75,15 +75,22 @@ def process_translation(text, audio_input, direction, is_hinglish):
     original_text = speech_to_text(audio_input) if audio_input else text
     if not original_text: return "", "", None
 
-    # 2. Handle Hinglish (NEW FEATURE)
-    # Only applies if translating TO Kurukh (User typing Hindi in English letters)
+    # 2. Handle Hinglish
     if direction == "Hindi -> Kurukh" and is_hinglish:
         original_text = transliterate_to_hindi(original_text)
 
-    # 3. Translate
+    # 3. Translate 
     target_pipeline = pipe_k2h if direction == "Kurukh -> Hindi" else pipe_h2k
     try:
-        results = target_pipeline(original_text, max_length=128)
+        
+        results = target_pipeline(
+            original_text, 
+            max_length=128, 
+            num_beams=5,             
+            no_repeat_ngram_size=2,  
+            repetition_penalty=2.0,  
+            early_stopping=True
+        )
         translated_text = results[0]['generated_text']
     except Exception as e:
         return str(e), "", None
@@ -93,7 +100,6 @@ def process_translation(text, audio_input, direction, is_hinglish):
     if direction == "Kurukh -> Hindi":
         audio_output = text_to_speech(translated_text, "hi")
     
-    # Return: (Updated Input Box), (Translation), (Audio)
     return original_text, translated_text, audio_output
 
 # --- THE UI ---