add recipe choice

app.py

@@ -1,6 +1,7 @@
 """kwest.dev is a saas tool to craft stories for your ideas."""
 
 import os
+from typing import Union
 
 import requests
 import streamlit as st
@@ -15,7 +16,7 @@ if not API_URL or not AUTH_TOKEN:
     raise ValueError("API_URL and AUTH_TOKEN secrets must be set in the environment variables.")
 
 
-def check_if_input_is_valid(input_str: str) -> bool:
+def check_if_input_is_valid(input_str: str) -> Union[dict[str, str], None]:
     """Check if the input string is valid for the API."""
     response = requests.post(
         f"{API_URL}input-check",
@@ -28,6 +29,32 @@ def check_if_input_is_valid(input_str: str) -> bool:
         return None
 
 
+def get_recipe_choices(input_str: str) -> Union[dict[str, str], None]:
+    """Get the recipe choices for the input string."""
+    response = requests.post(
+        f"{API_URL}recipe-choice",
+        headers={"Authorization": f"{AUTH_TOKEN}", "Accept": "application/json"},
+        json={"input": input_str},
+    )
+    if response.status_code == 200:
+        return response.json()["output"]["recipes"]
+    else:
+        return None
+
+
+def craft_story_from_recipe(input_str: str, recipe: str) -> Union[dict[str, str], None]:
+    """Craft a story from the input string and recipe."""
+    response = requests.post(
+        f"{API_URL}craft-from-recipe",
+        headers={"Authorization": f"{AUTH_TOKEN}", "Accept": "application/json"},
+        json={"input": input_str, "recipe": recipe},
+    )
+    if response.status_code == 200:
+        return response.json()["output"]
+    else:
+        return None
+
+
 # UI
 st.title("kwest.dev")
 st.subheader("Leverage the power of storytelling for any idea. 💡")
@@ -72,5 +99,16 @@ if check_input:
         )
 
         if input_is_valid["objective"] == "yes" and input_is_valid["context"] == "yes" and input_is_valid["audience"] == "yes":
+            recipe_choices = get_recipe_choices(input_str)
             st.write("Choose the recipe you want to use for your story.")
 
+            choice_1, choice_2 = st.columns(2)
+            other_choices = st.columns(5)
+
+            with choice_1:
+                recipe_1 = st.button(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"]}):
+                with other_choices[idx]:
+                    other_recipe = st.button(recipe)