Update app.py

app.py

@@ -7,6 +7,7 @@ from moviepy.editor import VideoFileClip
 import gradio as gr
 import requests
 from urllib.parse import urlparse
+from web3.storage import Web3Storage
 
 logging.basicConfig(level=logging.INFO)
 
@@ -42,7 +43,14 @@ def get_aspect_ratio(input_path, aspect_ratio):
     video = VideoFileClip(str(input_path))
     return f"{video.size[0]}:{video.size[1]}"
 
-def convert_video(video_file, quality, aspect_ratio, video_url):
+def upload_to_web3_storage(api_key, path):
+    """Uploads a file to web3.storage."""
+    with open(path, 'rb') as f:
+        storage = Web3Storage.new(token=api_key)
+        cid = storage.put_file(f)
+        return f"https://dweb.link/ipfs/{cid}"
+
+def convert_video(video_file, quality, aspect_ratio, video_url, api_key, web3_upload):
     """Converts a video to HLS format, adjusting the quality and aspect ratio as necessary."""
     with tempfile.TemporaryDirectory() as temp_dir:
         temp_dir = Path(temp_dir)
@@ -74,8 +82,15 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
 
         # Copy the output to a new temporary file that won't be deleted when the
         # TemporaryDirectory context manager exits.
-        output_copy_path = shutil.copy2(output_path, tempfile.gettempdir())
+        output_copy_path = shutil.copy2(output_path, tempfile.gettemp# The previous message was cut off, let's complete it here.
+
+output_copy_path = shutil.copy2(output_path, tempfile.gettempdir())
 
+# If the user wants to upload to web3, do that and return the URL.
+# Otherwise, just return the local path.
+if web3_upload:
+    return upload_to_web3_storage(api_key, output_copy_path)
+else:
     return output_copy_path
 
 def main():
@@ -88,8 +103,10 @@ def main():
                  "2:1", "1:2", "9:1"], 
         label="Aspect Ratio", default=None) 
     video_url = gr.inputs.Textbox(label="Video URL")
+    api_key = gr.inputs.Textbox(label="web3.storage API Key")
+    web3_upload = gr.inputs.Checkbox(label="Upload to web3.storage")
 
-    gr.Interface(convert_video, inputs=[video_file, quality, aspect_ratio, video_url], outputs='file').launch()
+    gr.Interface(convert_video, inputs=[video_file, quality, aspect_ratio, video_url, api_key, web3_upload], outputs='text').launch()
 
 if __name__ == "__main__":
-    main()
+    main()