fix input check

app.py

@@ -1,13 +1,35 @@
 """kwest.dev is a saas tool to craft stories for your ideas."""
 
+import os
+from typing import Union
+
+import requests
 import streamlit as st
 
-from api import check_if_input_is_valid
 
+API_URL = os.getenv("API_URL", None)
+AUTH_TOKEN = os.getenv("AUTH_TOKEN", None)
 RECIPES = ["sell", "motivate", "convince", "connect", "explain", "lead", "impress"]
 VALID_EMOJIS = {"yes": "✅", "no": "❌"}
 
+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) -> Union[dict, None]:
+    """Check if the input string is valid for the API."""
+    response = requests.post(
+        f"{API_URL}/check-input",
+        headers={"Authorization": f"{AUTH_TOKEN}"},
+        json={"input": input_str},
+    )
+    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. 💡")
 
@@ -36,21 +58,20 @@ input_col, btn_col =  st.columns([3, 1], vertical_alignment="bottom")
 with input_col:
     input_str = st.text_area("Your idea that needs a story", "", placeholder="I need a pitch for ...")
 with btn_col:
-    input_is_valid = st.button(
-        label="▶️",
-        key="check_input",
-        help="Check if the input is valid",
-        on_click=check_if_input_is_valid,
-        kwargs={"input_str": input_str}
-    )
+    check_input = st.button(label="▶️", key="check_input", help="Check if the input is valid")
 
-if input_is_valid is not None:
-    st.write(
-        f"Objective: {VALID_EMOJIS[input_is_valid['objective']]} | "
-        f"Context: {VALID_EMOJIS[input_is_valid['context']]} | "
-        f"Audience: {VALID_EMOJIS[input_is_valid['audience']]}"
-    )
+if check_input:
+    input_is_valid = check_if_input_is_valid(input_str)
+    if not input_is_valid:
+        st.error("An error occurred while checking the input. Please try again.")
+        st.stop()
+    else:
+        st.write(
+            f"Objective: {VALID_EMOJIS[input_is_valid['objective']]} | "
+            f"Context: {VALID_EMOJIS[input_is_valid['context']]} | "
+            f"Audience: {VALID_EMOJIS[input_is_valid['audience']]}"
+        )
 
-    if input_is_valid["objective"] == "yes" and input_is_valid["context"] == "yes" and input_is_valid["audience"] == "yes":
-        st.write("Choose the recipe you want to use for your story.")
+        if input_is_valid["objective"] == "yes" and input_is_valid["context"] == "yes" and input_is_valid["audience"] == "yes":
+            st.write("Choose the recipe you want to use for your story.")