import streamlit as st
from langchain_openai import ChatOpenAI
from langchain_core.prompts import ChatPromptTemplate
import os 

os.environ["OPENAI_API_KEY"] = os.getenv("k2")

st.title("Summarization Application :robot_face:")

text = st.text_area("Text Input: ", height=200)

if st.button("Submit"):
    llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
    prompt = ChatPromptTemplate.from_messages([("system", "You are a good assistant for summarization."),
                                               ("human", "{i}")])

    chain = prompt | llm

    response = chain.invoke({"i": text})

    st.write("Summary: ")
    st.write(response.content)