import streamlit as st
from langchain.llms import HuggingFaceHub


def get_answer(question,model_name="google/flan-t5-large"):
    model=HuggingFaceHub(repo_id=model_name)
    return model(question)


st.set_page_config(
    page_title="Simple Q&A App",
    page_icon=":robot:"

)

st.header("Do It!!!! Ask the Question.....")

col1,col2=st.columns([5,3])

question=col1.text_area("Question:")
model_name=col2.selectbox("Model Name",options=["google/flan-t5-large","deepset/roberta-base-squad2","AmazonScience/qanlu","gpt2-medium","EleutherAI/gpt-neo-125m"])

submit=st.button("Submit")

if submit:
    with st.spinner("In progress..."):
        response=get_answer(question,model_name)
    st.subheader("Response")
    st.write(f"{response}")