import streamlit as st
from langchain_community.llms import HuggingFaceHub

# Function to return the response
def load_answer(question):
    llm = HuggingFaceHub(repo_id="google/flan-t5-large")
    answer = llm(question)
    return answer

# App UI starts here
st.set_page_config(page_title="Simple Question & Answer App", page_icon=":robot:")
st.header("Simple Question & Answer App")

# Function to get user input
def get_text():
    input_text = st.text_input("You: ", key="input")
    return input_text

# Display text input field
user_input = get_text()

# If there is user input and the generate button is clicked
if user_input:
    submit = st.button('Generate')  

    # If generate button is clicked
    if submit:
        # Load the answer and display
        response = load_answer(user_input)
        st.subheader("Answer:")
        st.write(response)