Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,6 +9,7 @@ import http.server
|
|
| 9 |
import socketserver
|
| 10 |
import threading
|
| 11 |
import socket
|
|
|
|
| 12 |
import s3fs
|
| 13 |
import os
|
| 14 |
|
|
@@ -20,12 +21,14 @@ output_dir.mkdir(exist_ok=True)
|
|
| 20 |
AWS_S3_BUCKET = os.getenv("AWS_S3_BUCKET")
|
| 21 |
AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
|
| 22 |
AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
|
|
|
|
|
|
| 26 |
|
| 27 |
-
|
| 28 |
-
fs = s3fs.S3FileSystem(
|
| 29 |
|
| 30 |
logging.basicConfig(level=logging.INFO)
|
| 31 |
|
|
@@ -116,8 +119,13 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
|
|
| 116 |
output_links = []
|
| 117 |
for path in output_paths:
|
| 118 |
output_links.append(f'<a href="http://{server_ip}:8000{path}" target="_blank">Download {Path(path).stem}</a>')
|
|
|
|
|
|
|
|
|
|
| 119 |
print("Upload " + Path(path).stem + " to AWS")
|
| 120 |
-
fs.mv(str(output_dir / f"{output_path.stem}_%03d.ts"), f"s3://{AWS_S3_BUCKET}/{Path(path).stem}",)
|
|
|
|
|
|
|
| 121 |
|
| 122 |
output_html = "<br>".join(output_links)
|
| 123 |
return output_html
|
|
|
|
| 9 |
import socketserver
|
| 10 |
import threading
|
| 11 |
import socket
|
| 12 |
+
import boto3
|
| 13 |
import s3fs
|
| 14 |
import os
|
| 15 |
|
|
|
|
| 21 |
AWS_S3_BUCKET = os.getenv("AWS_S3_BUCKET")
|
| 22 |
AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
|
| 23 |
AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
|
| 24 |
+
session = boto3.Session(
|
| 25 |
+
aws_access_key_id=AWS_ACCESS_KEY_ID,
|
| 26 |
+
aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
|
| 27 |
+
region_name='us-east-1'
|
| 28 |
+
)
|
| 29 |
|
| 30 |
+
s3 = session.client('s3')
|
| 31 |
+
fs = s3fs.S3FileSystem(anon=False, client_kwargs={'region_name': 'us-east-1'})
|
| 32 |
|
| 33 |
logging.basicConfig(level=logging.INFO)
|
| 34 |
|
|
|
|
| 119 |
output_links = []
|
| 120 |
for path in output_paths:
|
| 121 |
output_links.append(f'<a href="http://{server_ip}:8000{path}" target="_blank">Download {Path(path).stem}</a>')
|
| 122 |
+
print("Read local file")
|
| 123 |
+
with open(path, 'rb') as file:
|
| 124 |
+
file_content = file.read()
|
| 125 |
print("Upload " + Path(path).stem + " to AWS")
|
| 126 |
+
# fs.mv(str(output_dir / f"{output_path.stem}_%03d.ts"), f"s3://{AWS_S3_BUCKET}/{Path(path).stem}",)
|
| 127 |
+
with fs.open(f"s3://{AWS_S3_BUCKET}/{Path(path).stem}", 'wb') as file:
|
| 128 |
+
file.write(your_file_content)
|
| 129 |
|
| 130 |
output_html = "<br>".join(output_links)
|
| 131 |
return output_html
|