requirements and app updated

app.py

@@ -19,89 +19,85 @@ def load_model(model_name):
 
 def translate_text(model_name, text):
     """
-    Translate text after detecting complete sentences.
+    Translate input text using the specified model.
     """
-    if not model_name or not text.strip():
-        return "Select a model and provide text."
-
+    if not model_name or not text:
+        return "Please select a model and provide text for translation."
+    
     try:
-        # Load the selected model
+        # Load the model and tokenizer
         model, tokenizer = load_model(model_name)
-
-        # Tokenize the input text
+        
+        # Tokenize the text
         tokens = tokenizer(text, return_tensors="pt", padding=True)
         if torch.cuda.is_available():
             tokens = {k: v.to('cuda') for k, v in tokens.items()}
-
-        # Generate translated tokens
+        
+        # Generate translation
         translated = model.generate(**tokens)
-
-        # Decode translation
         return tokenizer.decode(translated[0], skip_special_tokens=True)
+    
     except Exception as e:
         return f"Error: {str(e)}"
 
-def process_input(model_name, text):
-    """
-    Process user input to detect completed sentences and translate in real-time.
-    """
-    sentences = text.strip().split('. ')
-    translations = []
-    for sentence in sentences:
-        if sentence.endswith('.') or sentence.endswith('!') or sentence.endswith('?'):
-            translations.append(translate_text(model_name, sentence))
-        else:
-            # If the sentence is incomplete, skip translation
-            translations.append(f"Waiting for completion: {sentence}")
-    return '\n'.join(translations)
-
-# Define Gradio Interface
-with gr.Blocks() as app:
-    gr.Markdown("## 🌍 Real-Time Sentence Translation")
-    gr.Markdown("### Enter text in the textbox, and it will be translated after each sentence ends!")
-
-    model_dropdown = gr.Dropdown(
-        label="Select Translation Model",
-        choices=[
-            "Helsinki-NLP/opus-mt-tc-big-en-tr",  # English to Turkish
-            "Helsinki-NLP/opus-mt-tc-big-tr-en",  # Turkish to English
-            "Helsinki-NLP/opus-mt-tc-big-en-fr",  # English to French
-            "Helsinki-NLP/opus-mt-tc-big-fr-en",  # French to English
-            "Helsinki-NLP/opus-mt-en-de",        # English to German
-            "Helsinki-NLP/opus-mt-de-en",        # German to English
-            "Helsinki-NLP/opus-mt-tc-big-en-es", # English to Spanish
-            "Helsinki-NLP/opus-mt-es-en",        # Spanish to English
-            "Helsinki-NLP/opus-mt-tc-big-en-ar", # English to Arabic
-            "Helsinki-NLP/opus-mt-tc-big-ar-en", # Arabic to English
-            "Helsinki-NLP/opus-mt-en-ur",        # English to Urdu
-            "Helsinki-NLP/opus-mt-ur-en",        # Urdu to English
-            "Helsinki-NLP/opus-mt-en-hi",        # English to Hindi
-            "Helsinki-NLP/opus-mt-hi-en",        # Hindi to English
-            "Helsinki-NLP/opus-mt-en-zh",        # English to Chinese
-            "Helsinki-NLP/opus-mt-zh-en",        # Chinese to English",
-        ],
-        value="Helsinki-NLP/opus-mt-tc-big-en-tr",
-        interactive=True
-    )
-
-    input_textbox = gr.Textbox(
-        label="Enter Text:",
-        placeholder="Type here...",
-        lines=5,
-        interactive=True
-    )
+# Model options
+model_options = [
+    ("English to Turkish", "Helsinki-NLP/opus-mt-tc-big-en-tr"),
+    ("Turkish to English", "Helsinki-NLP/opus-mt-tc-big-tr-en"),
+    ("English to French", "Helsinki-NLP/opus-mt-tc-big-en-fr"),
+    ("French to English", "Helsinki-NLP/opus-mt-tc-big-fr-en"),
+    ("English to German", "Helsinki-NLP/opus-mt-en-de"),
+    ("German to English", "Helsinki-NLP/opus-mt-de-en"),
+    ("English to Spanish", "Helsinki-NLP/opus-mt-tc-big-en-es"),
+    ("Spanish to English", "Helsinki-NLP/opus-mt-es-en"),
+    ("English to Arabic", "Helsinki-NLP/opus-mt-tc-big-en-ar"),
+    ("Arabic to English", "Helsinki-NLP/opus-mt-tc-big-ar-en"),
+    ("English to Urdu", "Helsinki-NLP/opus-mt-en-ur"),
+    ("Urdu to English", "Helsinki-NLP/opus-mt-ur-en"),
+    ("English to Hindi", "Helsinki-NLP/opus-mt-en-hi"),
+    ("Hindi to English", "Helsinki-NLP/opus-mt-hi-en"),
+    ("English to Chinese", "Helsinki-NLP/opus-mt-en-zh"),
+    ("Chinese to English", "Helsinki-NLP/opus-mt-zh-en"),
+]
 
-    output_textbox = gr.Textbox(
-        label="Translated Text:",
-        lines=5,
-        interactive=False
+# Create Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# 🌍 Real-Time Sentence Translation")
+    
+    with gr.Row():
+        model_dropdown = gr.Dropdown(
+            label="Select Translation Model",
+            choices=[option[1] for option in model_options],
+            type="value",
+        )
+    
+    with gr.Row():
+        input_text = gr.Textbox(
+            label="Enter text (complete sentences)",
+            lines=5,
+            placeholder="Type here...",
+        )
+    
+    with gr.Row():
+        translate_button = gr.Button("Translate")
+        clear_button = gr.Button("Clear")
+    
+    output_text = gr.Textbox(label="Translated Text", interactive=False)
+    
+    def clear_inputs():
+        return "", ""
+    
+    translate_button.click(
+        fn=translate_text,
+        inputs=[model_dropdown, input_text],
+        outputs=output_text,
     )
-
-    input_textbox.change(
-        fn=process_input,
-        inputs=[model_dropdown, input_textbox],
-        outputs=[output_textbox],
+    
+    clear_button.click(
+        fn=clear_inputs,
+        inputs=[],
+        outputs=[input_text, output_text],
     )
 
-# Launch Gradio App
-app.launch()
+# Run the Gradio app
+demo.launch()

requirements.txt

@@ -1,4 +1,5 @@
 transformers
 torch
 gradio
-sentencepiece
+sentencepiece
+sacremoses