now taking into account saved details when prompting and parsing the answer of the LLM

app.py

@@ -8,7 +8,7 @@ from streamlit.runtime.scriptrunner_utils.exceptions import StopException
 from form.form import build_form_data_from_answers, write_pdf_form, work_categories
 from llm_manager.llm_parser import LlmParser
 from local_storage.entities import PersonalDetails, LocationDetails, ContractorDetails
-from local_storage.ls_manager import save_details, get_detail
+from local_storage.ls_manager import LocalStorageManager
 from prompts.prompts_manager import PromptsManager
 from enums import Questions as Q
 from repository.repository import build_repo_from_environment, get_repository
@@ -42,8 +42,9 @@ class UIManager:
         self.pm: PromptsManager = PromptsManager(work_categories=work_categories)
         self.repository = (build_repo_from_environment(self.pm.system_prompt) or
                            get_repository("testing",
-                                          Model("fakeModel", ModelRoles("a", "b", "c"))))
-
+                                          Model("fakeModel",
+                                                ModelRoles("a", "b", "c"))))
+        self.lsm = None
     @staticmethod
     def get_current_step() -> int:
         try:
@@ -76,8 +77,9 @@ class UIManager:
     def build_ui_for_parsing_answers(self):
         self._build_base_ui()
         with st.status("parsing user input for tags"):
+            self.lsm = LocalStorageManager() if not self.lsm else self.lsm
             tags = find_tags_regex.findall(ss["user_input"])
-            details = [get_detail(t) for t in tags]
+            details = [self.lsm.get_detail(t) for t in tags]
         with st.status("initialising LLM"):
             self.repository.init()
         with st.status("waiting for LLM"):
@@ -140,7 +142,7 @@ class UIManager:
             details = func(details_key)
             if details:
                 key = ss[details_key]  # get the name under which this data should be saved
-                save_details(details, key)
+                self.lsm.save_details(details, key)
         self.set_current_step(Steps.FIND_CATEGORIES)
         st.rerun()
 

local_storage/ls_manager.py

@@ -1,39 +1,38 @@
-import json
-
 from streamlit_local_storage import LocalStorage
 
 from local_storage.entities import SavedDetails, PersonalDetails, LocationDetails, ContractorDetails
 from enums import DetailsType
 
 
-def get_detail(key: str):
-    for detail_type in DetailsType.values():
-        detail = LocalStorage(key=detail_type.name).getItem(detail_type.name).get(key)
-        if detail:
-            if detail_type.value == DetailsType.PERSONAL_DETAILS.value:
-                return PersonalDetails(**detail)
-            elif detail_type.value == DetailsType.LOCATION_DETAILS.value:
-                return LocationDetails(**detail)
-            elif detail_type.value == DetailsType.CONTRACTOR_DETAILS.value:
-                return ContractorDetails(**detail)
-            return None
-    return None
-
-
-def save_details(details: SavedDetails, key: str):
-    if isinstance(details, PersonalDetails):
-        type_ = DetailsType.PERSONAL_DETAILS
-    elif isinstance(details, LocationDetails):
-        type_ = DetailsType.LOCATION_DETAILS
-    elif isinstance(details, ContractorDetails):
-        type_ = DetailsType.CONTRACTOR_DETAILS
-    else:
-        raise ValueError("Unexpected type: {}", type(details))
-    existing_data = (ls.getItem(type_.name) or {})
-    existing_data[key] = details.to_json()
-    ls.setItem(type_.name, existing_data, type_.name)
-
-
-def get_details(type_: DetailsType):
-    return ls.getItem(type_.name) or {}
-
+class LocalStorageManager:
+    def __init__(self):
+        self.ls: LocalStorage = LocalStorage()
+
+    def get_detail(self, key: str):
+        for detail_type in DetailsType.values():
+            detail = self.ls.getItem(detail_type.name).get(key)
+            if detail:
+                if detail_type.value == DetailsType.PERSONAL_DETAILS.value:
+                    return PersonalDetails(**detail)
+                elif detail_type.value == DetailsType.LOCATION_DETAILS.value:
+                    return LocationDetails(**detail)
+                elif detail_type.value == DetailsType.CONTRACTOR_DETAILS.value:
+                    return ContractorDetails(**detail)
+                return None
+        return None
+
+    def save_details(self, details: SavedDetails, key: str):
+        if isinstance(details, PersonalDetails):
+            type_ = DetailsType.PERSONAL_DETAILS
+        elif isinstance(details, LocationDetails):
+            type_ = DetailsType.LOCATION_DETAILS
+        elif isinstance(details, ContractorDetails):
+            type_ = DetailsType.CONTRACTOR_DETAILS
+        else:
+            raise ValueError("Unexpected type: {}", type(details))
+        existing_data = (self.ls.getItem(type_.name) or {})
+        existing_data[key] = details.to_json()
+        self.ls.setItem(type_.name, existing_data, type_.name)
+
+    def get_details(self, type_: DetailsType):
+        return self.ls.getItem(type_.name) or {}