Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from transformers import pipeline
-import tempfile, os, zipfile
+import tempfile, os, zipfile, traceback
 
 translator_cache = {}
 
@@ -26,13 +26,12 @@ def safe_translate(text, src, tgt):
             translator = get_translator(src, tgt)
             return translator(text, max_length=512)[0]["translation_text"]
         elif (src, tgt) == ("ja", "zh") or (src, tgt) == ("zh", "ja"):
-            # double translate via English
             mid = safe_translate(text, src, "en")
             return safe_translate(mid, "en", tgt)
         else:
             return f"[Unsupported: {src}->{tgt}]"
     except Exception as e:
-        return f"[Translation error: {e}]"
+        return f"[Translation error: {str(e)}]"
 
 def parse_srt(srt_text):
     blocks = srt_text.strip().split("\n\n")
@@ -49,35 +48,51 @@ def parse_srt(srt_text):
 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):
+def process_file(file_obj, src_lang, tgt_lang, output_dir, error_log):
     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 = []
 
-    subtitles = parse_srt(raw_text)
-    translated_subs = []
+        for idx, ts, txt in subtitles:
+            translated = safe_translate(txt, src_lang, tgt_lang)
+            bilingual = f"{txt}\n{translated}"
+            translated_subs.append((idx, ts, bilingual))
 
-    for idx, ts, txt in subtitles:
-        translated = safe_translate(txt, src_lang, tgt_lang)
-        bilingual = f"{txt}\n{translated}"
-        translated_subs.append((idx, ts, bilingual))
-
-    output_path = os.path.join(output_dir, os.path.basename(file_obj.name))
-    with open(output_path, "w", encoding="utf-8") as f:
-        f.write(reassemble_srt(translated_subs))
+        output_path = os.path.join(output_dir, os.path.basename(file_obj.name))
+        with open(output_path, "w", encoding="utf-8") as f:
+            f.write(reassemble_srt(translated_subs))
+    except Exception as e:
+        error_log.append(f"File {file_obj.name} failed: {str(e)}\n{traceback.format_exc()}")
 
 def batch_translate(files, src_lang, tgt_lang):
     tmp_dir = tempfile.mkdtemp()
+    error_log = []
+
     for file_obj in files:
-        process_file(file_obj, src_lang, tgt_lang, tmp_dir)
+        process_file(file_obj, src_lang, tgt_lang, tmp_dir, error_log)
 
     zip_path = os.path.join(tmp_dir, "translated_srt.zip")
-    with zipfile.ZipFile(zip_path, 'w') as zipf:
-        for name in os.listdir(tmp_dir):
-            if name.endswith(".srt"):
-                zipf.write(os.path.join(tmp_dir, name), arcname=name)
-    return zip_path
+    try:
+        with zipfile.ZipFile(zip_path, 'w') as zipf:
+            for name in os.listdir(tmp_dir):
+                path = os.path.join(tmp_dir, name)
+                if os.path.isfile(path) and name.endswith(".srt"):
+                    zipf.write(path, arcname=name)
+            if error_log:
+                log_path = os.path.join(tmp_dir, "log.txt")
+                with open(log_path, "w") as logf:
+                    logf.write("\n".join(error_log))
+                zipf.write(log_path, arcname="log.txt")
+        return zip_path
+    except Exception as e:
+        # fallback error
+        fail_zip = os.path.join(tmp_dir, "fail.zip")
+        with zipfile.ZipFile(fail_zip, 'w') as zipf:
+            with open(os.path.join(tmp_dir, "log.txt"), "w") as logf:
+                logf.write(f"ZIP error: {str(e)}\n\n{traceback.format_exc()}")
+            zipf.write(os.path.join(tmp_dir, "log.txt"), arcname="log.txt")
+        return fail_zip
 
 gr.Interface(
     fn=batch_translate,
@@ -88,5 +103,5 @@ gr.Interface(
     ],
     outputs=gr.File(label="Download Translated ZIP"),
     title="Batch SRT Translator (EN-ZH-JA)",
-    description="Upload .srt subtitle files and translate between English, Chinese, and Japanese. Dual-language output with original + translation. ZIP output.",
+    description="Upload .srt subtitle files and translate between English, Chinese, and Japanese. Dual-language output with original + translation. ZIP output. Errors will be logged.",
 ).launch()