Update app.py

app.py

@@ -9,11 +9,20 @@ import http.server
 import socketserver
 import threading
 import socket
+import s3fs
 
 # Define output directory
 output_dir = Path.cwd() / "output"
 output_dir.mkdir(exist_ok=True)
 
+# Set S3 credentials
+AWS_S3_BUCKET = os.getenv("AWS_S3_BUCKET")
+AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
+AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
+
+storage_options = {"key": AWS_ACCESS_KEY_ID, "secret": AWS_SECRET_ACCESS_KEY}
+fs = s3fs.S3FileSystem(**storage_options)
+
 logging.basicConfig(level=logging.INFO)
 
 standard_resolutions = [240, 360, 480, 720, 1080, 1440, 2160, 4320]
@@ -103,6 +112,8 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
     output_links = []
     for path in output_paths:
         output_links.append(f'<a href="http://{server_ip}:8000{path}" target="_blank">Download {Path(path).stem}</a>')
+        print("Upload " + Path(path).stem + " to AWS")
+        fs.mv(str(output_dir / f"{output_path.stem}_%03d.ts"), f"s3://{AWS_S3_BUCKET}/{Path(path).stem}",)
 
     output_html = "<br>".join(output_links)
     return output_html