Update app.py

app.py

@@ -1,12 +1,10 @@
 import logging
 import requests
-import gradio
 from urllib.parse import urlparse
 import subprocess
 from pathlib import Path
 from moviepy.editor import VideoFileClip
-import gradio as gr
-from gradio import components
+from gradio import components as gr
 import http.server
 import socketserver
 import threading
@@ -78,11 +76,11 @@ def get_aspect_ratio(input_path, aspect_ratio):
 
 def create_master_playlist(output_paths):
     master_playlist_path = output_dir / "master_playlist.m3u8"
-    with open(master_playlist_path, 'w') as f:
+    with open(master_playlist_path, "w") as f:
         f.write("#EXTM3U\n")
         for path in output_paths:
-            f.write(f"#EXT-X-STREAM-INF:BANDWIDTH={1000*1000},RESOLUTION={path.stem.split('_')[-1]}\n")
-            f.write(f"{path.name}\n")
+            f.write(f'#EXT-X-STREAM-INF:BANDWIDTH=8000000,RESOLUTION={Path(path).stem.split("_")[1]}\n')
+            f.write(f'{Path(path).name}\n')
     return master_playlist_path
 
 def convert_video(video_file, quality, aspect_ratio, video_url):
@@ -90,6 +88,8 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
     aspect_ratio = get_aspect_ratio(input_path, aspect_ratio)
 
     video = VideoFileClip(str(input_path))
+
+    # Get the original height to avoid upscaling
     original_height = video.size[1]
 
     output_paths = []
@@ -127,22 +127,19 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
     return output_html
 
 # Change "video" to "file"
-video_file = gradio.inputs.File(label="Your video file")
-quality = gradio.inputs.Slider(minimum=1, maximum=50, default=25, label="Quality (1:Speed - 50:Quality)")
-aspect_ratio = gradio.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 = gradio.inputs.Textbox(lines=1, placeholder="or paste video url here", label="Video URL")
-
-# Update the aspect ratio selection to include the comprehensive list
-# aspect_ratio = gr.inputs.Dropdown(choices=aspect_ratios, label="Aspect Ratio", default="16:9")
+video_file = gr.File(label="Your video file")
+quality = gr.Slider(minimum=1, maximum=50, default=25, label="Quality (1:Speed - 50:Quality)")
+aspect_ratio = gr.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.Textbox(lines=1, placeholder="or paste video url here", label="Video URL")
 
-interface = gradio.Interface(
+interface = gr.Interface(
     fn=convert_video, 
     inputs=[video_file, quality, aspect_ratio, video_url],
-    outputs=gradio.outputs.HTML(label="Download Links"),
+    outputs=gr.HTML(label="Download Links"),
     title="NEAR Hub Video Transcoder to m3u8",
     description="convert video files to m3u8 for VOD streaming",
-    allow_flagging=False
+    allow_flagging='never'
 )
 
 start_http_server()
-interface.launch(server_name="0.0.0.0", server_port=7860)
+interface.launch(server_name=get_ip_address(), server_port=7860)