import os
import streamlit as st
from streamlit_option_menu import option_menu
from youtube import youtube
from pornhub import pornhub
from other import cut_video, session, convert_size

# Navigasi Sidebar
options = ['Youtube', 'Pornhub', 'Iwara', 'Mega', 'Rule34', 'Paipancon', 'Trailer']
with st.sidebar:
    selected = option_menu("Video Downloader", options, 
        icons=['play', 'fire', 'star', 'moon','gear', 'house', 'lightning'], menu_icon="cast", default_index=0)

functions = [youtube, pornhub]

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.selectbox("Pilih Resolusi", (360, 480, 720), 2)
    
    cut = st.checkbox("Potong Video")
    if cut:
        start_time = st.text_input("Start Time", value='00:07:12.000')
        end_time = st.text_input("End Time", value='00:07:31.000')
    
        if st.button(f"Download and Cut {selected}"):
            video_file, judul_video, video_info = fungsi(video_link, resolution)
            video_file = cut_video(video_file, judul_video, start_time, end_time)
            file_size = os.path.getsize(video_file)
            session(video_info, video_file, cut)
            st.text_area(f"Video '{judul_video}' setelah diproses:", convert_size(file_size))
    else:
        if st.button(f"Download {selected}"):
            video_file, judul_video, video_info = fungsi(video_link, resolution)
            file_size = os.path.getsize(video_file)
            session(video_info, video_file, cut)
            st.text_area(f"Video '{judul_video}' setelah diproses:", convert_size(file_size))