thanhtd/YoutubeScriptWritingTool

1import streamlit as st 2from utils import generate_script3 4# Applying Styling5st.markdown("""6<style>7div.stButton > button:first-child {8    background-color: #0099ff;9    color:#ffffff;10}11div.stButton > button:hover {12    background-color: #00ff00;13    color:#FFFFFF;14    }15</style>""", unsafe_allow_html=True)16 17 18# Creating Session State Variable19if 'API_Key' not in st.session_state:20    st.session_state['API_Key'] =''21 22 23st.title('❤️ YouTube Script Writing Tool') 24 25# Sidebar to capture the OpenAi API key26st.sidebar.title("😎🗝️")27st.session_state['API_Key']= st.sidebar.text_input("What's your API key?",type="password")28st.sidebar.image('./Youtube.jpg',width=300, use_column_width=True)29 30 31# Captures User Inputs32prompt = st.text_input('Please provide the topic of the video',key="prompt")  # The box for the text prompt33video_length = st.text_input('Expected Video Length 🕒 (in minutes)',key="video_length")  # The box for the text prompt34creativity = st.slider('Words limit ✨ - (0 LOW || 1 HIGH)', 0.0, 1.0, 0.2,step=0.1)35 36submit = st.button("Generate Script for me")37 38 39if submit:40    41    if st.session_state['API_Key']:42        search_result,title,script = generate_script(prompt,video_length,creativity,st.session_state['API_Key'])43        #Let's generate the script44        st.success('Hope you like this script ❤️')45 46        #Display Title47        st.subheader("Title:🔥")48        st.write(title)49 50        #Display Video Script51        st.subheader("Your Video Script:📝")52        st.write(script)53 54        #Display Search Engine Result55        st.subheader("Check Out - DuckDuckGo Search:🔍")56        with st.expander('Show me 👀'): 57            st.info(search_result)58    else:59        st.error("Ooopssss!!! Please provide API key.....")