Update app.py

app.py

@@ -1,59 +1,39 @@
-from huggingface_hub import InferenceClient
 import gradio as gr
+from huggingface_hub import InferenceClient
 import re
 
+# Initialize the InferenceClient with the Mixtral model
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-def translate_subtitles(srt_content, target_language):
-    """
-    Translate the subtitles in the SRT file content to the target language.
-    """
-    # Split the SRT content into blocks
-    blocks = srt_content.split('\n\n')
-    translated_blocks = []
-    for block in blocks:
-        if block.strip() == "":
-            continue
-        lines = block.split('\n')
-        if len(lines) >= 3:
-            index = lines[0]
-            time_range = lines[1]
-            subtitle_text = '\n'.join(lines[2:])
-            # Translate the subtitle text
-            translation = client.translation(subtitle_text, target_language=target_language)
-            translated_text = translation[0]['translation_text']
-            translated_blocks.append(f"{index}\n{time_range}\n{translated_text}")
-    return '\n\n'.join(translated_blocks)
-
-def read_srt_file(file_path):
-    """
-    Read SRT file content.
-    """
-    with open(file_path, 'r', encoding='utf-8') as file:
-        return file.read()
-
-def save_translated_srt(content, output_path):
-    """
-    Save the translated subtitles to a new SRT file.
-    """
-    with open(output_path, 'w', encoding='utf-8') as file:
-        file.write(content)
-
-def translate_srt_interface(srt_file, target_language):
-    """
-    Gradio interface function to translate SRT file content.
-    """
-    srt_content = srt_file.read()
-    translated_content = translate_subtitles(srt_content, target_language)
-    return translated_content
-
+def translate_srt(file, target_language):
+    # Read the content of the SRT file
+    srt_content = file.read().decode("utf-8")
+    lines = srt_content.split('\n')
+    
+    translated_lines = []
+    for line in lines:
+        # Check if the line is a timestamp or a subtitle number
+        if re.match(r"^\d+$", line) or re.match(r"^\d{2}:\d{2}:\d{2},\d{3} --> \d{2}:\d{2}:\d{2},\d{3}$", line):
+            translated_lines.append(line)  # Copy timestamps and numbers directly
+        elif line.strip() == "":
+            translated_lines.append(line)  # Preserve empty lines for formatting
+        else:
+            # Translate the text line
+            response = client(inputs=line, parameters={"target_language": target_language})
+            translated_lines.append(response[0]["generated_text"])
+    
+    # Join the translated lines back into a single string
+    translated_srt_content = "\n".join(translated_lines)
+    return translated_srt_content
+
+# Gradio interface
 iface = gr.Interface(
-    fn=translate_srt_interface,
-    inputs=[gr.File(label="Upload SRT File"), gr.Textbox(label="Target Language Code", placeholder="Enter language code here, e.g., 'en' for English")],
-    outputs=gr.Textbox(),
+    fn=translate_srt,
+    inputs=[gr.inputs.File(label="Upload SRT File"), gr.inputs.Dropdown(["fr", "en", "es", "de", "it", "pt"], label="Target Language")],
+    outputs="text",
     title="SRT File Translator",
-    description="Translate SRT subtitle files to your desired language."
+    description="Translate SRT files to the selected language using Mixtral model."
 )
 
-
+# Launch the Gradio app
 iface.launch()