Update app.py

app.py

@@ -28,7 +28,6 @@ def start_http_server(port=8000):
 def get_ip_address():
     s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
     try:
-        # doesn't have to be reachable
         s.connect(('10.255.255.255', 1))
         IP = s.getsockname()[0]
     except Exception:
@@ -59,10 +58,6 @@ def get_output_path(input_path, res):
     output_path = output_dir / (Path(input_path).stem + f"_{res}p.m3u8")
     return output_path
 
-def get_aspect_ratio(input_path):
-    video = VideoFileClip(str(input_path))
-    return f"{video.size[0]}:{video.size[1]}"
-
 def create_master_playlist(output_paths):
     master_playlist_path = output_dir / "master_playlist.m3u8"
     with open(master_playlist_path, "w") as f:
@@ -72,7 +67,7 @@ def create_master_playlist(output_paths):
             f.write(f'{Path(path).name}\n')
     return master_playlist_path
 
-def convert_video(video_file, quality, video_url):
+def convert_video(video_file, quality, aspect_ratio, video_url):
     input_path = get_input_path(video_file, video_url)
 
     video = VideoFileClip(str(input_path))
@@ -104,24 +99,22 @@ def convert_video(video_file, quality, video_url):
     master_playlist_path = create_master_playlist(output_paths)
     output_paths.append(master_playlist_path)
 
-    # Get server IP and generate output links
-    server_ip = get_ip_address()  # or get_public_ip_address() if you need the public IP
+    server_ip = get_ip_address()
     output_links = []
     for path in output_paths:
-        # Replace 'your_server_ip' with server_ip
         output_links.append(f'<a href="http://{server_ip}:8000/{path}" target="_blank">Download {Path(path).stem}</a>')
 
     output_html = "<br>".join(output_links)
     return output_html
 
-# Change "video" to "file"
 video_file = gr.inputs.File(label="Your video file")
 quality = gr.inputs.Slider(minimum=1, maximum=50, label="Quality (1:Speed - 50:Quality)")
+aspect_ratio = gr.inputs.Dropdown(["16:9", "4:3", "1:1", "3:4", "9:16", "1:1", "2:1", "1:2"], label="Aspect Ratio", default="16:9")
 video_url = gr.inputs.Textbox(lines=1, placeholder="or paste video url here", label="Video URL")
 
 interface = gr.Interface(
     fn=convert_video, 
-    inputs=[video_file, quality, video_url],
+    inputs=[video_file, quality, aspect_ratio, video_url],
     outputs=gr.outputs.HTML(label="Download Links"),
     title="NEAR Hub Video Transcoder to m3u8",
     description="convert video files to m3u8 for VOD streaming",