progress on search

app.py

@@ -22,9 +22,10 @@ def download_model(select_model:str):
     except Exception as e:
         return False
 
-def search_docs(query:str, matcher:PhraseMatcher, documents:List[Doc], nlp):
+def search_docs(query:str, documents:List[Doc], nlp):
     terms = query.split('|')
     patterns = [nlp.make_doc(text) for text in terms]
+    matcher = PhraseMatcher(nlp.vocab)
     matcher.add(query, patterns)
     
     results = []
@@ -49,10 +50,10 @@ if language:
             nlp = spacy.load(select_model)
 
             nlp.max_length = 1200000
-            matcher = PhraseMatcher(nlp.vocab)
+            
 
             uploaded_files = st.file_uploader("Select files to process", accept_multiple_files=True)
-            query = st.sidebar.text_input(label="Enter your query", value="...")
+            query = st.sidebar.text_input(label="Enter your query (use | to separate search terms)", value="...")
             documents = []
             for uploaded_file in uploaded_files:
                 file_type = uploaded_file.type
@@ -70,12 +71,25 @@ if language:
                             st.sidebar.write(value)
                             for ent in doc.ents:
                                 if ent.label_ == key:
+                                    #TODO Make clickable button that updates query
+                                    #TODO also return distinct ents
                                     st.sidebar.write(ent.text)
                         
                 except Exception as e:
                     st.error(e)
-                
-            results = search_docs(query, matcher, documents,nlp)
+            
+            results_container = st.container()   
+            results = search_docs(query, documents,nlp)
+            for result in results:
+                doc = result.doc
+                sent_before = doc[result.sent.start:result.start]
+                sent_after = doc[result.end:result.sent.end]
+                results_container.markdown(f"""
+                <div class="card">
+                <div class='text'>{sent_before.text} <span class="result"> {result.text}</span>{sent_after.text}</div>
+                </div>
+                """, unsafe_allow_html=True)
+            
             st.write(results)
             #st.download_button('Download', '', 'text/plain')