model-pick / app.py
anmolsahai's picture
file
c8891d8
raw
history blame
1.76 kB
import streamlit as st
from langchain_pipeline import pipeline, model_names
st.title("Canarie AI Prototype")
st.subheader("Finding the canarie in the coal mine")
model_name = st.selectbox(
"Model",
model_names())
balance_type = st.selectbox(
"Do you charge on available balance or ledger balance?",
["available balance", "ledger balance"]
)
apsn_transactions = st.selectbox(
"Do you charge for APSN transactions?",
["yes", "no"]
)
max_fees_per_day = st.number_input(
"How many overdraft fees per day can be charged?",
min_value=0, max_value=10,
)
min_overdrawn_fee = st.number_input(
"What is the minimum amount overdrawn to incur a fee?",
min_value=0, max_value=500
)
min_transaction_overdraft = st.number_input(
"What is the minimum transaction amount to trigger an overdraft?",
min_value=0, max_value=500
)
uploaded_file = st.file_uploader("Choose a file")
if uploaded_file is not None:
with st.spinner('Please wait ...'):
try:
output_path = pipeline(
uploaded_file,
model_name,
balance_type,
apsn_transactions,
max_fees_per_day,
min_overdrawn_fee,
min_transaction_overdraft
)
st.success(f"Redlined document saved to {output_path}")
with open(output_path, "rb") as file:
st.download_button(
label="Download Redlined Document",
data=file,
file_name="Redlined_Reg_E_Notice.docx",
mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
)
except Exception as e:
st.exception(e)