Update app.py

app.py

@@ -5,27 +5,31 @@ from pornhub import ph_down
 from other import cut_video, session
 
 # Navigasi Sidebar
+options = ['Youtube', 'Pornhub', 'Iwara', 'Mega', 'Rule34', 'Trailer']
 with st.sidebar:
-    selected = option_menu("Video Downloader", ['Youtube', 'Pornhub', 'Iwara', 'Mega', 'Rule34', 'Trailer'], 
+    selected = option_menu("Video Downloader", options, 
         icons=['play', 'fire', 'star', 'star', 'moon','gear', 'lightning'], menu_icon="cast", default_index=0)
 
-if selected == 'Youtube':
-    st.title("YouTube Video Cutter")
-    st.write("Potong dan download sebagian video YouTube.")    
+functions = [yt_down, ph_down]
+
+if selected:
+    index = options.index(selected)
+    fungsi = functions[index]
+    st.title(f"{selected} Video Downloader and Cutter")
+    st.write(f"Potong dan download sebagian video {selected}.")    
     video_link = st.text_input("Link Video", value='https://www.youtube.com/watch?v=ZGltvcmVSAk')
     resolution = st.text_input("Resolusi", value='720')
+    
     cut = st.checkbox("Potong Video")
     if cut:
         start_time = st.text_input("Start Time", value='00:07:12.750')
         end_time = st.text_input("End Time", value='00:09:31.000')
     
-        if st.button("Download and Cut Youtube"):
-            filename, judul_video, video_info = yt_down(video_link, resolution)
-            output_file_path = cut_video(filename, start_time, end_time)
+        if st.button(f"Download and Cut {selected}"):
+            filename, judul_video, video_info = fungsi(video_link, resolution)
+            output_file_path = cut_video(filename, judul_video, start_time, end_time)
             session(video_info, video_file, cut)
     else:
-        if st.button("Download Youtube"):
+        if st.button(f"Download {selected}"):
             video_info, video_file = yt_down(video_link, resolution, start_time, end_time)
-            session(video_info, video_file, cut)
-    
-
+            session(video_info, video_file, cut)