Update app.py

app.py

@@ -6,15 +6,19 @@ from io import BytesIO
 from PIL import Image
 from transformers import pipeline
 
-pipe1 = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
-pipe2 = pipeline("text-classification", model="shahrukhx01/question-vs-statement-classifier")
+# Global pipelines
+pipe_english = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
+pipe_question = pipeline("text-classification", model="shahrukhx01/question-vs-statement-classifier")
 
 # Define maximum dimensions for the fortune image (in pixels)
 MAX_SIZE = (400, 400)
 
+# Initialize button click count in session state
+if "button_count_temp" not in st.session_state:
+    st.session_state.button_count_temp = 0
+
 # Set page configuration
 st.set_page_config(page_title="Fortuen Stick Enquiry", layout="wide")
-
 st.title("Fortuen Stick Enquiry")
 
 # Initialize session state variables
@@ -24,10 +28,12 @@ if "fortune_number" not in st.session_state:
     st.session_state.fortune_number = None
 if "fortune_row" not in st.session_state:
     st.session_state.fortune_row = None
+if "error_message" not in st.session_state:
+    st.session_state.error_message = ""
 
 if "fortune_data" not in st.session_state:
     try:
-        st.session_state.fortune_data = pd.read_csv("detail.csv")
+        st.session_state.fortune_data = pd.read_csv("/home/user/app/resources/detail.csv")
     except Exception as e:
         st.error(f"Error loading CSV: {e}")
         st.session_state.fortune_data = None
@@ -35,35 +41,35 @@ if "fortune_data" not in st.session_state:
 if "stick_clicked" not in st.session_state:
     st.session_state.stick_clicked = False
 
-# Update the check functions to use the "question" parameter properly.
 def check_sentence_is_english_model(question):
-    globals pipe1
-    # If the model predicts the label "en", we consider it English.
-    if pipe1(question)[0]['label'] == 'en':
-        return True
-    return False
+    return pipe_english(question)[0]['label'] == 'en'
 
 def check_sentence_is_question_model(question):
-    globals pipe1
-    # If the model predicts "LABEL_1", we consider it a question.
-    if pipe2(question)[0]['label'] == 'LABEL_1':
-        return True
-    return False
+    return pipe_question(question)[0]['label'] == 'LABEL_1'
 
-# Callback function for the submit button (no parameters now)
 def submit_text_callback():
-    # Get the user input from session state
     question = st.session_state.get("user_sentence", "")
+    # Clear any previous error message
+    st.session_state.error_message = ""
     
     if not check_sentence_is_english_model(question):
-        st.error("Please enter in English")
+        st.session_state.error_message = "Please enter in English!"
+        st.session_state.button_count_temp = 0
         return
 
     if not check_sentence_is_question_model(question):
-        st.error("This is not a question. Please enter another question")
+        st.session_state.error_message = "This is not a question. Please enter again!"
+        st.session_state.button_count_temp = 0
+        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.button_count_temp = 1
         return
 
     st.session_state.submitted_text = True
+    st.session_state.button_count_temp = 0  # Reset the counter once submission is accepted
+
     # Randomly generate a number from 1 to 100
     st.session_state.fortune_number = random.randint(1, 100)
     
@@ -78,7 +84,7 @@ def submit_text_callback():
                 "Luck": row.get("Luck", "N/A"),
                 "Description": row.get("Description", "No description available."),
                 "Detail": row.get("Detail", "No detail available."),
-                "HeaderLink": row.get("link", None)  # URL to the image
+                "HeaderLink": row.get("link", None)
             }
         else:
             st.session_state.fortune_row = {
@@ -89,7 +95,6 @@ def submit_text_callback():
                 "HeaderLink": None
             }
 
-# Function to load and resize local images using Pillow
 def load_and_resize_image(path, max_size=MAX_SIZE):
     try:
         img = Image.open(path)
@@ -99,7 +104,6 @@ def load_and_resize_image(path, max_size=MAX_SIZE):
         st.error(f"Error loading image: {e}")
         return None
 
-# Function to download image from URL and resize it
 def download_and_resize_image(url, max_size=MAX_SIZE):
     try:
         response = requests.get(url)
@@ -122,16 +126,13 @@ left_col, _, right_col = st.columns([3, 1, 5])
 with left_col:
     left_top = st.container()
     left_bottom = st.container()
-    # Top container: Input area and submit button
     with left_top:
-        # Text area for user input (saved into session_state with key "user_sentence")
-        user_sentence = st.text_area("Enter your question in English", key="user_sentence", height=150)
-        # The callback no longer requires a parameter; it accesses st.session_state directly.
+        st.text_area("Enter your question in English", key="user_sentence", height=150)
         st.button("submit", key="submit_button", on_click=submit_text_callback)
-    # Left Bottom: Centered "Cfu Explain" button
+        if st.session_state.error_message:
+            st.error(st.session_state.error_message)
     if st.session_state.submitted_text:
         with left_bottom:
-            # Add vertical spacing to approximately center the button vertically
             for _ in range(5):
                 st.write("")
             col1, col2, col3 = st.columns(3)
@@ -142,42 +143,33 @@ with left_col:
 
 # ---- Right Column ----
 with right_col:
-    # Top container: Fortune image area
     with st.container():
         col_left, col_center, col_right = st.columns([1, 2, 1])
         with col_center:
             if st.session_state.submitted_text and st.session_state.fortune_row:
                 header_link = st.session_state.fortune_row.get("HeaderLink")
                 if header_link:
-                    # Download the image from the URL
                     img_from_url = download_and_resize_image(header_link)
                     if img_from_url:
                         st.image(img_from_url, use_container_width=False)
                     else:
-                        # Fallback: display a default image if download fails
-                        img = load_and_resize_image("error.png")
+                        img = load_and_resize_image("/home/user/app/resources/error.png")
                         if img:
                             st.image(img, use_container_width=False)
                 else:
-                    # Fallback: display a default image if no header link is provided
-                    img = load_and_resize_image("error.png")
+                    img = load_and_resize_image("/home/user/app/resources/error.png")
                     if img:
                         st.image(img, use_container_width=False)
             else:
-                # Before submit, show the default image
-                img = load_and_resize_image("fortune.png")
+                img = load_and_resize_image("/home/user/app/resources/fortune.png")
                 if img:
                     st.image(img, caption="Your Fortune", use_container_width=False)
-    
-    # Bottom container: Display fortune details using text areas
     with st.container():
         if st.session_state.fortune_row:
-            header_text = st.session_state.fortune_row.get("Header", "N/A")
             luck_text = st.session_state.fortune_row.get("Luck", "N/A")
             description_text = st.session_state.fortune_row.get("Description", "No description available.")
             detail_text = st.session_state.fortune_row.get("Detail", "No detail available.")
             
-            # Create a summary with larger text using HTML styling
             summary = f"""
             <div style="font-size: 28px; font-weight: bold;">
                 Fortune stick number: {st.session_state.fortune_number}<br>
@@ -186,8 +178,5 @@ with right_col:
             """
             st.markdown(summary, unsafe_allow_html=True)
             
-            # Second text area: Description
             st.text_area("Description", value=description_text, height=150, disabled=True)
-            
-            # Third text area: Detail
             st.text_area("Detail", value=detail_text, height=150, disabled=True)