prevents error from crashing hte entire site

app.py

@@ -19,7 +19,7 @@ def run_with_context(func: Callable) -> Callable:
 
 def generate_questions(
     page_lower, page_higher, question_number, taxonomy
-) -> tuple[str, dict[str, Any], dict[str, Any]]:
+) -> tuple[str, dict[str, Any]]:
     if machine.value and machine.value.current_state_value == "start":
         machine.value.start_machine()  # Start the machine!
     if not question_number or question_number <= 0:
@@ -60,13 +60,17 @@ def generate_questions(
                 },
             },
         )
-    return (
-        "## Questions Ready for Download Below",
-        gr.update(
-            visible=True, value=f"{Path.cwd().joinpath('outputs/professor_guide.csv')}"
-        ),
-        gr.update(visible=False),
+    button = gr.update(
+        visible=True,
+        value=f"{Path.cwd().joinpath('outputs/professor_guide.csv')}",
     )
+    markdown = "## Questions Ready for Download Below"
+    if machine.value.errored:
+        button = gr.update()
+        markdown = "## Questions Not Ready for Download"
+        exception = "Bad Page Range Selected"
+        raise gr.Error(exception)
+    return (markdown, button)
 
 
 def create_statemachine() -> None:
@@ -115,7 +119,7 @@ with gr.Blocks() as demo:
     start_button.click(
         fn=run_with_context(generate_questions),
         inputs=[page_lower, page_higher, question_number, taxonomy],
-        outputs=[output, download_professor, start_button],
+        outputs=[output, download_professor],
     )
     download_professor.click(
         fn=run_with_context(questions_downloaded), outputs=[start_button]

outputs/professor_guide.csv

@@ -1,3 +1,3 @@
-MC,,1,Identify the typical run time for a crowdfunding campaign according to the textbook.,5,The typical run time for a crowdfunding campaign is 120 days.,The typical run time for a crowdfunding campaign is 15 weeks.,The typical run time for a crowdfunding campaign is 90 days.,The typical run time for a crowdfunding campaign is 10 days.,The typical run time for a crowdfunding campaign is 30 to 60 days.,266-269,Knowledge
-MC,,1,Identify the purpose of a 'Pretend to Own' prototype.,4,A 'Pretend to Own' prototype is used to analyze technical feasibility.,A 'Pretend to Own' prototype is used to assess production costs.,A 'Pretend to Own' prototype is used to evaluate design aesthetics.,A 'Pretend to Own' prototype is used to determine if a solution fits into the customer's day-to-day life.,A 'Pretend to Own' prototype is used to test market demand.,208-213,Knowledge
-MC,,1,Identify the primary problem Topology Eyewear aims to solve with their product.,1,Poorly fitting glasses.,Fragile frames that break easily.,High cost of prescription lenses.,Limited style options for eyewear.,Uncomfortable nose pads on glasses.,300-305,Knowledge
+MC,,1,Identify the percentage of working time the Core Team dedicates weekly.,2,The Core Team dedicates 25% of working time weekly.,The Core Team dedicates 9% of working time weekly.,The Core Team dedicates 20% of working time weekly.,The Core Team dedicates 12% of working time weekly.,The Core Team dedicates 15% of working time weekly.,68-81,Knowledge
+MC,,1,Identify the percentage of working time the Core Team spends on ceremonies.,3,The Core Team spends 10% of working time on ceremonies.,The Core Team spends 7% of working time on ceremonies.,The Core Team spends 9% of working time on ceremonies.,The Core Team spends 11% of working time on ceremonies.,The Core Team spends 8% of working time on ceremonies.,68-81,Knowledge
+MC,,1,Identify the frequency of the Biweekly Retrospective meetings as mentioned in the textbook.,5,Biweekly Retrospective meetings occur twice a month.,Biweekly Retrospective meetings occur every four weeks.,Biweekly Retrospective meetings occur once a month.,Biweekly Retrospective meetings occur every three weeks.,Biweekly Retrospective meetings occur every two weeks.,68-81,Knowledge

poetry.lock

@@ -530,7 +530,7 @@ loaders-sql = ["sqlalchemy (>=2.0.31,<3.0.0)"]
 type = "git"
 url = "https://github.com/griptape-ai/griptape.git"
 reference = "main"
-resolved_reference = "03b65295cabcea9c57343494c86c6d12ac0faa1c"
+resolved_reference = "794ee6ef4c2872dba4dbf08722a52642446b937e"
 
 [[package]]
 name = "h11"
@@ -1546,13 +1546,13 @@ image = ["Pillow"]
 
 [[package]]
 name = "pyright"
-version = "1.1.391"
+version = "1.1.392.post0"
 description = "Command line wrapper for pyright"
 optional = false
 python-versions = ">=3.7"
 files = [
-    {file = "pyright-1.1.391-py3-none-any.whl", hash = "sha256:54fa186f8b3e8a55a44ebfa842636635688670c6896dcf6cf4a7fc75062f4d15"},
-    {file = "pyright-1.1.391.tar.gz", hash = "sha256:66b2d42cdf5c3cbab05f2f4b76e8bec8aa78e679bfa0b6ad7b923d9e027cadb2"},
+    {file = "pyright-1.1.392.post0-py3-none-any.whl", hash = "sha256:252f84458a46fa2f0fd4e2f91fc74f50b9ca52c757062e93f6c250c0d8329eb2"},
+    {file = "pyright-1.1.392.post0.tar.gz", hash = "sha256:3b7f88de74a28dcfa90c7d90c782b6569a48c2be5f9d4add38472bdaac247ebd"},
 ]
 
 [package.dependencies]

uw_programmatic/base_machine.py

@@ -95,6 +95,7 @@ class UWBaseMachine(StateMachine):
         # To keep vector stores on track
         self.kb_ids = {}
         self.rejected_questions: list = []
+        self.errored: bool = False
 
         self.state_status: dict[str, bool] = {}
 
@@ -331,6 +332,8 @@ class UWBaseMachine(StateMachine):
         )
         question_machine.send("start_up")
         if question_machine.rejected:
+            if question_machine.reject_reason == "BAD KB PAGE RANGE":
+                return InfoArtifact("Bad KB Range")
             self.rejected_questions.append(question_machine.generated_question)
             return InfoArtifact("Question is Rejected")
         return TextArtifact(question_machine.generated_question)
@@ -355,6 +358,10 @@ class UWBaseMachine(StateMachine):
         questions = []
         for outputs in parent_outputs.values():
             if outputs.type == "InfoArtifact":
+                if outputs.value == "Bad KB Range":
+                    self.errored = True
+                    self.send("error_to_start")
+                    return ListArtifact([])
                 continue
             questions.append(outputs)
         return ListArtifact(questions)

uw_programmatic/config.yaml

@@ -23,6 +23,10 @@ events:
     transitions:
     - from: output_q
       to: end
+  error_to_start:
+    transitions:
+    - from: need_more_q
+      to: gather_parameters
 rulesets:
   frame_question_best_practices:
     name: Frame Question with KB

uw_programmatic/single_question_machine.py

@@ -78,6 +78,7 @@ TRANSITIONS = [
             {"from": "start", "to": "random_selection"},
         ],
     },
+    {"event": "end_state", "transitions": [{"from": "random_selection", "to": "end"}]},
 ]
 RULESETS = {
     "specific_question_creator": [
@@ -211,6 +212,12 @@ class SingleQuestion(StateMachine):
         self.taxonomy_prompt = taxonomy_prompt[self.taxonomy]
         # get the random page range and GTCVectorStoreDriver
         pages, driver = self.get_vector_store_id_from_page()
+        if driver is None:
+            self.send("end_state")
+            self.rejected = True
+            self.reject_reason = "BAD KB PAGE RANGE"
+            print(self.reject_reason)
+            return
         self.pages = pages
         self.driver = driver
         self.send("next_state")
@@ -405,7 +412,7 @@ class SingleQuestion(StateMachine):
 
     def get_vector_store_id_from_page(
         self,
-    ) -> tuple[str, GriptapeCloudVectorStoreDriver]:
+    ) -> tuple[str, GriptapeCloudVectorStoreDriver | None]:
         possible_kbs = {}
         for name, kb_id in self.kb_ids.items():
             page_nums = name.split("p")[1:]
@@ -413,6 +420,8 @@ class SingleQuestion(StateMachine):
             end_page = int(page_nums[1])
             if end_page <= self.page_range[1] and start_page >= self.page_range[0]:
                 possible_kbs[kb_id] = f"{start_page}-{end_page}"
+        if not len(list(possible_kbs.keys())):
+            return ("No KBs in range", None)
         kb_id = random.choice(list(possible_kbs.keys()))
         page_value = possible_kbs[kb_id]
         return page_value, GriptapeCloudVectorStoreDriver(

uw_programmatic/uw_machine.py

@@ -33,6 +33,7 @@ class UWMachine(UWBaseMachine):
         self.current_question_count = 0
         self.give_up_count = 0
         self.question_list = []
+        self.rejected_questions = []
 
     # The first state: Listens for Gradio and then gives us the parameters to search for.
     # Reinitializes the Give Up counter.
@@ -45,7 +46,11 @@ class UWMachine(UWBaseMachine):
                 self.page_range = parameters["page_range"]
                 self.question_number = parameters["question_number"]
                 self.taxonomy = parameters["taxonomy"]
+                self.errored = False
                 self.send("next_state")
+            case "griptape_event":
+                if event_value["structure_id"] == "create_question_workflow":
+                    pass
             case _:
                 err_msg = f"Unexpected Transition Event ID: {event_value}."
                 raise ValueError(err_msg)