Sakit

app.py

@@ -32,8 +32,9 @@ if selected:
     else:
         video_link = st.text_input("Link Video")
     
-    cut = st.checkbox("Potong Video")
-    if cut:
+    choice = st.radio('Pilih salah satu:', ['Potong Video', 'Compress Video', 'Cuma Download'], 2)
+
+    if choice == 'Potong Video':
         start_time = st.text_input("Start Time", value='00:07:12.000')
         end_time = st.text_input("End Time", value='00:07:31.000')
     
@@ -46,7 +47,22 @@ if selected:
                 video_file, judul_video, video_info, thumbnail_file = fungsi(video_link)
             video_file = cut_video(video_file, judul_video, start_time, end_time)
             file_size = os.path.getsize(video_file)
-            session(video_info, video_file, thumbnail_file, cut)
+            session(video_info, video_file, thumbnail_file, choice)
+            st.text_input(f"Video '{judul_video}' setelah diproses:", convert_size(file_size))
+
+    elif choice == 'Compress Video':
+        compress = st.selectbox("Pilih Resolusi Compress", (360, 480, 720), 2)
+    
+        if st.button(f"Download and Cut {selected}"):
+            if selected == 'Youtube' or selected == 'Pornhub':
+                video_file, judul_video, video_info, thumbnail_file = fungsi(video_link, resolution)
+            elif selected == 'Iwara' or selected == 'Mega':
+                video_file, judul_video, video_info, thumbnail_file = fungsi(video_link, name)
+            else:
+                video_file, judul_video, video_info, thumbnail_file = fungsi(video_link)
+            video_file = convert_videos(compress, video_file)
+            file_size = os.path.getsize(video_file)
+            session(video_info, video_file, thumbnail_file, choice)
             st.text_input(f"Video '{judul_video}' setelah diproses:", convert_size(file_size))
 
     else:
@@ -58,5 +74,5 @@ if selected:
             else:
                 video_file, judul_video, video_info, thumbnail_file = fungsi(video_link)
             file_size = os.path.getsize(video_file)
-            session(video_info, video_file, thumbnail_file, cut)
+            session(video_info, video_file, thumbnail_file, choice)
             st.text_input(f"Video '{judul_video}' setelah diproses:", convert_size(file_size))

others.py

@@ -51,12 +51,14 @@ def cut_video(filename, judul_video, start_time, end_time):
 
     return output_file_path
     
-def session(video_info, video_file, thumbnail_file, cut):
+def session(video_info, video_file, thumbnail_file, choice):
     st.session_state.video_info = video_info
     st.session_state.video_file = video_file
     st.session_state.thumbnail_file = thumbnail_file
-    if cut:
-        hasil = "Hasil Potongan "
+    if choice:
+        hasil = "Hasil Potongan"
+    elif choice:
+        hasil = "Hasil Compress"
     else:
         hasil = ""
     if 'video_info' in st.session_state:
@@ -215,8 +217,9 @@ def get_video_resolution(input_file):
         print(f'Error getting video resolution: {result.stderr}')
         return None
     
-def convert_videos(height, download_folder):
-    konversi_folder  = '/home/user/app/Hasil Konversi'
+def convert_videos(height, filename):
+    download_folder = os.path.dirname(filename)
+    konversi_folder  = f'{download_folder}/Hasil Konversi'
     if not os.path.exists(konversi_folder):
         os.makedirs(konversi_folder)
 
@@ -240,8 +243,8 @@ def convert_videos(height, download_folder):
 
                     # Convert video using ffmpeg
                     if height == 720:
-                        command = f'ffmpeg -i "{input_file}" -s {width}x{height} -b:v 850k -bufsize 1000k -r 30 -b:a 160k -ar 44100 -ac 2 "{output_file}"'
-                        # command = f'ffmpeg -i "{input_file}" -s {width}x{height} -b:v 1350k -bufsize 1500k -r 30 -b:a 160k -ar 48000 -ac 2 "{output_file}"'
+                        # command = f'ffmpeg -i "{input_file}" -s {width}x{height} -b:v 850k -bufsize 1000k -r 30 -b:a 160k -ar 44100 -ac 2 "{output_file}"'
+                        command = f'ffmpeg -i "{input_file}" -s {width}x{height} -b:v 1350k -bufsize 1500k -r 30 -b:a 160k -ar 48000 -ac 2 "{output_file}"'
                     elif height == 480:
                         command = f'ffmpeg -i "{input_file}" -s {width}x{height} -b:v 600k -bufsize 700k -r 24 -b:a 95k -ar 44100 -ac 2 "{output_file}"'
                     elif height == 360:
@@ -255,8 +258,8 @@ def convert_videos(height, download_folder):
 
                     # Convert video using ffmpeg
                     if width == 720:
-                        command = f'ffmpeg -i "{input_file}" -s {width}x{height} -b:v 850k -bufsize 1000k -r 30 -b:a 160k -ar 44100 -ac 2 "{output_file}"'
-                        # command = f'ffmpeg -i "{input_file}" -s {width}x{height} -b:v 1350k -bufsize 1500k -r 30 -b:a 160k -ar 48000 -ac 2 "{output_file}"'
+                        # command = f'ffmpeg -i "{input_file}" -s {width}x{height} -b:v 850k -bufsize 1000k -r 30 -b:a 160k -ar 44100 -ac 2 "{output_file}"'
+                        command = f'ffmpeg -i "{input_file}" -s {width}x{height} -b:v 1350k -bufsize 1500k -r 30 -b:a 160k -ar 48000 -ac 2 "{output_file}"'
                     elif width == 480:
                         command = f'ffmpeg -i "{input_file}" -s {width}x{height} -b:v 600k -bufsize 700k -r 24 -b:a 95k -ar 44100 -ac 2 "{output_file}"'