Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
 import tempfile
-from transformers import pipeline, MarianMTModel, MarianTokenizer
+from transformers import pipeline
 from deep_translator import GoogleTranslator
 from langdetect import detect
 from fpdf import FPDF
@@ -30,6 +30,9 @@ def detect_language(text):
         return "en"  # Default to English if detection fails
 
 def translate_text(text, target_language):
+    if not text or not target_language:
+        return "No text to translate"
+        
     source_lang = detect_language(text)
     
     # Skip translation if source and target are the same
@@ -42,20 +45,16 @@ def translate_text(text, target_language):
         translation = translator.translate(text)
         return translation
     except Exception as e:
-        # Fallback to Hugging Face models if Google fails
-        try:
-            model_name = f"Helsinki-NLP/opus-mt-{source_lang}-{LANGUAGES[target_language]}"
-            translator = pipeline("translation", model=model_name)
-            translation = translator(text, max_length=500)
-            return translation[0]['translation_text']
-        except:
-            return f"Translation failed: {str(e)}"
+        return f"Translation failed: {str(e)}"
 
 def extract_text_from_document(file_path):
+    if not file_path:
+        return ""
+        
     _, file_extension = os.path.splitext(file_path)
     
     if file_extension.lower() == '.txt':
-        with open(file_path, 'r', encoding='utf-8') as file:
+        with open(file_path, 'r', encoding='utf-8', errors='replace') as file:
             return file.read()
     elif file_extension.lower() in ['.docx', '.doc']:
         return docx2txt.process(file_path)
@@ -71,7 +70,10 @@ def text_to_pdf(text, output_path):
     lines = text.split('\n')
     for line in lines:
         # Handle non-ASCII characters
-        pdf.multi_cell(0, 10, line.encode('latin-1', 'replace').decode('latin-1'))
+        try:
+            pdf.multi_cell(0, 10, line.encode('latin-1', 'replace').decode('latin-1'))
+        except Exception:
+            pdf.multi_cell(0, 10, "[Text contains unsupported characters]")
     
     pdf.output(output_path)
     return output_path
@@ -80,8 +82,8 @@ def translate_and_save(input_text, input_file, target_language):
     # Determine input source (text or file)
     if input_text:
         text_to_translate = input_text
-    elif input_file:
-        text_to_translate = extract_text_from_document(input_file.name)
+    elif input_file is not None:
+        text_to_translate = extract_text_from_document(input_file)
     else:
         return None, "Please provide either text or a document for translation."
     
@@ -96,27 +98,35 @@ def translate_and_save(input_text, input_file, target_language):
     
     return pdf_path, translated_text
 
-# Create Gradio interface
+# Create Gradio interface with proper component configurations
 with gr.Blocks(title="Context-Aware Translation Tool") as demo:
     gr.Markdown("# Context-Aware Language Translation")
     gr.Markdown("This tool translates text while preserving context, idioms, and phrases.")
     
     with gr.Row():
         with gr.Column():
+            # Input components with explicit types
             input_text = gr.Textbox(label="Enter text to translate", lines=5)
-            input_file = gr.File(label="Or upload a document (.txt, .docx)")
-            target_language = gr.Dropdown(label="Target Language", choices=list(LANGUAGES.keys()), value="English")
+            input_file = gr.File(label="Or upload a document (.txt, .docx)", type="filepath")
+            target_language = gr.Dropdown(
+                label="Target Language", 
+                choices=list(LANGUAGES.keys()), 
+                value="English"
+            )
             translate_button = gr.Button("Translate")
         
         with gr.Column():
+            # Output components with explicit types
             output_text = gr.Textbox(label="Translation", lines=5)
             output_pdf = gr.File(label="Download as PDF")
     
+    # Connect the components with properly typed inputs and outputs
     translate_button.click(
         fn=translate_and_save,
         inputs=[input_text, input_file, target_language],
         outputs=[output_pdf, output_text]
     )
 
+# Launch with specific options to avoid errors
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(show_error=True)