Update app.py

app.py

@@ -11,7 +11,10 @@ def search_years(search_mode, query):
         matches = df[df['YEAR'].str.startswith(query[:4])]
     elif search_mode == "Search Years by Keywords":
         keyword_list = [keyword.strip() for keyword in query.split(',')]
-        matches = df[df['TEXT'].apply(lambda x: any(keyword in x for keyword in keyword_list))]
+        matches = df[df['KEYWORDS'].apply(lambda x: any(keyword in x for keyword in keyword_list))]  # Search in 'KEYWORDS'
+    elif search_mode == "Search Years 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'
     else:
         return [], "Please select exactly one search mode."
 
@@ -36,10 +39,10 @@ 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"], label="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'
     
     # Row for search query and button
-    search_input = gr.Textbox(label="Search Query: e.g., 2024 (by YEAR) or tapping (by Keywords)", placeholder="Enter year or keywords")
+    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")
     search_button = gr.Button("Click to Search")
     search_output = gr.Dropdown(label="Results (file names)", choices=[], visible=False)
     status_output = gr.Textbox(label="Status", visible=False)