Update app.py

app.py

@@ -36,23 +36,23 @@ def translate_text(text, source_language_code, target_language_code):
     
     return translated_text
 
-def translate_srt(file_content, source_language_code, target_language_code):
-    # Load the SRT content using pysrt and io.BytesIO
-    subs = pysrt.from_string(file_content.decode("utf-8"))
+def translate_srt(file_info, source_language_code, target_language_code):
+    # Assuming file_info is a dictionary with 'content' holding the file's bytes
+    file_content = file_info['content']  # Correctly access the bytes content of the file
+
+    # Use pysrt to load subtitles from the file content
+    subs = pysrt.open(io.BytesIO(file_content))
 
     # Translate each subtitle
     for sub in subs:
         translated_text = translate_text(sub.text, source_language_code, target_language_code)
         sub.text = translated_text
-    
-    # Convert the subtitles back to an SRT string
-    srt_content = '\n'.join(subs.to_string())
 
-    # Save the translated SRT content to a temporary file
+    # Save the translated subtitles to a temporary file
     output_path = "/mnt/data/translated_srt.srt"
     with open(output_path, "w", encoding="utf-8") as file:
-        file.write(srt_content)
-    
+        subs.save(file, encoding='utf-8')
+
     return output_path
 
 source_language_dropdown = gr.Dropdown(choices=language_options, label="Source Language")