Update app.py

app.py

@@ -29,6 +29,8 @@ if "error_message" not in st.session_state:
     st.session_state.error_message = ""
 if "cfu_explain_text" not in st.session_state:
     st.session_state.cfu_explain_text = ""
+if "stick_clicked" not in st.session_state:
+    st.session_state.stick_clicked = False
 
 if "fortune_data" not in st.session_state:
     try:
@@ -37,9 +39,6 @@ if "fortune_data" not in st.session_state:
         st.error(f"Error loading CSV: {e}")
         st.session_state.fortune_data = None
 
-if "stick_clicked" not in st.session_state:
-    st.session_state.stick_clicked = False
-
 def load_finetuned_classifier_model(question):
     label_list = ["Geomancy", "Lost Property", "Personal Well-Being", "Future Prospect", "Traveling"]
     # Create a mapping dictionary to convert the default "LABEL_x" output.
@@ -71,7 +70,6 @@ def analysis(row_detail, classifiy, question):
         result = match.group(1)
         # If you want to generate a custom answer, you can call generate_answer()
         return generate_answer(question, result)
-        # return result
     else:
         return "Heaven's secret cannot be revealed."
 
@@ -99,7 +97,7 @@ def submit_text_callback():
         return
 
     if st.session_state.button_count_temp == 0:
-        st.session_state.error_message = "Please take a moment to quietly reflect on your question in your mind, then click submit."
+        st.session_state.error_message = "Please take a moment to quietly reflect on your question in your mind, then click submit again!"
         st.session_state.button_count_temp = 1
         return
 
@@ -133,10 +131,6 @@ def submit_text_callback():
                 "HeaderLink": None
             }
     print(row_detail)
-    classifiy = load_finetuned_classifier_model(question)
-    cfu_explain = analysis(row_detail, classifiy, question)
-    # Save the returned value in session state for later display
-    st.session_state.cfu_explain_text = cfu_explain
 
 def load_and_resize_image(path, max_size=MAX_SIZE):
     try:
@@ -160,6 +154,18 @@ def download_and_resize_image(url, max_size=MAX_SIZE):
         return None
 
 def stick_enquiry_callback():
+    # Retrieve the user's question and the fortune detail
+    question = st.session_state.get("user_sentence", "")
+    if not st.session_state.fortune_row:
+        st.error("Fortune data is not available. Please submit your question first.")
+        return
+    row_detail = st.session_state.fortune_row.get("Detail", "No detail available.")
+    # Run the classifier model after the image has loaded
+    classifiy = load_finetuned_classifier_model(question)
+    # Generate the explanation using the analysis function
+    cfu_explain = analysis(row_detail, classifiy, question)
+    # Save the returned value in session state for later display
+    st.session_state.cfu_explain_text = cfu_explain
     st.session_state.stick_clicked = True
 
 # Main layout: Left (input) and Right (fortune display)