Update app.py

app.py

@@ -10,6 +10,7 @@ from urllib.parse import urlparse
 import http.server
 import socketserver
 import threading
+import atexit
 
 logging.basicConfig(level=logging.INFO)
 
@@ -28,6 +29,11 @@ def start_server():
 t = threading.Thread(target=start_server)
 t.start()
 
+# Cleanup function to remove the temporary directory when the script is exited
+@atexit.register
+def cleanup():
+    shutil.rmtree(temp_dir)
+
 logging.basicConfig(level=logging.INFO)
 
 def download_file(url, destination):
@@ -72,6 +78,46 @@ def create_master_playlist(output_paths, temp_dir):
             f.write(f"{path.name}\n")
     return master_playlist_path
 
+import logging
+import shutil
+import tempfile
+import subprocess
+from pathlib import Path
+from moviepy.editor import VideoFileClip
+import gradio as gr
+import requests
+from urllib.parse import urlparse
+import http.server
+import socketserver
+import threading
+import atexit
+
+logging.basicConfig(level=logging.INFO)
+
+PORT = 8000
+temp_dir = Path(tempfile.mkdtemp())  # Create a temporary directory using mkdtemp() instead of TemporaryDirectory()
+
+class Handler(http.server.SimpleHTTPRequestHandler):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, directory=str(temp_dir), **kwargs)
+
+def start_server():
+    with socketserver.TCPServer(("", PORT), Handler) as httpd:
+        print(f"Serving at port {PORT}")
+        httpd.serve_forever()
+
+t = threading.Thread(target=start_server)
+t.start()
+
+# Cleanup function to remove the temporary directory when the script is exited
+@atexit.register
+def cleanup():
+    shutil.rmtree(temp_dir)
+
+logging.basicConfig(level=logging.INFO)
+
+# ... rest of your code ...
+
 def convert_video(video_file, quality, aspect_ratio, video_url):
     # Use the already-created temp_dir instead of creating a new one
     # temp_dir = Path(tempfile.mkdtemp())
@@ -106,21 +152,24 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
     master_playlist_path = create_master_playlist(output_paths, temp_dir)
     output_paths.append(master_playlist_path)
 
-    return output_paths  # Return the list of completed output files
+    # Convert Path objects to URLs before returning
+    return [{"name": path.stem, "url": f"http://localhost:{PORT}/{path.name}"} for path in output_paths]
 
 video_file = gr.File(label="Video File")
 quality = gr.Dropdown(
     choices=["18", "23", "27", "28", "32"], label="Quality", default="27")
 aspect_ratio = gr.Textbox(default="16:9", lines=1, label="Aspect ratio (width:height)")
-standard_resolutions = [4320, 2160, 1440, 1080, 720, 480, 360, 240, 144]  # 8K, 4K, 2K, Full HD, HD, SD in pixels
+standard_resolutions = gr.Dropdown(
+    choices=[4320, 2160, 1440, 1080, 720, 480, 360, 240, 144]  # 8K, 4K, 2K, Full HD, HD, SD in pixels
 video_url = gr.Textbox(label="Or enter video URL")
 
 interface = gr.Interface(
     fn=convert_video, 
     inputs=[video_file, quality, aspect_ratio, video_url],
-    outputs=gr.Textbox(label="Output"),
+    outputs=gr.outputs.HTML(label="Download Links"),
     title="Video Converter",
     description="A simple video converter app",
     allow_flagging=False, 
 )
 interface.launch()
+