Update app.py

app.py

@@ -13,14 +13,10 @@ def format_prompt(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-def generate_from_srt(file_path, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
-    # Assuming the file content is directly usable as a prompt
-    with open(file_path, "r", encoding="utf-8") as file:
-        file_content = file.loads()
-    
+def generate_from_srt(file_content, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
     # Process the SRT file content as needed before using it as a prompt
     # For example, extracting text and removing timestamps if necessary
-    # Here, directly using the file content for simplicity
+    # Directly using the file content for simplicity here
     
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -44,17 +40,19 @@ def generate_from_srt(file_path, temperature=0.9, max_new_tokens=256, top_p=0.95
         output += response.token.text
     return output
 
-def handle_file(file, **kwargs):
-    # Save the uploaded file temporarily to read it
-    file_path = file.name
-    with open(file_path, "wb") as f:
-        f.write(file.loads())
+def handle_file(file_info):
+    # Directly use the file content if it's a text file
+    if isinstance(file_info, str):
+        file_content = file_info
+    else:
+        # If file_info is not a string, it might be a binary file
+        file_content = file_info.decode('utf-8')
     
-    return generate_from_srt(file_path, **kwargs)
+    return generate_from_srt(file_content)
 
 iface = gr.Interface(
     fn=handle_file,
-    inputs=gr.File(label="Upload SRT File"),
+    inputs=gr.File(label="Upload SRT File", type="text"),  # Specify file type as text
     outputs="text",
     title="SRT File Translation",
     concurrency_limit=20,