import streamlit as st 

from langchain.llms import OpenAI

#App UI starts here
st.set_page_config(page_title='LangChain Demo', page_icon=":robot:")
st.header('LangChain Demo')

#Get user input
def get_text():
    input_text = st.text_input("You: ", key="input")
    return input_text

#Generate response
def generate_response(user_input):
    llm = OpenAI(model_name="text-davinci-003", temperature=0)
    response = llm(user_input)
    return response

user_input = get_text()

submit = st.button("Generate")

if submit:
    st.subheader("Answer: ")
    response = generate_response(user_input)
    st.write(response)