Update app.py

app.py

@@ -11,10 +11,25 @@ import http.server
 import socketserver
 import threading
 import atexit
+import socket
 
 logging.basicConfig(level=logging.INFO)
 
-PORT = 8000
+PORT = 8000  # Define PORT here
+
+def get_available_port(start_port):
+    while True:
+        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+            try:
+                s.bind(("", start_port))
+                return start_port
+            except OSError as e:
+                if e.errno == 98:  # errno 98 means address already in use
+                    start_port += 1  # If address in use, increment port and try again
+                    continue
+                else:
+                    raise
+
 temp_dir = Path(tempfile.mkdtemp())  # Create a temporary directory using mkdtemp() instead of TemporaryDirectory()
 
 class Handler(http.server.SimpleHTTPRequestHandler):
@@ -23,19 +38,11 @@ class Handler(http.server.SimpleHTTPRequestHandler):
 
 def start_server():
     global PORT  # Make sure to define PORT as global
-    while True:
-        try:
-            with socketserver.TCPServer(("", PORT), Handler) as httpd:
-                print(f"Serving at port {PORT}")
-                httpd.serve_forever()
-        except OSError as e:
-            if e.errno == 98:  # errno 98 means address already in use
-                PORT += 1  # If address in use, increment port and try again
-                continue
-            else:
-                raise
-        break  # If server started successfully, break the loop
-
+    PORT = get_available_port(PORT)
+    with socketserver.TCPServer(("", PORT), Handler) as httpd:
+        print(f"Serving at port {PORT}")
+        httpd.serve_forever()
+       
 t = threading.Thread(target=start_server)
 t.start()
 
@@ -88,46 +95,6 @@ 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())
@@ -169,8 +136,7 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
 video_file = gr.File(label="Video File")
 quality = gr.Dropdown(
     "27", ["18", "23", "27", "28", "32"], label="Quality")
-aspect_ratio = gr.Dropdown("16:9", ["16:9",  "1:1", "4:3", "3:2", "5:4", "21:9", "1.85:1", "2.35:1", "3:1", "360", "9:16", "2:1", "1:2", "9:1"],
-       label="Aspect ratio (width:height)")
+aspect_ratio = gr.Dropdown(["16:9",  "1:1", "4:3", "3:2", "5:4", "21:9", "1.85:1", "2.35:1", "3:1", "360", "9:16", "2:1", "1:2", "9:1"], label="Aspect ratio (width:height)", default="16:9")
 standard_resolutions = [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")