Spaces:
Running
Running
File size: 3,178 Bytes
887083d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import streamlit as st
from form.form import build_form_data_from_answers, write_pdf_form
from llm_manager.llm_parser import LlmParser
from utils.parsing_utils import check_for_missing_answers
def build_ui_for_initial_state(help_):
with st.form("Please describe your request"):
user_input = st.text_area("Your input", height=700, label_visibility="hidden", placeholder=help_, help=help_)
signature = st.file_uploader("Your signature", key="file_upload")
st.session_state["signature"] = signature
submit_button = st.form_submit_button()
if submit_button:
st.session_state["user_input"] = user_input
st.session_state["step"] = "parsing_answers"
st.rerun()
def build_ui_for_parsing_answers(repository, pm):
with st.status("initialising LLM"):
repository.init()
with st.status("waiting for LLM"):
answer = repository.send_prompt(pm.verify_user_input_prompt(st.session_state["user_input"]))
st.write(f"answers from LLM: {answer['content']}")
with st.status("Checking for missing answers"):
st.session_state["answers"] = LlmParser.parse_verification_prompt_answers(answer['content'])
st.session_state["missing_answers"] = check_for_missing_answers(st.session_state["answers"])
if not st.session_state.get("missing_answers"):
st.session_state["step"] = "check_category"
else:
st.session_state["step"] = "ask_again"
st.rerun()
def build_ui_for_ask_again(pm):
with st.form("form1"):
for ma in st.session_state["missing_answers"]:
st.text_input(pm.questions[ma].lower(), key=ma)
submitted = st.form_submit_button("Submit answers")
if submitted:
for ma in st.session_state["missing_answers"]:
st.session_state["answers"][ma] = st.session_state[ma]
st.session_state["step"] = "check_category"
st.rerun()
def build_ui_for_check_category(repository, pm):
with st.status("finding the work categories applicable to your work"):
answer = repository.send_prompt(pm.get_work_category(st.session_state["answers"][1]))
categories = LlmParser.parse_get_categories_answer(answer['content'])
with st.status("categories found, creating PDF form"):
form_data, filename = build_form_data_from_answers(st.session_state["answers"], categories,
st.session_state.get("signature"))
pdf_form = write_pdf_form(form_data)
pdf_form_filename = filename
st.session_state["pdf_form"] = pdf_form
st.session_state["pdf_form_filename"] = pdf_form_filename
st.session_state["step"] = "form_created"
st.rerun()
def build_ui_for_form_created():
st.download_button("download form", st.session_state["pdf_form"],
file_name=st.session_state["pdf_form_filename"], mime="application/pdf")
start_over_button = st.button("Start over")
if start_over_button:
del st.session_state["step"]
del st.session_state["pdf_form"]
del st.session_state["pdf_form_filename"]
st.rerun() |