Update app.py

app.py

@@ -4,8 +4,8 @@ import gradio as gr
 # Initialize the InferenceClient with the Mixtral model
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-def translate_srt(srt_file, source_lang, target_lang):
-    srt_content = srt_file.read().decode("utf-8")
+def translate_srt(srt_content, source_lang, target_lang):
+    # Assuming srt_content is now a string with the content of the SRT file
     translated_lines = []
     
     # Split the SRT file into lines
@@ -16,19 +16,21 @@ def translate_srt(srt_file, source_lang, target_lang):
         if line.strip().isdigit() or "-->" in line or line == "":
             translated_lines.append(line)
         else:
-            # Translate the text line
-            response = client(text=line, parameters={"src_lang": source_lang, "tgt_lang": target_lang})
+            # Prepare the text for translation
+            formatted_prompt = f"Translate the following text from {source_lang} to {target_lang}: {line}"
+            # Using the Mixtral API for translation; adjust parameters as necessary
+            response = client(text=formatted_prompt)
+            # Assuming the response includes a 'generated_text' field with the translated text
             translated_text = response["generated_text"]
             translated_lines.append(translated_text)
     
-    translated_srt = "\n".join(translated_lines)
-    return translated_srt
+    return "\n".join(translated_lines)
 
 # Define the Gradio interface
 iface = gr.Interface(
     fn=translate_srt,
     inputs=[
-        gr.File(label="Upload SRT File"),
+        gr.TextArea(label="SRT File Content"),
         gr.Textbox(label="Source Language (ISO Code)", placeholder="e.g., en for English"),
         gr.Textbox(label="Target Language (ISO Code)", placeholder="e.g., es for Spanish")
     ],