Update app.py

app.py

@@ -1,11 +1,11 @@
 import os
+import requests
 from datetime import datetime, timedelta
 from pytube import YouTube
 from moviepy.editor import VideoFileClip
 from tqdm import tqdm
 import gradio as gr
 
-# Function to download YouTube video
 def download_youtube(url, nama_channel, new_name, extension):
     response = requests.get(url, stream=True)
     file_name = new_name + "." + extension
@@ -27,34 +27,7 @@ def download_youtube(url, nama_channel, new_name, extension):
         progress_bar.close()
         print("")
 
-    return f'File berhasil diunduh dan disimpan di direktori:\n\n{filename}', filename
-
-# Function to cut and save the video
-def cut_video(link, start_time_str, end_time_str):
-    yt = YouTube(link)
-    stream = yt.streams.filter(progressive=True, resolution='720p').first()
-
-    if stream is None:
-        return "Maaf, resolusi yang Anda pilih tidak tersedia."
-
-    nama_channel = yt.author
-    judul_video = yt.title.replace('/', ' ')
-    filename = download_youtube(stream.url, nama_channel, judul_video, 'mp4')
-
-    start_time_parts = start_time_str.split(':')
-    end_time_parts = end_time_str.split(':')
-    start_time_seconds = int(start_time_parts[0]) * 3600 + int(start_time_parts[1]) * 60 + float(start_time_parts[2])
-    end_time_seconds = int(end_time_parts[0]) * 3600 + int(end_time_parts[1]) * 60 + float(end_time_parts[2])
-    potong = "/content/Hasil Potong"
-    if not os.path.exists(potong):
-        os.makedirs(potong)
-    output_file_path = f'{potong}/{judul_video}.mp4'
-
-    with VideoFileClip(filename) as video:
-        subclip = video.subclip(start_time_seconds, end_time_seconds)
-        subclip.write_videofile(output_file_path)
-
-    return output_file_path
+    return filename
 
 def format_number(number):
     if number < 1000:
@@ -65,40 +38,63 @@ def format_number(number):
         return f"{round(number / 1000000)} juta"
     else:
         return f"{round(number / 1000000000)} miliar"
+    
+def cut_video(link, resolusi_input, start_time_str, end_time_str):
+    yt = YouTube(link)
 
-link = gr.inputs.Textbox(lines=1, label="Input Link Youtube")
-yt = YouTube(link)
-
-nama_channel = yt.author
-judul_video = yt.title.replace('/', ' ')
-tanggal_upload = yt.publish_date.strftime("%-d %B %Y")
-jumlah_viewer = format_number(yt.views)
-selisih_hari = (datetime.now() - yt.publish_date).days
-rata2_viewer_per_hari = format_number(int(yt.views if selisih_hari < 1 else yt.views / selisih_hari))
-durasi_video = str(timedelta(seconds=yt.length))
+    nama_channel = yt.author
+    judul_video = yt.title.replace('/', ' ')
+    tanggal_upload = yt.publish_date.strftime("%-d %B %Y")
+    jumlah_viewer = format_number(yt.views)
+    selisih_hari = (datetime.now() - yt.publish_date).days
+    rata2_viewer_per_hari = format_number(int(yt.views if selisih_hari < 1 else yt.views / selisih_hari))
+    durasi_video = str(timedelta(seconds=yt.length))
 
-print("Nama Channel:", nama_channel)
-print("Judul Video:", judul_video)
-print("Tanggal Upload:", tanggal_upload)
-print("Jumlah Viewer:", jumlah_viewer)
-print("Rata-rata Viewer per Hari:", rata2_viewer_per_hari)
-print("Durasi Video:", durasi_video)
+    print("Nama Channel:", nama_channel)
+    print("Judul Video:", judul_video)
+    print("Tanggal Upload:", tanggal_upload)
+    print("Jumlah Viewer:", jumlah_viewer)
+    print("Rata-rata Viewer per Hari:", rata2_viewer_per_hari)
+    print("Durasi Video:", durasi_video)
 
-resolusi_tersedia = [stream.resolution for stream in yt.streams.filter(progressive=True)]
-print("Resolusi yang tersedia:", ", ".join(resolusi_tersedia))
+    resolusi_tersedia = [stream.resolution for stream in yt.streams.filter(progressive=True)]
+    print("Resolusi yang tersedia:", ", ".join(resolusi_tersedia))
 
-resolusi_input = gr.inputs.Dropdown(choices=resolusi_tersedia, label="Pilih Resolusi")
-start_time_input = gr.inputs.Textbox(lines=1, label="Start Time (HH:MM:SS.MS)")
-end_time_input = gr.inputs.Textbox(lines=1, label="End Time (HH:MM:SS.MS)")
+    resolusi = resolusi_input + "p"
+    stream = yt.streams.filter(progressive=True, resolution=resolusi).first()
 
-cut_button = gr.outputs.Button(label="Cut Video")
+    if stream is None:
+        print("Maaf resolusi yang anda masukkan tidak ada")
+        return None
+    else:
+        filename = download_youtube(stream.url, nama_channel, judul_video, 'mp4')
+    
+        start_time_parts = start_time_str.split(':')
+        end_time_parts = end_time_str.split(':')
+        start_time_seconds = int(start_time_parts[0]) * 3600 + int(start_time_parts[1]) * 60 + float(start_time_parts[2])
+        end_time_seconds = int(end_time_parts[0]) * 3600 + int(end_time_parts[1]) * 60 + float(end_time_parts[2])
+        potong = "Hasil Potong"  # Ganti dengan path yang sesuai di server Hugging Face
+        if not os.path.exists(potong):
+            os.makedirs(potong)
+        output_file_path = f'{potong}/{judul_video}.mp4'
+
+        with VideoFileClip(filename) as video:
+            subclip = video.subclip(start_time_seconds, end_time_seconds)
+            subclip.write_videofile(output_file_path)
+
+        return output_file_path
 
 iface = gr.Interface(
     fn=cut_video,
-    inputs=[link, start_time_input, end_time_input, cut_button],
+    inputs=[
+        gr.inputs.Textbox(lines=1, label="Link Video"),
+        gr.inputs.Dropdown(choices=resolusi_tersedia, label="Resolusi"),
+        gr.inputs.Textbox(lines=1, label="Start Time (HH:MM:SS.MS)"),
+        gr.inputs.Textbox(lines=1, label="End Time (HH:MM:SS.MS)"),
+    ],
     outputs=gr.outputs.File(label="Download Cut Video"),
     title="YouTube Video Cutter",
-    description="Cut and download portions of YouTube videos.",
+    description="Potong dan download sebagian video YouTube.",
 )
 
 iface.launch()