Update app.py

app.py

@@ -80,6 +80,21 @@ def extract_and_paragraph(pdf1, pdf2, paragraph):
     
     return paragraphs_1, paragraphs_2
 
+# Filter
+def filter_paragraphs(keyword):
+    global stored_paragraphs_1, stored_paragraphs_2
+    if not keyword:
+        return gr.update(choices=stored_paragraphs_1, value=None), gr.update(choices=stored_paragraphs_2, value=None)  # No keyword entered, return original list
+    
+    # Filter paragraphs that contain the keyword (case-insensitive)
+    filtered1 = [p for p in stored_paragraphs_1 if keyword.lower() in p.lower()]
+    filtered2 = [p for p in stored_paragraphs_2 if keyword.lower() in p.lower()]
+    # Update dropdown with filtered results
+    return gr.update(choices=filtered1, value=None), gr.update(choices=filtered2, value=None) 
+    
+def clear_paragraphs():
+    global stored_paragraphs_1, stored_paragraphs_2
+    return gr.update(choices=stored_paragraphs_1, value=None), gr.update(choices=stored_paragraphs_2, value=None)
 def process_paragraph_1_sum(paragraph):
                 try:
                     paragraph_index = int(paragraph.split(':')[0].replace('Paragraph ', '')) - 1
@@ -464,7 +479,14 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
                 b1 = gr.Button("Extract and Display Paragraphs")
                 paragraph_1_dropdown = gr.Dropdown(label="Select Paragraph from PDF 1")
                 paragraph_2_dropdown = gr.Dropdown(label="Select Paragraph from PDF 2")
-        
+                
+                keyword_input = gr.Textbox(label="Enter keyword to search")
+                # Button to trigger the filtering
+                with gr.Row():
+                    search_button = gr.Button("Search")
+                    clear_button = gr.Button("Clear")
+                search_button.click(filter_paragraphs, inputs=keyword_input, outputs=[pdf1, pdf2])
+                clear_button.click(clear_parargraphs, inputs=[], outputs=[pdf1, pdf2])
                 # Extract paragraphs from PDFs and update dropdowns
                 def update_paragraphs(pdf1, pdf2):
                     global stored_paragraphs_1, stored_paragraphs_2