Update app.py

app.py

@@ -1,7 +1,9 @@
 import streamlit as st
 from streamlit_option_menu import option_menu
 import streamlit as st
-from youtube import cut_video
+from youtube import yt_down
+from pornhub import ph_down
+from other import cut_video
 
 # Navigasi Sidebar
 with st.sidebar:
@@ -13,25 +15,18 @@ if selected == 'Youtube':
     st.write("Potong dan download sebagian video YouTube.")    
     video_link = st.text_input("Link Video", value='https://www.youtube.com/watch?v=ZGltvcmVSAk')
     resolution = st.text_input("Resolusi", value='720')
-    start_time = st.text_input("Start Time", value='00:07:12.750')
-    end_time = st.text_input("End Time", value='00:09:31.000')
+    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 Video"):
-        video_info, video_result, video_file = cut_video(video_link, resolution, start_time, end_time)
-        st.session_state.video_info = video_info
-        st.session_state.video_result = video_result
-        st.session_state.video_file = video_file
+        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)
+            session(video_info, video_file, cut)
+    else:
+        if st.button("Download Youtube"):
+            video_info, video_file = yt_down(video_link, resolution, start_time, end_time)
+            session(video_info, video_file, cut)
     
-    if 'video_info' in st.session_state:
-        num_lines = st.session_state.video_info.count('\n') + 1
-        text_area_height = 25 * num_lines
-        st.text_area("Informasi Video", st.session_state.video_info, height=text_area_height)
-        st.video(st.session_state.video_result)    
-        with open(st.session_state.video_file, 'rb') as f:
-            file_contents = f.read()
-        st.download_button(
-            label="Download Hasil Potongan Video",
-            data=file_contents,
-            file_name=st.session_state.video_file.replace('/home/user/app/Hasil Potong', '').title().replace('Mp4', 'mp4'),
-            mime='application/octet-stream'
-        )
+