Update app.py

app.py

@@ -89,6 +89,9 @@ def filter_paragraphs(keyword):
         return gr.update(choices=paragraph1, value=None), gr.update(choices=paragraph2, value=None)  # No keyword entered, return original list
     filtered1 = [f"Paragraph {i+1}: {p[:100]}..." for i, p in enumerate(stored_paragraphs_1) if keyword.lower() in p.lower()]
     filtered2 = [f"Paragraph {i+1}: {p[:100]}..." for i, p in enumerate(stored_paragraphs_2) if keyword.lower() in p.lower()]
+    global filter_paragraphs_1, filter_paragraphs_2
+    filter_paragraphs_1 = filtered1
+    filter_paragraphs_2 = filtered2
     # Filter paragraphs that contain the keyword (case-insensitive)
     # Update dropdown with filtered results
     return gr.update(choices=filtered1, value=None), gr.update(choices=filtered2, value=None) 
@@ -99,6 +102,16 @@ def clear_paragraphs():
     paragraph2 = [f"Paragraph {i+1}: {p[:100]}..." for i, p in enumerate(stored_paragraphs_2)]
     return gr.update(choices=paragraph1, value=None), gr.update(choices=paragraph2, value=None)
     
+def filtered_close_paragraph(p, keyword, pdf):
+    if not keyword:
+        if pdf == "1":
+            return lib.comparison.compare_selected_paragraph(p, stored_paragraphs_1)
+        else:
+            return lib.comparison.compare_selected_paragraph(p, stored_paragraphs_2)
+    if pdf == "1":
+        return lib.comparison.compare_selected_paragraph(p, filter_paragraphs_1)
+    else:
+        return lib.comparison.compare_selected_paragraph(p, filter_paragraphs_2)      
 def process_paragraph_1_sum(paragraph):
                 try:
                     paragraph_index = int(paragraph.split(':')[0].replace('Paragraph ', '')) - 1
@@ -412,7 +425,8 @@ def generate_text(df, country, theme):
 # Global variable
 stored_paragraphs_1 = []
 stored_paragraphs_2 = []
-
+filter_paragraphs_1 = []
+filter_paragraphs_2 = []
 stored_df1 = []
 stored_df2 = []
 
@@ -510,8 +524,8 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
                 paragraph_1_dropdown.select(fn=show1, inputs = paragraph_1_dropdown, outputs=selected_paragraph_1)
                 close_paragraph_1 = gr.Textbox(label="Closest Paragraph 1 Content", lines=4)
                 paragraph_1_dropdown.select(
-                    fn=lambda p: lib.comparison.compare_selected_paragraph(p, stored_paragraphs_2),  # Use stored_paragraphs_2 inside the function
-                    inputs=paragraph_1_dropdown,
+                    fn=lambda p, keyword: filtered_close_paragraph(p, keyword, "2"),  # Use stored_paragraphs_2 inside the function
+                    inputs=[paragraph_1_dropdown, keyword_input],
                     outputs=close_paragraph_1
                 )
                 with gr.Group():
@@ -548,8 +562,8 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
                 paragraph_2_dropdown.select(fn=show2, inputs = paragraph_2_dropdown, outputs=selected_paragraph_2)
                 close_paragraph_2 = gr.Textbox(label="Closest Paragraph 2 Content", lines=4)
                 paragraph_2_dropdown.select(
-                    fn=lambda p: lib.comparison.compare_selected_paragraph(p, stored_paragraphs_1),  # Use stored_paragraphs_2 inside the function
-                    inputs=paragraph_2_dropdown,
+                    fn=lambda p, keyword: filtered_close_paragraph(p, keyword, "1"),  # Use stored_paragraphs_2 inside the function
+                    inputs=[paragraph_2_dropdown, keyword_input],
                     outputs=close_paragraph_2
                 )
                 with gr.Group():