import streamlit as st
from transformers import pipeline

pipe = pipeline("question-answering", model="distilbert/distilbert-base-cased-distilled-squad")
context = r"""
Hyarchis has been assisting its customers in the financial services field in managing their most important data for more than 30 years.
We help customers work smartly, efficiently, and in compliance with laws and regulations, relying on proprietary AI technologies that are unique within our markets.
Today, hundreds of Europe's top financial service providers, such as banks, accountancy firms, pension funds, and tens of thousands of users, rely on Hyarchis solutions.
We set the standard for document management, compliance, and KYC lifecycle management platforms."""

st.header(':purple[Hyarchis support bot]', divider='violet')
st.markdown(context)
question = st.text_input("Ask a question about Hyarchis:")
if question:
    result = pipe(question=question, context=context)
    st.write(result['answer'])