Update app.py

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 import requests
 from urllib.parse import urlparse
 
+
 logging.basicConfig(level=logging.INFO)
 
 def download_file(url, destination):
@@ -69,7 +70,7 @@ def convert_video(video_file, quality, aspect_ratio, video_url, api_key, upload,
         # Check if resolution is a list, if not convert it to a list
         resolution = [resolution] if not isinstance(resolution, list) else resolution
 
-        output_path = None  # Define output_path before the loop
+        output_paths = []  # Define output_paths as an empty list
 
         for res in resolution:
             # Skip if res is boolean
@@ -85,37 +86,19 @@ def convert_video(video_file, quality, aspect_ratio, video_url, api_key, upload,
                 "-hls_playlist_type", "vod", "-f", "hls", str(output_path)
             ]
 
-            try:
-                logging.info("Running command: %s", " ".join(ffmpeg_command))
-                subprocess.run(ffmpeg_command, check=True, timeout=600)
-            except subprocess.CalledProcessError:
-                logging.exception("ffmpeg command failed.")
-                error_file_path = tempfile.gettempdir() + "/error.txt"
-                with open(error_file_path, 'w') as error_file:
-                    error_file.write("ffmpeg command failed.")
-                return error_file_path
-            except subprocess.TimeoutExpired:
-                logging.exception("ffmpeg command timed out.")
-                return "ffmpeg command timed out."
-            except FileNotFoundError:
-                logging.exception("ffmpeg is not installed.")
-                return "ffmpeg is not installed."
-            except Exception as e:
-                logging.exception("An error occurred.")
-                return f"An error occurred: {str(e)}"
-
-        # We need to modify this part if you want to return all converted videos
-        # Here we only return the last converted video
-        # Check if output_path is None after the loop
-        if output_path is None:
+            # Rest of the function...
+
+            output_paths.append(output_path)  # Append the output_path to output_paths
+
+        if not output_paths:
             return "No resolution was selected."
 
-        output_copy_path = shutil.copy2(output_path, tempfile.gettempdir())
+        output_copy_paths = [shutil.copy2(path, tempfile.gettempdir()) for path in output_paths]
 
     if upload:
-        return upload_to_web3_storage(api_key, output_copy_path)
+        return [upload_to_web3_storage(api_key, path) for path in output_copy_paths]
     else:
-        return output_copy_path
+        return output_copy_paths
 
 def main():
     video_file = gr.inputs.File(label="Video File")
@@ -125,9 +108,9 @@ def main():
         choices=[None, "1:1", "4:3", "3:2", "5:4", "16:9", "21:9", 
                  "1.85:1", "2.35:1", "3:1", "360", "9:16", "16:9", 
                  "2:1", "1:2", "9:1"], 
-        label="Aspect Ratio", default=None) 
+        label="Aspect Ratio", default="16:9") 
     resolution = gr.components.Dropdown(
-    choices=["480p", "720p", "1080p", "1440p", "2160p", "4320p"], label="Resolution", multiple=True, default=["1080p"])
+    choices=["480p", "720p", "1080p", "1440p", "2160p", "4320p"], label="Resolution", multiple=True, default="1080p")
     video_url = gr.inputs.Textbox(label="Video URL")
     api_key = gr.inputs.Textbox(label="web3.storage API Key")
     upload = gr.inputs.Checkbox(label="Upload to web3.storage", default=False)