import streamlit as st

from langchain.llms import HuggingFaceEndpoint

def load_answer(question):
    llm = HuggingFaceEndpoint(repo_id="mistralai/Mistral-7B-instruct-v0.2" )
    answer = llm.invoke(question)
    return answer


st.set_page_config(page_title = "Langchain Chatbot" , 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()

submit = st.button('Generate')

if submit:
    if not user_input.strip():
        st.write("Please enter a question. ")

    else:
        response= load_answer(user_input)
        st.subheader("Answer: ")
        st.write(response)