add craft story

app.py

@@ -42,7 +42,7 @@ def get_recipe_choices(input_str: str) -> Union[dict[str, str], None]:
         return None
 
 
-def craft_story_from_recipe(input_str: str, recipe: str) -> Union[dict[str, str], None]:
+def craft_story_from_recipe(input_str: str, recipe: str) -> None:
     """Craft a story from the input string and recipe."""
     response = requests.post(
         f"{API_URL}craft-from-recipe",
@@ -50,9 +50,9 @@ def craft_story_from_recipe(input_str: str, recipe: str) -> Union[dict[str, str]
         json={"input": input_str, "recipe": recipe},
     )
     if response.status_code == 200:
-        return response.json()["output"]
+        st.session_state.story = response.json()["output"]["story"]
     else:
-        return None
+        raise ValueError("An error occurred while crafting the story.")
 
 
 # UI
@@ -106,9 +106,22 @@ if check_input:
             other_choices = st.columns(5)
 
             with choice_1:
-                recipe_1 = st.button(recipe_choices["recipe"])
+                recipe_1 = st.button(
+                    recipe_choices["recipe"],
+                    on_click=lambda: craft_story_from_recipe(input_str, recipe_choices["recipe"])
+                )
             with choice_2:
-                recipe_2 = st.button(recipe_choices["second_recipe"])
-            for idx, recipe in enumerate(RECIPES - {recipe_choices["recipe"], recipe_choices["second_recipe"]}):
+                recipe_2 = st.button(
+                    recipe_choices["second_recipe"],
+                    on_click=lambda: craft_story_from_recipe(input_str, recipe_choices["second_recipe"])
+                )
+            for idx, recipe in enumerate(RECIPES - [recipe_choices["recipe"], recipe_choices["second_recipe"]]):
                 with other_choices[idx]:
-                    other_recipe = st.button(recipe)
+                    other_recipe = st.button(
+                        recipe,
+                        on_click=lambda: craft_story_from_recipe(input_str, recipe)
+                    )
+
+if "story" in st.session_state:
+    st.markdown(st.session_state.story)
+    del st.session_state.story