Update app.py

app.py

@@ -36,41 +36,38 @@ aspect_ratio = gr.inputs.Dropdown(choices=[
 
 video_url = gr.inputs.Textbox(label="Video URL")
 
-def convert_video(video_file: dict, quality, aspect_ratio, video_url):
+def convert_video(video_file, quality, aspect_ratio, video_url):
     # If a file was uploaded
     if video_file is not None:
         # Get the file extension
-        file_extension = video_file['name'].split(".")[-1]
+        file_extension = video_file.name.split(".")[-1]
 
         # Define output file name
-        output_file_name = video_file['name'][:-len(file_extension)] + ".m3u8"
+        output_file_name = video_file.name[:-len(file_extension)] + ".m3u8"
         
-        # Write the uploaded file to the temporary directory
-        with open(os.path.join(temp_dir, video_file['name']), 'wb') as f:
-            f.write(video_file['content'])
-
-        input_path = os.path.join(temp_dir, video_file['name'])
+        input_path = video_file.name
     
     # If a URL was provided
     elif video_url is not None:
+        # Get the file name from URL
+        file_name = video_url.split("/")[-1]
+
         # Get the file extension
-        file_extension = video_url.split(".")[-1]
+        file_extension = file_name.split(".")[-1]
 
         # Define output file name
-        output_file_name = video_url.split("/")[-1][:-len(file_extension)] + ".m3u8"
+        output_file_name = file_name[:-len(file_extension)] + ".m3u8"
         
         # Download the file to the temporary directory
         response = requests.get(video_url)
         
         if response.status_code == 200:
-            with open(os.path.join(temp_dir, output_file_name), 'wb') as f:
-                f.write(response.content)
-
             input_path = os.path.join(temp_dir, output_file_name)
+            with open(input_path, 'wb') as f:
+                f.write(response.content)
         else:
             print("Failed to download file from URL.")
             return None
-
     else:
         print("No input was provided.")
         return None
@@ -82,7 +79,6 @@ def convert_video(video_file: dict, quality, aspect_ratio, video_url):
     if os.path.exists(output_path):
         # The file already exists, so we can just display it in the output viewer
         return gr.outputs.File(output_path, label="Download")
-
     else:
         # The file does not exist, so we need to convert it
         ffmpeg_command = f"ffmpeg -i {input_path} -c:v libx264 -crf {quality} -vf scale=-1:720,setsar=1:1 -hls_time 6 -hls_playlist_type vod -f hls {output_path}"