app.py

import streamlit as st

from langchain.llms import OpenAI

#Function to return response
def load_answer(question):
    llm = OpenAI(model_name="text-davinci-003", temperature=0)
    answer=llm(question)
    return answer

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

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

user_input=get_input()
response = load_answer(user_input)

submit = st.button('Generate')

#If generate button is clicked
if submit:
    st.subheader("Answer:")
    st.write(response)