fix(app): correct task checklist generation and update token handling

- Fix the logic to include only tasks with a value of 1 from the parsed dictionary.
- Remove unnecessary null check from the task checklist generation logic.
- Update the recovery token input to reflect changes in the checklist generation function.
- Ensure proper component initialization within the Gradio Blocks context.
- Streamline the submit button click function to process data and update UI components.
- Update dependency management to include `pytest` and fix `iniconfig` and `pluggy` versions.

app.py

@@ -120,9 +120,7 @@ def show_checkbox_groups(token):
                 tasks,
                 label=f"{row['name']}",
                 value=[
-                    tasks[i]
-                    for i, v in enumerate(parsed_dict[row["plant"]])
-                    if v == 1 and df.notnull().at[index, TASKS[i]]
+                    tasks[i] for i, v in enumerate(parsed_dict[row["plant"]]) if v == 1
                 ],
                 type="index",
             )
@@ -336,81 +334,81 @@ def generate_img_by_plotly(df):
     return image
 
 
-with gr.Blocks() as app:
-    gr.Markdown(
-        """
-        <center><font size=8>👩🏻‍🌾Herbology Almanac Checklist Generator📝</font></center>
-        This is a simple web app that generates an almanac checklist for your plants.
-        """
-    )
+if __name__ == "__main__":
+    with gr.Blocks() as app:
+        gr.Markdown(
+            """
+            <center><font size=8>👩🏻‍🌾Herbology Almanac Checklist Generator📝</font></center>
+            This is a simple web app that generates an almanac checklist for your plants.
+            """
+        )
 
-    gr.Markdown(
-        """
-        # RECOVERY TOKEN
-        """
-    )
-    recovery_token = gr.Textbox(
-        value="",
-        label="Recovery Token",
-        info="Save this token or paste your saved one here",
-        placeholder="Keep this token to restore your previous input".upper(),
-        interactive=True,
-    )
+        gr.Markdown(
+            """
+            # RECOVERY TOKEN
+            """
+        )
+        recovery_token = gr.Textbox(
+            value="",
+            label="Recovery Token",
+            info="Save this token or paste your saved one here",
+            placeholder="Keep this token to restore your previous input".upper(),
+            interactive=True,
+        )
 
-    gr.Markdown(
-        """
-        # YOUR RESEARCH TASKS
-        """
-    )
-    checklist_inputs = show_checkbox_groups(recovery_token.value)
+        gr.Markdown(
+            """
+            # YOUR RESEARCH TASKS
+            """
+        )
+        checklist_inputs = show_checkbox_groups(recovery_token.value)
 
-    gr.Markdown(
-        """
-        # IMAGE STYLE
-        """
-    )
-    plot_library = gr.Radio(
-        ["plotly", "matplotlib"],
-        label="Plot Library",
-        value="plotly",
-        info="Choose your plot library",
-    )
+        gr.Markdown(
+            """
+            # IMAGE STYLE
+            """
+        )
+        plot_library = gr.Radio(
+            ["plotly", "matplotlib"],
+            label="Plot Library",
+            value="plotly",
+            info="Choose your plot library",
+        )
 
-    submit_button = gr.Button("Generate Image and Token")
+        submit_button = gr.Button("Generate Image and Token")
 
-    # df_out = gr.Dataframe(label="Output Dataframe", interactive=False)
+        # df_out = gr.Dataframe(label="Output Dataframe", interactive=False)
 
-    generated_img = gr.Image(label="Generated Image", format="png", type="pil")
+        generated_img = gr.Image(label="Generated Image", format="png", type="pil")
 
-    logs = gr.Markdown(
+        logs = gr.Markdown(
+            """
+            # CHANGELOG
+            - 2024/08/31: Initial release
+            - 2024/09/03: Fix a mistake in the tasks of mimbulus
+            - 2024/09/04: Correct Radiant count for water hyacinth
+            - 2024/09/05: Support image generated by plotly
+            - 2024/09/13: Update Water Lily tasks and color options
         """
-        # CHANGELOG
-        - 2024/08/31: Initial release
-        - 2024/09/03: Fix a mistake in the tasks of mimbulus
-        - 2024/09/04: Correct Radiant count for water hyacinth
-        - 2024/09/05: Support image generated by plotly
-        - 2024/09/13: Update Water Lily tasks and color options
-    """
-    )
-
-    submit_button.click(
-        process_data,
-        inputs=checklist_inputs + [plot_library],
-        outputs=[generated_img, recovery_token],
-    )
+        )
 
-    recovery_token.change(
-        show_checkbox_groups,
-        inputs=[recovery_token],
-        outputs=checklist_inputs,
-    )
+        submit_button.click(
+            process_data,
+            inputs=checklist_inputs + [plot_library],
+            outputs=[generated_img, recovery_token],
+        )
 
-    # generate_button.click(
-    #     generate_img,
-    #     inputs=[df_out],
-    #     outputs=[generated_img],
-    # )
+        recovery_token.change(
+            show_checkbox_groups,
+            inputs=[recovery_token],
+            outputs=checklist_inputs,
+        )
 
+        # generate_button.click(
+        #     generate_img,
+        #     inputs=[df_out],
+        #     outputs=[generated_img],
+        # )
 
-app.queue()
-app.launch()
+    app.queue()
+    app.launch()

poetry.lock

@@ -652,6 +652,17 @@ enabler = ["pytest-enabler (>=2.2)"]
 test = ["jaraco.test (>=5.4)", "pytest (>=6,!=8.1.*)", "zipp (>=3.17)"]
 type = ["pytest-mypy"]
 
+[[package]]
+name = "iniconfig"
+version = "2.0.0"
+description = "brain-dead simple config-ini parsing"
+optional = false
+python-versions = ">=3.7"
+files = [
+    {file = "iniconfig-2.0.0-py3-none-any.whl", hash = "sha256:b6a85871a79d2e3b22d2d1b94ac2824226a63c6b741c88f7ae975f18b6778374"},
+    {file = "iniconfig-2.0.0.tar.gz", hash = "sha256:2d91e135bf72d31a410b17c16da610a82cb55f6b0477d1a902134b24a455b8b3"},
+]
+
 [[package]]
 name = "jinja2"
 version = "3.1.4"
@@ -1333,6 +1344,21 @@ files = [
 packaging = "*"
 tenacity = ">=6.2.0"
 
+[[package]]
+name = "pluggy"
+version = "1.5.0"
+description = "plugin and hook calling mechanisms for python"
+optional = false
+python-versions = ">=3.8"
+files = [
+    {file = "pluggy-1.5.0-py3-none-any.whl", hash = "sha256:44e1ad92c8ca002de6377e165f3e0f1be63266ab4d554740532335b9d75ea669"},
+    {file = "pluggy-1.5.0.tar.gz", hash = "sha256:2cffa88e94fdc978c4c574f15f9e59b7f4201d439195c3715ca9e2486f1d0cf1"},
+]
+
+[package.extras]
+dev = ["pre-commit", "tox"]
+testing = ["pytest", "pytest-benchmark"]
+
 [[package]]
 name = "pydantic"
 version = "2.8.2"
@@ -1495,6 +1521,26 @@ files = [
 [package.extras]
 diagrams = ["jinja2", "railroad-diagrams"]
 
+[[package]]
+name = "pytest"
+version = "8.3.3"
+description = "pytest: simple powerful testing with Python"
+optional = false
+python-versions = ">=3.8"
+files = [
+    {file = "pytest-8.3.3-py3-none-any.whl", hash = "sha256:a6853c7375b2663155079443d2e45de913a911a11d669df02a50814944db57b2"},
+    {file = "pytest-8.3.3.tar.gz", hash = "sha256:70b98107bd648308a7952b06e6ca9a50bc660be218d53c257cc1fc94fda10181"},
+]
+
+[package.dependencies]
+colorama = {version = "*", markers = "sys_platform == \"win32\""}
+iniconfig = "*"
+packaging = "*"
+pluggy = ">=1.5,<2"
+
+[package.extras]
+dev = ["argcomplete", "attrs (>=19.2)", "hypothesis (>=3.56)", "mock", "pygments (>=2.7.2)", "requests", "setuptools", "xmlschema"]
+
 [[package]]
 name = "python-dateutil"
 version = "2.9.0.post0"
@@ -2056,4 +2102,4 @@ files = [
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.12"
-content-hash = "53f7ebd990eb3771759d3adf626d8fbe951bd6fc6a77c8ca2df15eb3bb7a5aea"
+content-hash = "a8f9e09ac59568c26e085b319749b7dc9fd7363892a41b1dfc04e96fa3b406fd"

pyproject.toml

@@ -11,6 +11,7 @@ gradio = "4.38.1"
 uvicorn = "^0.30.6"
 plotly = "^5.24.0"
 pytz = "^2024.1"
+pytest = "^8.3.3"
 
 
 [build-system]

test_token.py

@@ -0,0 +1,9 @@
+from app import bin_ls2base64, base64_to_binary
+from numpy import random
+
+
+def test_translation():
+    for _ in range(100):
+        ls0 = random.randint(0, 2, 24 * 10).tolist()
+        ls1 = [int(n) for n in base64_to_binary(bin_ls2base64(ls0))]
+        assert ls0 == ls1