import streamlit as st

from langchain.llms import HuggingFaceHub

def load_answer(question):
    llm = HuggingFaceHub(repo_id = "google/flan-t5-large")
    ans = llm.invoke(question)
    return ans

st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
st.header("LangChain Demo")

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

user_input=get_text()

if not user_input:
    st.warning("Please enter a question before generating")
else:
    response=load_answer(user_input)

submit=st.button('Generate')

if submit:
    st.subheader('Answer:')
    st.write(response)