temp fix

app.py

@@ -133,7 +133,8 @@ def format_result(result, highlight_terms, exact_search, datasets_filter=None):
     docid_html = get_docid_html(docid)
     language = "FIXME"
     result_html = """{}
-        <span style='font-size:14px; font-family: Arial; color:#7978FF; text-align: left;'>Document ID: {}</span><br>
+        <span style='font-size:14px; font-family: Arial; color:#7978FF; text-align: left;'>Document ID: {}</span>
+        <button type="button" onclick="alert('Hello world!')">Flag result</button><br>
         <!--  <span style='font-size:12px; font-family: Arial; color:MediumAquaMarine'>Language: {}</span><br> -->
         <span style='font-family: Arial;'>{}</span><br>
         <br>
@@ -169,6 +170,10 @@ def extract_results_from_payload(query, language, payload, exact_search):
         num_results = payload["num_results"]
     else:
         highlight_terms = payload["highlight_terms"]
+        results = []
+        for lang, res_for_lang in payload["results"].items():
+            for result in res_for_lang:
+                results.append(result)
 
     for result in results:
         text = result["text"]
@@ -287,20 +292,13 @@ if __name__ == "__main__":
                 label="Datasets Filter",
                 multiselect=True,
             )
-        with gr.Row() as results_row:
-            header_html = gr.HTML(label="Header")
-            results_html = gr.HTML(label="Results")
+        with gr.Row():
+            header_html = gr.HTML(label="Header", value="hello")
+        results_html = []
+        for i in range(100):
+            results_html.append(gr.HTML(label="Results"))
         with gr.Row(visible=False) as pagination:
             next_page_btn = gr.Button("Next Page")
-        with gr.Column(visible=False) as flagging_form:
-            flag_txt = gr.Textbox(
-                lines=1,
-                placeholder="Type here...",
-                label="""If you choose to flag your search, we will save the query, language and the number of results
-                    you requested. Please consider adding relevant additional context below:""",
-            )
-            flag_btn = gr.Button("Flag Results")
-            flag_btn.click(flag, inputs=[query, lang, k, flag_txt], outputs=[flag_txt])
 
         def run_query(query, lang, k, dropdown_input, max_page_size, received_results):
             query = query.strip()
@@ -346,7 +344,7 @@ if __name__ == "__main__":
                     num_results
                 )
             # print("processed_results", processed_results)
-            results_html_new = format_result_page(
+            results_html = format_result_page(
                 processed_results, highlight_terms, num_results, exact_search
             )
             return (
@@ -354,7 +352,7 @@ if __name__ == "__main__":
                 highlight_terms,
                 num_results,
                 exact_search,
-                results_html_new,
+                results_html,
                 ds,
             )
 
@@ -364,24 +362,24 @@ if __name__ == "__main__":
                 highlight_terms,
                 num_results,
                 exact_search,
-                results_html_new,
+                results_html,
                 datasets,
             ) = run_query(query, lang, k, dropdown_input, max_page_size, 0)
             has_more_results = exact_search and (num_results > max_page_size)
-            return {
-                processed_results_state: processed_results,
-                highlight_terms_state: highlight_terms,
-                num_results_state: num_results,
-                exact_search_state: exact_search,
-                results_html: results_html_new,
-                flagging_form: gr.update(visible=True),
-                datasets_filter: gr.update(visible=True),
-                available_datasets: gr.Dropdown.update(
-                    choices=datasets, value=datasets
-                ),
-                pagination: gr.update(visible=has_more_results),
-                received_results_state: len(processed_results),
-            }
+            return (
+                [
+                    processed_results,
+                    highlight_terms,
+                    num_results,
+                    exact_search,
+                    gr.update(visible=True),
+                    gr.Dropdown.update(choices=datasets, value=datasets),
+                    gr.update(visible=has_more_results),
+                    len(processed_results),
+                ]
+                + results_html
+                + [gr.update(visible=False)] * (100 - len(results_html))
+            )
 
         def next_page(
             query,
@@ -397,7 +395,7 @@ if __name__ == "__main__":
                 highlight_terms,
                 num_results,
                 exact_search,
-                results_html_new,
+                results_html,
                 datasets,
             ) = run_query(
                 query, lang, k, dropdown_input, max_page_size, received_results
@@ -407,22 +405,23 @@ if __name__ == "__main__":
             print("num_processed_results", num_processed_results)
             print("has_more_results", has_more_results)
             print("received_results", received_results)
-            return {
-                processed_results_state: processed_results,
-                highlight_terms_state: highlight_terms,
-                num_results_state: num_results,
-                exact_search_state: exact_search,
-                results_html: results_html_new,
-                flagging_form: gr.update(visible=True),
-                datasets_filter: gr.update(visible=True),
-                available_datasets: gr.Dropdown.update(
-                    choices=datasets, value=datasets
-                ),
-                pagination: gr.update(
-                    visible=num_processed_results >= max_page_size and has_more_results
-                ),
-                received_results_state: received_results + num_processed_results,
-            }
+            return (
+                [
+                    processed_results,
+                    highlight_terms,
+                    num_results,
+                    exact_search,
+                    gr.update(visible=True),
+                    gr.Dropdown.update(choices=datasets, value=datasets),
+                    gr.update(
+                        visible=num_processed_results >= max_page_size
+                        and has_more_results
+                    ),
+                    received_results + num_processed_results,
+                ]
+                + results_html
+                + [gr.update(visible=False)] * (100 - len(results_html))
+            )
 
         def filter_datasets(
             lang,
@@ -432,14 +431,14 @@ if __name__ == "__main__":
             exact_search,
             datasets_filter,
         ):
-            results_html_new = format_result_page(
+            results_html = format_result_page(
                 processed_results,
                 highlight_terms,
                 num_results,
                 exact_search,
                 datasets_filter,
             )
-            return {results_html: results_html_new}
+            return results_html + [gr.update(visible=False)] * (100 - len(results_html))
 
         query.submit(
             fn=submit,
@@ -449,13 +448,12 @@ if __name__ == "__main__":
                 highlight_terms_state,
                 num_results_state,
                 exact_search_state,
-                results_html,
-                flagging_form,
                 datasets_filter,
                 available_datasets,
                 pagination,
                 received_results_state,
-            ],
+            ]
+            + results_html,
         )
         submit_btn.click(
             submit,
@@ -465,13 +463,12 @@ if __name__ == "__main__":
                 highlight_terms_state,
                 num_results_state,
                 exact_search_state,
-                results_html,
-                flagging_form,
                 datasets_filter,
                 available_datasets,
                 pagination,
                 received_results_state,
-            ],
+            ]
+            + results_html,
         )
 
         next_page_btn.click(
@@ -490,13 +487,12 @@ if __name__ == "__main__":
                 highlight_terms_state,
                 num_results_state,
                 exact_search_state,
-                results_html,
-                flagging_form,
                 datasets_filter,
                 available_datasets,
                 pagination,
                 received_results_state,
-            ],
+            ]
+            + results_html,
         )
 
         available_datasets.change(
@@ -509,6 +505,18 @@ if __name__ == "__main__":
                 exact_search_state,
                 available_datasets,
             ],
-            outputs=[results_html],
+            outputs=results_html,
+        )
+        results_html[0].change(
+            filter_datasets,
+            inputs=[
+                lang,
+                processed_results_state,
+                highlight_terms_state,
+                num_results_state,
+                exact_search_state,
+                available_datasets,
+            ],
+            outputs=results_html,
         )
     demo.launch(enable_queue=True, debug=True)