import streamlit as st
from langchain.llms import OpenAI

#Function to return the response
def load_answer(question):
    llm = OpenAI(model_name="gpt-3.5-turbo",temperature=0)
    answer=llm(question)
    return answer


#App UI starts here
st.set_page_config(page_title="Entz's LLM LangChain-OpenAI", page_icon=":ant:")
st.header("Entz's LLM LangChain-OpenAI Base Model")

#Gets the user input
def question():
    input_text = st.text_input("Enter your question: ", key="input")
    return input_text


user_input = question()
response = load_answer(user_input)

submit = st.button('Entz AI Answer')  

#If generate button is clicked
if submit:

    st.subheader("According to Entz AI:")

    st.write(response)