Update app.py

app.py

@@ -4,19 +4,15 @@ import pandas as pd
 import requests
 from io import BytesIO
 from PIL import Image
-
-import os
-print("Current Working Directory:", os.getcwd())
-print("Files in the directory:", os.listdir('.'))
-
+from transformers import pipeline
 
 # Define maximum dimensions for the fortune image (in pixels)
 MAX_SIZE = (400, 400)
 
 # Set page configuration
-st.set_page_config(page_title="Fortune Stick Enquiry", layout="wide")
+st.set_page_config(page_title="Fortuen Stick Enquiry", layout="wide")
 
-st.title("Fortune Stick Enquiry")
+st.title("Fortuen Stick Enquiry")
 
 # Initialize session state variables
 if "submitted_text" not in st.session_state:
@@ -25,9 +21,10 @@ 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 "fortune_data" not in st.session_state:
     try:
-        st.session_state.fortune_data = pd.read_csv(r"/home/user/app/detail.csv")
+        st.session_state.fortune_data = pd.read_csv("detail.csv")
     except Exception as e:
         st.error(f"Error loading CSV: {e}")
         st.session_state.fortune_data = None
@@ -35,9 +32,34 @@ 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):
+    pipe = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
+    # If the model predicts the label "en", we consider it English.
+    if pipe(question)[0]['label'] == 'en':
+        return True
+    return False
+
+def check_sentence_is_question_model(question):
+    pipe = pipeline("text-classification", model="shahrukhx01/question-vs-statement-classifier")
+    # If the model predicts "LABEL_1", we consider it a question.
+    if pipe(question)[0]['label'] == 'LABEL_1':
+        return True
+    return False
 
-# Callback function for the submit button:
+# 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", "")
+    
+    if not check_sentence_is_english_model(question):
+        st.error("Please enter in English")
+        return
+
+    if not check_sentence_is_question_model(question):
+        st.error("This is not a question. Please enter another question")
+        return
+
     st.session_state.submitted_text = True
     # Randomly generate a number from 1 to 100
     st.session_state.fortune_number = random.randint(1, 100)
@@ -99,13 +121,12 @@ with left_col:
     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.button("submit", key="submit_button", on_click=submit_text_callback)
-    # (The previous Stick Enquiry button has been removed)
-    
+    # Left Bottom: Centered "Cfu Explain" button
     if st.session_state.submitted_text:
-    # Left Bottom: Centered "解籤/Stick Enquiry" button
         with left_bottom:
             # Add vertical spacing to approximately center the button vertically
             for _ in range(5):
@@ -114,7 +135,6 @@ with left_col:
             with col2:
                 st.button("Cfu Explain", key="stick_button", on_click=stick_enquiry_callback)
             if st.session_state.stick_clicked:
-                # st.write("Here are the stick enquiry words...")  # Placeholder text
                 st.text_area(' ', value="Here are the stick enquiry words...", height=300, disabled=True)
 
 # ---- Right Column ----
@@ -132,17 +152,17 @@ with right_col:
                         st.image(img_from_url, use_container_width=False)
                     else:
                         # Fallback: display a default image if download fails
-                        img = load_and_resize_image(r"/home/user/app/fortune/error.png")
+                        img = load_and_resize_image("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(r"/home/user/app/fortune/error.png")
+                    img = load_and_resize_image("error.png")
                     if img:
                         st.image(img, use_container_width=False)
             else:
                 # Before submit, show the default image
-                img = load_and_resize_image(r"/home/user/app/fortune.png")
+                img = load_and_resize_image("fortune.png")
                 if img:
                     st.image(img, caption="Your Fortune", use_container_width=False)