fix: finance-content-checker logic repair (#13)

Author: yanomaly

finance-content-checker/.wf/components-page-0-c0f99a9e-5004-4e75-a6c6-36f17490b134.jsonl

@@ -0,0 +1,47 @@
+{"id": "c0f99a9e-5004-4e75-a6c6-36f17490b134", "type": "page", "content": {"pageMode": "compact"}, "handlers": {}, "isCodeManaged": false, "parentId": "root", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "bebc5fe9-63a7-46a7-b0fa-62303555cfaf", "type": "header", "content": {"text": "@{my_app.title}"}, "handlers": {}, "isCodeManaged": false, "parentId": "c0f99a9e-5004-4e75-a6c6-36f17490b134", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "oifvuxkfgni30c04", "type": "image", "content": {"caption": "", "maxHeight": "30", "maxWidth": "", "src": "@{img_path}"}, "handlers": {}, "isCodeManaged": false, "parentId": "bebc5fe9-63a7-46a7-b0fa-62303555cfaf", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "cfdn0lwtc3bc2e1i", "type": "message", "content": {"message": "@{message}"}, "handlers": {}, "isCodeManaged": false, "parentId": "c0f99a9e-5004-4e75-a6c6-36f17490b134", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "3emssu3oapg9aaeb", "type": "columns", "content": {}, "handlers": {}, "isCodeManaged": false, "parentId": "c0f99a9e-5004-4e75-a6c6-36f17490b134", "position": 2, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "0r3y4groykgktb1l", "type": "column", "content": {"width": "1"}, "handlers": {}, "isCodeManaged": false, "parentId": "3emssu3oapg9aaeb", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "hjtxdwb3aeylsfh6", "type": "section", "content": {"title": "Input content to check"}, "handlers": {}, "isCodeManaged": false, "parentId": "0r3y4groykgktb1l", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "f82m2sdbhhgfgbb1", "type": "textareainput", "binding": {"eventType": "wf-change", "stateRef": "content"}, "content": {"label": "Paste in some content to check (<500 words)", "placeholder": "", "rows": "5"}, "handlers": {"wf-change": "content_to_HMTL"}, "isCodeManaged": false, "parentId": "hjtxdwb3aeylsfh6", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "3wckjuynvz0od6w9", "type": "section", "content": {"title": ""}, "handlers": {}, "isCodeManaged": false, "parentId": "0r3y4groykgktb1l", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "c30ch4pibyw5juug", "type": "html", "content": {"htmlInside": "@{html_content}", "styles": "{\n  \"padding\": \"16px\",\n\"padding-top\":\"20px\",\n  \"min-height\": \"64px\",\n\"max-height\": \"420px\",\n\"overflow\":\"auto\",\n  \"min-width\": \"64px\",\n  \"border-radius\": \"8px\"\n}"}, "handlers": {}, "isCodeManaged": false, "parentId": "3wckjuynvz0od6w9", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "43cp015ju9156xvo", "type": "column", "content": {"width": "1"}, "handlers": {}, "isCodeManaged": false, "parentId": "3emssu3oapg9aaeb", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "z4toz7gq7kpdhbvs", "type": "multiselectinput", "binding": {"eventType": "wf-options-change", "stateRef": "suggestion_flags_selected"}, "content": {"accentColor": "gray", "chipTextColor": "", "label": "Select one or more Suggestion Flags", "maximumCount": "5", "options": "@{suggestion_flags}", "placeholder": "Select up to 3 options..."}, "handlers": {"wf-options-change": "enable_suggestion_tabs"}, "isCodeManaged": false, "parentId": "43cp015ju9156xvo", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "e9re4hpngsrtszns", "type": "button", "content": {"icon": "arrow_forward", "isDisabled": "@{generate_btn_disabled}", "text": "Generate suggestions"}, "handlers": {"wf-click": "handle_generate_button_click"}, "isCodeManaged": false, "parentId": "43cp015ju9156xvo", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "amo4m8mssxwfztng", "type": "tabs", "content": {}, "handlers": {}, "isCodeManaged": false, "parentId": "43cp015ju9156xvo", "position": 2, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "y2co003xc2z7ckld", "type": "tab", "content": {"name": "Outcome language"}, "handlers": {"click": "get_current_tab"}, "isCodeManaged": false, "parentId": "amo4m8mssxwfztng", "position": 0, "visible": {"binding": "tab_enabled.outcome_language", "expression": "custom", "reversed": false}}
+{"id": "8cgl5i0vbhppyh4i", "type": "columns", "content": {}, "handlers": {}, "isCodeManaged": false, "parentId": "y2co003xc2z7ckld", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "p8jduqivbny7bdxy", "type": "column", "content": {"width": "1"}, "handlers": {}, "isCodeManaged": false, "parentId": "8cgl5i0vbhppyh4i", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "hyfos0bkdr3v50nb", "type": "section", "content": {"title": "Original text"}, "handlers": {"click": "highlight_string"}, "isCodeManaged": false, "parentId": "p8jduqivbny7bdxy", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "h4k2l0kghizi1osw", "type": "message", "content": {"message": "@{output.outcome_language.text}"}, "handlers": {"change": "highlight_string"}, "isCodeManaged": false, "parentId": "hyfos0bkdr3v50nb", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "atmnc5edtlm52idp", "type": "section", "content": {"title": "Suggestion"}, "handlers": {}, "isCodeManaged": false, "parentId": "p8jduqivbny7bdxy", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "fli8kybxrmqrrfaw", "type": "message", "content": {"message": "+@{output.outcome_language.suggestion}"}, "handlers": {}, "isCodeManaged": false, "parentId": "atmnc5edtlm52idp", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "shhxankpkgzxdp6o", "type": "column", "content": {"width": "1"}, "handlers": {}, "isCodeManaged": false, "parentId": "8cgl5i0vbhppyh4i", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "v41sm3g27nf7qxsx", "type": "tags", "content": {"tags": "{\n  \"outcome_language\": \"outcome language\"\n}"}, "handlers": {}, "isCodeManaged": false, "parentId": "shhxankpkgzxdp6o", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "ijghkvhd8yxwxboq", "type": "message", "content": {"message": "!@{output.outcome_language.description}"}, "handlers": {}, "isCodeManaged": false, "parentId": "shhxankpkgzxdp6o", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "1h3fup2ixqij4zzl", "type": "pagination", "content": {"page": "@{pagination.outcome_language.current_page}", "pageSize": "1", "pageSizeOptions": "", "totalItems": "@{pagination.outcome_language.data_size}"}, "handlers": {"wf-change-page": "set_page"}, "isCodeManaged": false, "parentId": "y2co003xc2z7ckld", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "kdx0s5tf0t6t272y", "type": "tab", "content": {"name": "Hyperbole"}, "handlers": {"click": "get_current_tab"}, "isCodeManaged": false, "parentId": "amo4m8mssxwfztng", "position": 1, "visible": {"binding": "tab_enabled.hyperbole", "expression": "custom", "reversed": false}}
+{"id": "4ut5mwq967zm0049", "type": "columns", "content": {}, "handlers": {}, "isCodeManaged": false, "parentId": "kdx0s5tf0t6t272y", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "p6hezc459yfkcvrb", "type": "column", "content": {"width": "1"}, "handlers": {}, "isCodeManaged": false, "parentId": "4ut5mwq967zm0049", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "9n7r2yllx9h6d9vh", "type": "section", "content": {"title": "Original text"}, "handlers": {}, "isCodeManaged": false, "parentId": "p6hezc459yfkcvrb", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "giyebjqjd4tuojwy", "type": "message", "content": {"message": "@{output.hyperbole.text}"}, "handlers": {}, "isCodeManaged": false, "parentId": "9n7r2yllx9h6d9vh", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "8avbiu3lz7hzk4u4", "type": "section", "content": {"title": "Suggestion"}, "handlers": {}, "isCodeManaged": false, "parentId": "p6hezc459yfkcvrb", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "cumxx6s29p0kf3p1", "type": "message", "content": {"message": "+@{output.hyperbole.suggestion}"}, "handlers": {}, "isCodeManaged": false, "parentId": "8avbiu3lz7hzk4u4", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "jdjc5g79v7qffben", "type": "column", "content": {"width": "1"}, "handlers": {}, "isCodeManaged": false, "parentId": "4ut5mwq967zm0049", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "smuxxei8eqmzd3i5", "type": "tags", "content": {"tags": "{\n  \"Hyperbole\": \"hyperbole\"\n}"}, "handlers": {}, "isCodeManaged": false, "parentId": "jdjc5g79v7qffben", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "fhomxd6iobyhvq22", "type": "message", "content": {"message": "!@{output.hyperbole.description}"}, "handlers": {}, "isCodeManaged": false, "parentId": "jdjc5g79v7qffben", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "br9ri32d6dwg6btf", "type": "pagination", "content": {"page": "@{pagination.hyperbole.current_page}", "pageSize": "1", "pageSizeOptions": "", "totalItems": "@{pagination.hyperbole.data_size}"}, "handlers": {"wf-change-page": "set_page"}, "isCodeManaged": false, "parentId": "kdx0s5tf0t6t272y", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "1ntjnjorgokmrksh", "type": "tab", "content": {"name": "We pronoun"}, "handlers": {"click": "get_current_tab"}, "isCodeManaged": false, "parentId": "amo4m8mssxwfztng", "position": 2, "visible": {"binding": "tab_enabled.we_pronoun", "expression": "custom", "reversed": false}}
+{"id": "220abedg02oxr0df", "type": "columns", "content": {}, "handlers": {}, "isCodeManaged": false, "parentId": "1ntjnjorgokmrksh", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "ash4sqj4675xby5u", "type": "column", "content": {"width": "1"}, "handlers": {}, "isCodeManaged": false, "parentId": "220abedg02oxr0df", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "pz1uzfd2sgvj35wf", "type": "section", "content": {"title": "Original text"}, "handlers": {}, "isCodeManaged": false, "parentId": "ash4sqj4675xby5u", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "wufcwgrrybpweoc2", "type": "message", "content": {"message": "@{output.we_pronoun.text}"}, "handlers": {}, "isCodeManaged": false, "parentId": "pz1uzfd2sgvj35wf", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "4euoiqzlcfgkrd51", "type": "section", "content": {"title": "Suggestion"}, "handlers": {}, "isCodeManaged": false, "parentId": "ash4sqj4675xby5u", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "gg5gv7ukdcbp0ys7", "type": "message", "content": {"message": "+@{output.we_pronoun.suggestion}"}, "handlers": {}, "isCodeManaged": false, "parentId": "4euoiqzlcfgkrd51", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "0jydi26cfk0lj4zg", "type": "column", "content": {"width": "1"}, "handlers": {}, "isCodeManaged": false, "parentId": "220abedg02oxr0df", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "hddj4kf5pbu7nltg", "type": "tags", "content": {"tags": "{\n  \"we_pronoun\": \"we pronoun\"\n}"}, "handlers": {}, "isCodeManaged": false, "parentId": "0jydi26cfk0lj4zg", "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "tbherv4idvftfhr5", "type": "message", "content": {"message": "!@{output.we_pronoun.description}"}, "handlers": {}, "isCodeManaged": false, "parentId": "0jydi26cfk0lj4zg", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}
+{"id": "tu8ujfgmwm1cviiu", "type": "pagination", "content": {"page": "@{pagination.we_pronoun.current_page}", "pageSize": "1", "pageSizeOptions": "", "totalItems": "@{pagination.we_pronoun.data_size}"}, "handlers": {"wf-change-page": "set_page"}, "isCodeManaged": false, "parentId": "1ntjnjorgokmrksh", "position": 1, "visible": {"binding": "", "expression": true, "reversed": false}}

finance-content-checker/.wf/components-root.jsonl

@@ -0,0 +1 @@
+{"id": "root", "type": "root", "content": {"appName": "Finance Copy Compliance Checker"}, "handlers": {}, "isCodeManaged": false, "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}

finance-content-checker/.wf/components-workflows_root.jsonl

@@ -0,0 +1 @@
+{"id": "workflows_root", "type": "workflows_root", "content": {}, "handlers": {}, "isCodeManaged": false, "position": 0, "visible": {"binding": "", "expression": true, "reversed": false}}

finance-content-checker/.wf/metadata.json

@@ -0,0 +1,3 @@
+{
+    "writer_version": "0.8.2"
+}

finance-content-checker/main.py

@@ -1,25 +1,33 @@
+import asyncio
+
 import writer as wf
+from writerai import AsyncWriter
+from dotenv import load_dotenv
 import utils as ut
 from prompts import _hyperbole_prompt, _outcome_language_prompt, _we_pronoun_prompt, input_text, initial_output
 
+load_dotenv()
+
 class Rule:
     def __init__(self, name, prompt_function, state):
         self.name = name
         self.prompt_function = prompt_function
         self.state = state
 
-    def fetch_data(self):
+    async def fetch_data(self):
         formatted_content = self.state["content"].replace('\n\n', ' ')
         prompt = self.prompt_function(formatted_content)
-        output = wf.ai.complete(prompt)
-        data = ut.convert_string_to_dict_list(output)
+        output = await AsyncWriter().completions.create(
+            model="palmyra-x-004", prompt=prompt
+        )
+        data = ut.convert_string_to_dict_list(output.choices[0].text)
         if data:
             first_entry = data[0]
             ut.set_page_output(first_entry, self.state, self.name)
         self.state[self.name] = data
         self.state["pagination"][self.name]["data_size"] = len(data)
 
-def process_rules(state):
+async def process_rules(state):
     """Process all rules and update the state accordingly."""
 
     # Define rules
@@ -29,20 +37,27 @@ def process_rules(state):
         Rule("we_pronoun", _we_pronoun_prompt, state)
     ]
 
+    tasks = []
+
+    state["message"] = "Performing analysis..."
+
     # Fetch data for each rule
     for rule in rules:
         if(rule.name in state["suggestion_flags_selected"]):
             formatted_name = rule.name.replace("_", " ")
-            state["message"] = f"Processing {formatted_name}..."
-            rule.fetch_data()
+            tasks.append(rule.fetch_data())
+
+    await asyncio.gather(*tasks)
 
 
 
 # Event handlers
 # Handler to add/remove tabs from being displayed when suggestion flags are selected. Called by change event on multiselect suggestion flags UI component.
 def enable_suggestion_tabs(state,context):
     """
-    Handler to add/remove tabs from being displayed when suggestion flags are selected. Called by change event on multiselect suggestion flags UI component.
+    Handler to add/remove tabs from being displayed when suggestion
+    flags are selected. Called by change event on multiselect suggestion
+     flags UI component.
     """
     all_options = list(flags.values())
 
@@ -53,13 +68,19 @@ def enable_suggestion_tabs(state,context):
 
 
 def set_page(state,payload):
-    """This is a handler called by the pagination component when a page is clicked ("wf-change-page event). The payload returns the number of the current page "
+    """This is a handler called by the pagination component when
+    a page is clicked ("wf-change-page event). The payload returns
+    the number of the current page "
     """
     index = payload - 1
     curr_tab = state["current_tab"]
     state["pagination"][curr_tab]["current_page"] = payload
 
-    """The tab name of the current tab is used to determine which suggestion rule is in focus. This means that the tab names need to line up with the rules. The logic here is that based on the page clicked on in the active tab, we index that data dictionary and set the page output accordingly, as well as highlight the text in the html output."""
+    """The tab name of the current tab is used to determine which suggestion rule is in focus. 
+    This means that the tab names need to line up with the rules. The logic here is that based 
+    on the page clicked on in the active tab, we index that data dictionary and set the page 
+    output accordingly, as well as highlight the text in the html output.
+    """
     rule = state["current_tab"]
     data = state[rule]
     ut.set_page_output(data=data[index],state=state,rule=rule)
@@ -68,15 +89,16 @@ def set_page(state,payload):
 
 
 def content_to_HMTL(state):
-    """handler to set up the input text as html for the html element. Called whenever there is a change in the input text."""
+    """handler to set up the input text as html for the html element.
+    Called whenever there is a change in the input text."""
     formatted_content = state["content"].replace('\n\n', '<br/><br/>')
     print(formatted_content)
     html = """ <p class="description">{content}</p>""".format(content=formatted_content)
     state["html_content"] = html
     return html
 
 # Generate button click handler
-def handle_generate_button_click(state):
+async def handle_generate_button_click(state):
     # Disable button when clicked to prevent double clicking
     state["generate_btn_disabled"] = "yes"
 
@@ -85,15 +107,19 @@ def handle_generate_button_click(state):
     # This collection is stored in the state[rule] field and the 
     # number of pages is set based on the number of items in the collection.
 
-    process_rules(state)
+    await process_rules(state)
 
     # Re-enable button once processing is completed.
     state["message"] = ""
     state["generate_btn_disabled"] = "no"
     ut.highlight_string(state,state["current_tab"])
 
 def get_current_tab(state,context :dict):
-    """This is a handler that is tied to the tab click event. Return the current tab: this uses an approach that will eventually be deprecated where we look at the context and grab the target from it which gives the component id of the UI component"""
+    """This is a handler that is tied to the tab click event.
+     Return the current tab: this uses an approach that will
+    eventually be deprecated where we look at the context and
+    grab the target from it which gives the component id of
+    the UI component"""
 
     # mapping of component ids to tab names.
     tabs = {"y2co003xc2z7ckld": "outcome_language", "kdx0s5tf0t6t272y": "hyperbole", "1ntjnjorgokmrksh": "we_pronoun"}
@@ -123,7 +149,7 @@ def get_current_tab(state,context :dict):
     },
     "message": "",
     "content": input_text(),
-    "html_content":f""" <p class="description">{input_text().replace('\n\n', '<br/><br/>')}</p>""",
+    "html_content":'<p class="description">{text}</p>'.format(text=input_text().replace("\n\n", "<br/><br/>")),
     "outcome_language": "",
     "we_pronoun":"",
     "hyperbole":"",