import streamlit as st

# from langchain.llms import OpenAI
from langchain.llms import HuggingFace

## Function to return the response
def load_answer(question):
    # llm = OpenAI(model_name="gpt-3.5-turbo")
    llm = HuggingFace(model_name="google/flan-t5-large")
    answer = llm(question)
    return answer

## App UI starts here
st.set_page_config(page_title="LangChain Demo", page_icon="🦜")
st.header("🦜️🔗 LangChain Demo")

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

user_input = get_text()
response = load_answer(user_input)

submit = st.button("Generate")

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

    st.write(response)