File size: 3,230 Bytes
508ee70 12671eb 508ee70 52cdd17 12671eb 508ee70 12671eb 52cdd17 12671eb 52cdd17 12671eb 52cdd17 12671eb 52cdd17 7db74a9 52cdd17 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
import streamlit as st
from dotenv import load_dotenv, set_key
from generate_script import generate_script
import os
import json
# Load environment variables
load_dotenv()
# Initialize the API key
api_key = st.session_state.get("OPENAI_API_KEY", None)
serper_api_key = st.secrets.get("SERPER_API_KEY")
# Check if API keys are available
if not api_key:
st.error("API Key not provided! Please enter it in the main app.")
else:
os.environ["OPENAI_API_KEY"] = api_key
# File to store script history
history_file = "script_history.json"
# Function to load history from file
def load_history():
if os.path.exists(history_file):
with open(history_file, "r") as file:
return json.load(file)
return []
# Function to save history to file
def save_history(history):
with open(history_file, "w") as file:
json.dump(history, file, indent=4)
# Load history from JSON file
script_history = load_history()
# Layout with two columns
col1, col2 = st.columns(2)
# Left column for inputs
with col1:
st.subheader("Input 👇")
prompt = st.text_input("Provide the topic of the video:", placeholder="e.g., How to bake a cake")
video_length = st.number_input("Specify length in minutes", min_value=1.0, step=0.5, value=10.0)
with st.sidebar:
st.title("Options")
creativity = st.slider("Set creativity level:", min_value=0.0, max_value=1.0, value=0.5)
# Button to generate the video script
generate_script_button = st.button("Generate Script")
# Right column for results
with col2:
st.subheader("Output ✨")
if generate_script_button:
if not prompt:
st.error("Please provide a topic for the video.")
else:
try:
# Generate the video script using the utility function
with st.spinner("Generating your script..."):
title, script, search_data = generate_script(prompt, video_length, creativity, api_key, serper_api_key)
# Display the results
st.success("Script generated successfully!")
st.subheader(f"Title: {title}")
st.write(f"Script: {script}")
# Save the new script directly to the JSON file
new_entry = {
"title": title,
"script": script,
"search_data": search_data,
}
script_history.append(new_entry)
save_history(script_history)
# Add a download button for the generated script
st.download_button(
label="Download Script",
data=script,
file_name="generated_script.txt",
mime="text/plain",
)
# Display additional search data
with st.expander("Show search data for the script"):
st.write(search_data)
except Exception as e:
# Handle any errors that occur during script generation
st.error("An error occurred. Try again.")
with st.expander("Expand to see details"):
st.write(f"{e}")
|