Change to call the API inside on_change

app.py

@@ -1,10 +1,15 @@
+import logging
+
 import streamlit as st
 import os
 import requests
 
 
-@st.cache_data
+logger = logging.getLogger(__name__)
+
+
 def query(text):
+    logger.info("Query the API with the text: %s" % text)
     API_URL = "https://api-inference.huggingface.co/models/j-hartmann/emotion-english-distilroberta-base"
     headers = {"Authorization": "Bearer " + os.environ["HF_API_KEY"]}
     payload = {"inputs": text}
@@ -20,21 +25,20 @@ col1, col2 = st.columns(2)
 
 with col1:
     if st.button("Add query"):
-        st.session_state.prompts.append("")
+        st.session_state.prompts.append(("", None))
     if st.button("Remove query"):
         st.session_state.prompts.pop(0)
 
 
 def card(index: int):
     def on_change():
-        # Update the `prompts[index]` session state with the value from the text input which is specified by the key, `f"prompt_{index}"`.
-        st.session_state.prompts[index] = st.session_state[f"prompt_{index}"]
-    st.text_input("Prompt:", key=f"prompt_{index}", value=st.session_state.prompts[index], on_change=on_change)
+        prompt = st.session_state[f"prompt_{index}"]
+        result = query(prompt)
+        st.session_state.prompts[index] = (prompt, result)
 
-    prompt = st.session_state.prompts[index]
+    st.text_input("Prompt:", key=f"prompt_{index}", value=st.session_state.prompts[index][0], on_change=on_change)
 
-    # This `query` function is cached, so it will only be re-run if the input `prompt` changes.
-    result = query(prompt)
+    prompt, result = st.session_state.prompts[index]
 
     st.write(f'Prompt: "{prompt}"')
     st.json(result)