Update app.py

app.py

@@ -35,11 +35,12 @@ def extract_audio_from_video(video_path):
 
 def generate_output(file_obj, model_size, task, output_format, language):
     with tempfile.TemporaryDirectory() as tmpdirname:
-        # Save the uploaded file
+        # Save the uploaded file to a temporary directory
         file_name = os.path.join(tmpdirname, file_obj.name)
         with open(file_name, 'wb') as f:
-            f.write(file_obj.read())
+            f.write(file_obj.read())  # Read the file content and write it to the temp file
 
+        # Proceed with the rest of the processing using file_name
         # If it's a video file, extract the audio
         if is_video_file(file_name):
             audio_path = extract_audio_from_video(file_name)
@@ -56,8 +57,6 @@ def generate_output(file_obj, model_size, task, output_format, language):
             language=None if language == "Auto-detect" else language
         )
 
-        base_filename = os.path.splitext(file_name)[0]
-
         # Prepare the output file
         if 'Subtitles' in output_format:
             # Generate SRT content
@@ -71,13 +70,13 @@ def generate_output(file_obj, model_size, task, output_format, language):
                 subtitles.append(subtitle)
 
             srt_content = srt.compose(subtitles)
-            output_file = base_filename + '.srt'
+            output_file = file_name + '.srt'
             with open(output_file, "w", encoding='utf-8') as file:
                 file.write(srt_content)
         else:
             # Generate TXT content
             transcription_text = " ".join([segment['text'] for segment in result['segments']])
-            output_file = base_filename + '.txt'
+            output_file = file_name + '.txt'
             with open(output_file, "w", encoding='utf-8') as file:
                 file.write(transcription_text)
 
@@ -94,6 +93,7 @@ with gr.Blocks() as demo:
         file_input = gr.File(
             label="Upload Video or Audio File", 
             file_types=['video', 'audio']
+            type='file'
         )
 
     with gr.Row():