Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from transformers import pipeline
-import tempfile, os, zipfile, shutil
+import tempfile, os, zipfile
 
 translator_cache = {}
 
@@ -21,14 +21,17 @@ def get_translator(src_lang, tgt_lang):
     return translator_cache[key]
 
 def translate_text(text, src, tgt):
-    if (src, tgt) in MODEL_MAP:
-        translator = get_translator(src, tgt)
-        return translator(text, max_length=512)[0]["translation_text"]
-    elif (src == "zh" and tgt == "ja") or (src == "ja" and tgt == "zh"):
-        mid = translate_text(text, src, "en")
-        return translate_text(mid, "en", tgt)
-    else:
-        raise ValueError("Unsupported language pair")
+    try:
+        if (src, tgt) in MODEL_MAP:
+            translator = get_translator(src, tgt)
+            return translator(text, max_length=512)[0]["translation_text"]
+        elif (src == "zh" and tgt == "ja") or (src == "ja" and tgt == "zh"):
+            mid = translate_text(text, src, "en")
+            return translate_text(mid, "en", tgt)
+        else:
+            return f"[Unsupported language pair: {src}→{tgt}]"
+    except Exception as e:
+        return f"[Translation error: {e}]"
 
 def parse_srt(srt_text):
     blocks = srt_text.strip().split("\n\n")
@@ -47,15 +50,16 @@ def reassemble_srt(subtitles):
     return "\n\n".join(f"{idx}\n{ts}\n{txt}" for idx, ts, txt in subtitles)
 
 def process_file(file_obj, src_lang, tgt_lang, output_dir):
-    raw_text = file_obj.read().decode("utf-8")
+    try:
+        raw_text = file_obj.read().decode("utf-8")
+    except:
+        raw_text = file_obj.read().decode("utf-8", errors="ignore")
+
     subtitles = parse_srt(raw_text)
 
     translated_subs = []
     for idx, ts, txt in subtitles:
-        try:
-            translated = translate_text(txt, src_lang, tgt_lang)
-        except Exception as e:
-            translated = f"[Error: {e}]"
+        translated = translate_text(txt, src_lang, tgt_lang)
         translated_subs.append((idx, ts, translated))
 
     output_path = os.path.join(output_dir, os.path.basename(file_obj.name))