Update app.py

app.py

@@ -56,28 +56,33 @@ def convert_video(video_file: File, quality, aspect_ratio, video_url):
   # Set the output path to the temp directory
   output_path = os.path.join(temp_dir, output_path)
 
-  # Create the temporary file
-  os.mknod(output_path)
-
-  ffmpeg_command = f"ffmpeg -i {video_file} -c:v libx264 -crf {quality} -f mp4 -aspect {aspect_ratio} {output_path}"
-
-  try:
-    subprocess.run(ffmpeg_command, shell=True, timeout=10)
-  except subprocess.TimeoutExpired:
-    print("ffmpeg timed out")
-    return None
-  except FileNotFoundError:
-    print("ffmpeg is not installed.")
-    return None
-
-  for i in range(10):
-    if os.path.exists(output_path):
-      break
-    else:
-      time.sleep(1)
-
-  # Return the output file directly
-  return output_path
+  if os.path.exists(output_path):
+    # The file already exists, so we can just display it in the output viewer
+    gr.outputs.Video(output_path)
+  else:
+    # The file does not exist, so we need to convert it
+    ffmpeg_command = f"ffmpeg -i {video_file} -c:v libx264 -crf {quality} -f mp4 -aspect {aspect_ratio} {output_path}"
+
+    try:
+      subprocess.run(ffmpeg_command, shell=True, timeout=10)
+    except subprocess.TimeoutExpired:
+      print("ffmpeg timed out")
+      return None
+    except FileNotFoundError:
+      print("ffmpeg is not installed.")
+      return None
+
+    for i in range(10):
+      if os.path.exists(output_path):
+        break
+      else:
+        time.sleep(1)
+
+    # The file has now been converted, so we can display it in the output viewer
+    gr.outputs.Video(output_path)
+
+    # Add a button to download the file
+    gr.outputs.Button(label="Download", click=lambda: gr.download(output_path))
 
 from gradio import outputs