Update app.py

app.py

@@ -8,15 +8,15 @@ df = pd.read_csv(url, encoding='utf-8-sig')
 # Function to search years based on the selected mode
 def search_years(search_mode, query):
     if search_mode == "Search questions by YEAR":
-        matches = df[df['YEAR'].str.startswith(query[:4])]
+        matches = df[df['YEAR'].str.startswith(query.strip()[:4])]  # Ensure query is stripped of whitespace
     elif search_mode == "Search questions by Keywords":
-        keyword_list = [keyword.strip() for keyword in query.split(',')]
-        matches = df[df['KEYWORDS'].apply(lambda x: any(keyword in x for keyword in keyword_list))]  # Search in 'KEYWORDS'
-    elif search_mode == "Search questions by Words":  # New option for 'Search by Words'
-        word_list = [word.strip() for word in query.split(',')]
-        matches = df[df['TEXT'].apply(lambda x: any(word in x for word in word_list))]  # Search in 'TEXT'
+        keyword_list = [keyword.strip().lower() for keyword in query.split(',')]
+        matches = df[df['KEYWORDS'].str.lower().apply(lambda x: any(keyword in x for keyword in keyword_list))]
+    elif search_mode == "Search questions by Words":
+        word_list = [word.strip().lower() for word in query.split(',')]
+        matches = df[df['TEXT'].str.lower().apply(lambda x: any(word in x for word in word_list))]
     else:
-        return [], "Please select exactly one search mode."
+        return [], "Please select a valid search mode."
 
     if matches.empty:
         return [], "No results found for your query."
@@ -39,7 +39,7 @@ with gr.Blocks() as app:
     gr.Markdown("## ❄️ [1] Search Data")
     
     # Radio buttons to select search mode
-    search_mode = gr.Radio(choices=["Search by YEAR", "Search Years by Keywords", "Search Years by Words"], label="Search Mode")  # Added 'Search by Words'
+    search_mode = gr.Radio(choices=["Search questions by YEAR", "Search questions by Keywords", "Search questions by Words"], label="Search Mode")
     
     # Row for search query and button
     search_input = gr.Textbox(label="Search Query: e.g., 2024 (by YEAR) or tapping (by Keywords or Words)", placeholder="Enter year or keywords or words")