Update app.py

app.py

@@ -7,7 +7,7 @@ from moviepy.editor import VideoFileClip
 import gradio as gr
 import requests
 from urllib.parse import urlparse
-
+from ftplib import FTP
 
 logging.basicConfig(level=logging.INFO)
 
@@ -31,6 +31,13 @@ def get_input_path(video_file, video_url, temp_dir):
     else:
         raise ValueError("No input was provided.")
 
+def upload_to_ftp(server, username, password, source_file, destination_file):
+    """Uploads a file to an FTP server."""
+    with FTP(server, username, password) as ftp:
+        with open(source_file, 'rb') as f:
+            ftp.storbinary(f'STOR {destination_file}', f)
+
+
 def get_output_path(input_path, temp_dir, res):
     """Returns the path to the output file, creating it if necessary."""
     output_path = temp_dir / (Path(input_path).stem + f"_{res}.m3u8")
@@ -103,6 +110,16 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
 
             output_paths.append(output_path)  # Append the output_path to output_paths
 
+        if upload:
+            ftp_files = []
+            for path in [master_playlist_copy_path] + output_copy_paths:
+                destination = ftp_path + "/" + path.name
+                upload_to_ftp(ftp_server, ftp_username, ftp_password, path, destination)
+                ftp_files.append(destination)
+            return ftp_files
+        else:
+            return [master_playlist_copy_path] + output_copy_paths
+
         if not output_paths:
             return "The video is smaller than the smallest standard resolution."
 
@@ -140,9 +157,15 @@ def main():
         label="Aspect Ratio", default="16:9") 
     video_url = gr.inputs.Textbox(label="Video URL")
 
+    ftp_server = gr.inputs.Textbox(label="FTP Server")
+    ftp_username = gr.inputs.Textbox(label="FTP Username")
+    ftp_password = gr.inputs.Textbox(label="FTP Password", type="password")
+    ftp_path = gr.inputs.Textbox(label="FTP Path")
+    upload = gr.inputs.Checkbox(label="Upload to FTP server", default=False)
+
     gr.Interface(
         convert_video, 
-        inputs=[video_file, quality, aspect_ratio, video_url],
+        inputs=[video_file, quality, aspect_ratio, video_url, ftp_server, ftp_username, ftp_password, ftp_path, upload],
         outputs=gr.outputs.Textbox(),
         output_processor=process_output,
         allow_flagging=False,