import os

import streamlit as st
from langchain.llms import OpenAI

def load_anwer(question):
    llm = OpenAI(model_name="text-davinci-003",temperature=0)
    answer = llm(question)
    return answer

st.set_page_config(page_title="Langchain Demo")
st.header("Langchain Demo")

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

user = get_text()
response = load_anwer(user)

submit = st.button("Generate")
if submit:
    st.subheader("Answer: ")
    st.write(response)