New features!

app.py

@@ -111,8 +111,10 @@ def search_index(search_text, sd, ks, sort_by, max_results, user_num, search_tit
             output_df2 = output_df2.sort_values(by = ['Date Created'], ascending = False)
         elif sort_by == 'A-Z':
             output_df2 = output_df2.sort_values(by = ['title'], ascending = True)
-            
-        output_df2 = output_df2.head(int(max_results))
+
+        total_returned = 'No. of Results to Return (Total: ' + str(output_df2.shape[0]) + ')'
+        if max_results != 'All':
+            output_df2 = output_df2.head(int(max_results))
         output_df2 = output_df2[['url']].reset_index(drop = True)
         
         max_cols = 5
@@ -134,7 +136,7 @@ def search_index(search_text, sd, ks, sort_by, max_results, user_num, search_tit
         
     return('<center>' + 
            final_df.to_html(escape = False, render_links = True, index = False, header = False) +
-           '</center>')
+           '</center>', gr.update(label = total_returned))
     
 
 def search_logging(x: str, request: gr.Request):
@@ -194,14 +196,14 @@ with gr.Blocks(css=style) as app:
                 shared_drive = gr.Dropdown(choices = ['Illustrations - 01-10 to 07-22', 'Illustrations - Now'], multiselect = True, label = 'Shared Drive', value = ['Illustrations - 01-10 to 07-22', 'Illustrations - Now'])
                 key_stage = gr.Dropdown(choices = ['EYFS', 'KS1', 'KS2'], multiselect = True, label = 'Key Stage', value = ['EYFS', 'KS1', 'KS2'])
                 sort_by = gr.Dropdown(choices = ['Relevance', 'Date Created', 'A-Z'], value = 'Relevance', multiselect = False, label = 'Sort By')
-                max_return = gr.Dropdown(choices = ['10', '25', '50', '75', '100', '250', '500'], value = '10', multiselect = False, label = 'No. of Results to Return')
+                max_return = gr.Dropdown(choices = ['10', '25', '50', '75', '100', '250', '500', '1000', '5000', '10000', 'All'], value = '50', multiselect = False, label = 'No. of Results to Return (Total: 0)')
             with gr.Row():
                 search_button = gr.Button(value="Search!")
             with gr.Row(): 
                 output_df = gr.HTML() 
             back_top_btn = gr.HTML(back_to_top_btn_html)
-    search_button.click(search_index, inputs=[search_prompt, shared_drive, key_stage, sort_by, max_return, user_num, title_search], outputs=output_df) 
-    search_prompt.submit(search_index, inputs=[search_prompt, shared_drive, key_stage, sort_by, max_return, user_num, title_search], outputs=output_df)
+    search_button.click(search_index, inputs=[search_prompt, shared_drive, key_stage, sort_by, max_return, user_num, title_search], outputs=[output_df, max_return]) 
+    search_prompt.submit(search_index, inputs=[search_prompt, shared_drive, key_stage, sort_by, max_return, user_num, title_search], outputs=[output_df, max_return])
     search_button.click(search_logging, inputs=[search_prompt], outputs=None) 
     search_prompt.submit(search_logging, inputs=[search_prompt], outputs=None)