import streamlit as st
from dotenv import load_dotenv
import os
from utils import generate_script

# Load environment variables
load_dotenv()

# App title and header
st.title("YouTube Scriptwriting Tool")
st.header("Generate a video script by specifying a topic, length, and creativity level.")

# Sidebar for API configuration
st.sidebar.title("API Configuration")
google_api_key = st.sidebar.text_input("Enter your Gemini API key:", type="password", key="api_key")

# Main input fields
prompt = st.text_input("Provide the topic of the video:")
video_length = st.number_input("Specify video length (in minutes):", min_value=1.0, step=0.5)
creativity = st.slider("Set creativity level:", min_value=0.0, max_value=1.0, value=0.5)

# Button to generate script
if st.button("Generate Script"):
    if not google_api_key:
        st.error("Please provide a valid Gemini API key.")
    elif not prompt:
        st.error("Topic cannot be empty.")
    else:
        # Call the script generation utility
        try:
            title, script, search_data = generate_script(prompt, video_length, creativity, google_api_key)
            st.success("Script generated successfully!")
            st.subheader(f"Title: {title}")
            st.write(f"Script:\n\n{script}")
            with st.expander("Show search data used for the script"):
                st.write(search_data)
        except Exception as e:
            st.error(f"An error occurred: {e}")