Update app.py

app.py

@@ -30,13 +30,11 @@ LANGUAGES = {
     # Major Indian Languages
     "Hindi": "hi_IN",
     "Bengali": "bn_IN",
-    "Gujarati": "gu_IN",
+    "Gujarati": "gu_IN", 
     "Marathi": "mr_IN",
     "Tamil": "ta_IN",
     "Telugu": "te_IN",
     "Malayalam": "ml_IN",
-    "Punjabi": "pa_IN",  # Note: Using closest available in mBART
-    "Kannada": "kn_IN",  # Note: Using closest available in mBART
     "Urdu": "ur_PK"
 }
 
@@ -79,13 +77,24 @@ def save_as_pdf(text, output_path):
     pdf.add_page()
     pdf.set_font("Arial", size=12)
     
-    # Split text into lines and add to PDF
-    # Encode to handle unicode characters
-    encoded_text = text.encode('latin-1', 'replace').decode('latin-1')
-    pdf.multi_cell(0, 10, encoded_text)
+    # Handle encoding safely
+    try:
+        # Try UTF-8 first
+        encoded_text = text
+        pdf.multi_cell(0, 10, encoded_text)
+    except Exception:
+        try:
+            # Fall back to latin-1 with replacement
+            encoded_text = text.encode('latin-1', 'replace').decode('latin-1')
+            pdf.multi_cell(0, 10, encoded_text)
+        except Exception as e:
+            return f"Error creating PDF: {str(e)}"
     
-    pdf.output(output_path)
-    return output_path
+    try:
+        pdf.output(output_path)
+        return output_path
+    except Exception as e:
+        return f"Error saving PDF: {str(e)}"
 
 # Translation function
 def translate(text, source_lang, target_lang, max_length=1024):
@@ -128,6 +137,9 @@ def translate(text, source_lang, target_lang, max_length=1024):
 # Process uploads and handle translation
 def process_file(file, source_lang, target_lang):
     """Process uploaded file and translate its content"""
+    if file is None:
+        return None, "No file uploaded."
+        
     try:
         # Save uploaded file temporarily
         temp_file_path = file.name
@@ -146,62 +158,61 @@ def process_file(file, source_lang, target_lang):
         translated_text = translate(text, source_lang, target_lang)
         
         # Save translation as PDF
-        output_pdf_path = temp_file_path + "_translated.pdf"
-        save_as_pdf(translated_text, output_pdf_path)
+        output_pdf_path = os.path.join(os.path.dirname(temp_file_path), f"translated_{os.path.basename(temp_file_path)}.pdf")
+        result = save_as_pdf(translated_text, output_pdf_path)
         
+        if isinstance(result, str) and result.startswith("Error"):
+            return None, result
+            
         return output_pdf_path, translated_text
     
     except Exception as e:
         return None, f"Error processing file: {str(e)}"
 
-# Gradio interface
-def gradio_interface():
-    with gr.Blocks(title="Indian Language Translator", theme=gr.themes.Soft()) as interface:
-        gr.Markdown("# Indian & Global Language Translator")
-        gr.Markdown("Translate text with understanding of idioms and cultural expressions")
+# Create Gradio blocks
+with gr.Blocks(title="Indian Language Translator") as demo:
+    gr.Markdown("# Indian & Global Language Translator")
+    gr.Markdown("Translate text with understanding of idioms and cultural expressions")
+    
+    with gr.Tab("Text Translation"):
+        with gr.Row():
+            source_lang_text = gr.Dropdown(list(LANGUAGES.keys()), value="English", label="Source Language")
+            target_lang_text = gr.Dropdown(list(LANGUAGES.keys()), value="Hindi", label="Target Language")
         
-        with gr.Tab("Text Translation"):
-            with gr.Row():
-                source_lang_text = gr.Dropdown(list(LANGUAGES.keys()), value="English", label="Source Language")
-                target_lang_text = gr.Dropdown(list(LANGUAGES.keys()), value="Hindi", label="Target Language")
-            
-            with gr.Row():
-                input_text = gr.Textbox(label="Enter text to translate", lines=5, placeholder="Type or paste text here...")
-                output_text = gr.Textbox(label="Translation", lines=5)
-            
-            translate_btn = gr.Button("Translate Text", variant="primary")
-            translate_btn.click(
-                fn=translate,
-                inputs=[input_text, source_lang_text, target_lang_text],
-                outputs=output_text
-            )
+        with gr.Row():
+            input_text = gr.Textbox(label="Enter text to translate", lines=5, placeholder="Type or paste text here...")
+            output_text = gr.Textbox(label="Translation", lines=5)
         
-        with gr.Tab("Document Translation"):
-            with gr.Row():
-                source_lang_doc = gr.Dropdown(list(LANGUAGES.keys()), value="English", label="Source Language")
-                target_lang_doc = gr.Dropdown(list(LANGUAGES.keys()), value="Hindi", label="Target Language")
-            
-            file_input = gr.File(label="Upload Document (PDF, DOCX, TXT)", file_types=[".pdf", ".docx", ".txt"])
-            with gr.Row():
-                output_file = gr.File(label="Translated PDF")
-                output_preview = gr.Textbox(label="Translation Preview", lines=8)
-            
-            translate_doc_btn = gr.Button("Translate Document", variant="primary")
-            translate_doc_btn.click(
-                fn=process_file,
-                inputs=[file_input, source_lang_doc, target_lang_doc],
-                outputs=[output_file, output_preview]
-            )
+        translate_btn = gr.Button("Translate Text", variant="primary")
+        translate_btn.click(
+            fn=translate,
+            inputs=[input_text, source_lang_text, target_lang_text],
+            outputs=output_text
+        )
+    
+    with gr.Tab("Document Translation"):
+        with gr.Row():
+            source_lang_doc = gr.Dropdown(list(LANGUAGES.keys()), value="English", label="Source Language")
+            target_lang_doc = gr.Dropdown(list(LANGUAGES.keys()), value="Hindi", label="Target Language")
+        
+        file_input = gr.File(label="Upload Document (PDF, DOCX, TXT)", file_types=[".pdf", ".docx", ".txt"])
+        with gr.Row():
+            output_file = gr.File(label="Translated PDF")
+            output_preview = gr.Textbox(label="Translation Preview", lines=8)
         
-        gr.Markdown("### Supported File Types: PDF, DOCX, TXT")
-        gr.Markdown("### Features:")
-        gr.Markdown("- Supports major Indian languages including Hindi, Bengali, Tamil, Telugu, Malayalam")
-        gr.Markdown("- Context-aware translation that understands idioms and cultural expressions")
-        gr.Markdown("- Document translation with PDF output")
+        translate_doc_btn = gr.Button("Translate Document", variant="primary")
+        translate_doc_btn.click(
+            fn=process_file,
+            inputs=[file_input, source_lang_doc, target_lang_doc],
+            outputs=[output_file, output_preview]
+        )
     
-    return interface
+    gr.Markdown("### Supported File Types: PDF, DOCX, TXT")
+    gr.Markdown("### Features:")
+    gr.Markdown("- Supports major Indian languages including Hindi, Bengali, Tamil, Telugu, Malayalam")
+    gr.Markdown("- Context-aware translation that understands idioms and cultural expressions")
+    gr.Markdown("- Document translation with PDF output")
 
-# Launch the application
+# Launch the interface
 if __name__ == "__main__":
-    app = gradio_interface()
-    app.launch(share=True)  # Remove share=True in production
+    demo.launch(share=True)