| 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) | |