File size: 933 Bytes
7e6626e
 
 
8553bb0
7e6626e
8553bb0
b45b155
61338ac
8553bb0
 
 
7e6626e
8553bb0
b45b155
 
7e6626e
b45b155
a91ceef
b45b155
 
 
 
 
7e6626e
8553bb0
b45b155
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
import streamlit as st
from langchain_google_genai import ChatGoogleGenerativeAI

st.title("Text Summarizer")

def generate_summary(text):
    prompt = f"Generate workable scenes from the following script: {script}"
    llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["API_KEY"])
    summary = llm.invoke(prompt)
    summary = summary.content
    return summary

with st.form("text_summarizer_form"):
    script = st.text_area("Enter the script or screenplay")
    submitted = st.form_submit_button("Generate Scenes")
    
    if submitted and script:
        scenes = generate_summary(script)
        st.subheader("Generated Scenes:")
        st.write(scenes)
    elif submitted and not script:
        st.error("Please enter a script to generate scenes.")


if st.button("Need Inspiration?"):
    st.write("Breaking down a script into scenes can help visualize the flow of the story better. Try it out!")