import streamlit as st
from langchain.llms import HuggingFaceHub
import json

def getAnswer(question):
    llm = HuggingFaceHub(repo_id="google/flan-t5-large")
    response = llm(question)
    return response

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

def getQuestion():
    question = st.text_input("You: ", key="input")
    return question

question = getQuestion()

if st.button('Generate') and question:  # Check if Generate button is clicked and question is not empty
    response = getAnswer(question)
    st.subheader("Answer:")
    st.write(response)