import streamlit as st

from langchain_huggingface import HuggingFaceEndpoint

import os

api_key = os.environ.get("HUGGINGFACEHUB_API_TOKEN")

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

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 user_input != "":
    response = load_answer(user_input)

submit = st.button('Generate')
if submit:
    st.subheader("Answer:")
    st.write(response)