AshikAi96/Youtube_video_scripting_tool

1import streamlit as st 
2from utils import generate_script
3
4# Applying Styling
5st.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 Variable
19if '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 key
26st.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 Inputs
32prompt = st.text_input('Please provide the topic of the video',key="prompt")  # The box for the text prompt
33video_length = st.text_input('Expected Video Length 🕒 (in minutes)',key="video_length")  # The box for the text prompt
34creativity = st.slider('Creativity 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 script
44        st.success('Hope you like this script 😊')
45
46        #Display Title
47        st.subheader("Title: ")
48        st.write(title)
49
50        #Display Video Script
51        st.subheader("Your Video Script:📝")
52        st.write(script)
53
54        #Display Search Engine Result
55        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.....")
60