turn to DEEP work

app.py

@@ -3,7 +3,7 @@ import textract
 import tempfile
 from typing import List
 import spacy
-from spacy.tokens import DocBin, Doc
+from spacy.tokens import DocBin, Doc, Span
 from collections import Counter
 import srsly
 from spacy.matcher import PhraseMatcher
@@ -22,7 +22,7 @@ def download_model(select_model:str):
     except Exception as e:
         return False
 
-def search_docs(query:str, documents:List[Doc], nlp):
+def search_docs(query:str, documents:List[Doc], nlp) -> List[Span]:
     terms = query.split('|')
     patterns = [nlp.make_doc(text) for text in terms]
     matcher = PhraseMatcher(nlp.vocab)
@@ -34,7 +34,7 @@ def search_docs(query:str, documents:List[Doc], nlp):
         for match in matches:
             results.append(doc[match[1]:match[2]])
             
-    return results #List[Span]
+    return results
 
 models = srsly.read_json('models.json')
 models[''] = [] #require the user to choose a language
@@ -64,10 +64,14 @@ if language:
                     text = textract.process(temp.name)
                     text = text.decode('utf-8')
                     doc = nlp(text)
+                    doc.user_data['filename'] = uploaded_file.name
                     documents.append(doc)
                     ent_freq = Counter([ent.label_ for ent in doc.ents])
                     for key, value in ent_freq.items():
-                        if st.sidebar.button(key):
+                        if st.sidebar.button(key, key=key):
+                            #There are multiple identical st.button widgets with key='DATE'.
+                            #To fix this, please make sure that the key argument is unique for each st.button you create.
+                            # need to pool results from several files, then create buttons and search links
                             st.sidebar.write(value)
                             for ent in doc.ents:
                                 if ent.label_ == key:
@@ -85,11 +89,10 @@ if language:
                 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 style="border: 2px solid #202d89;border-radius: 15px;"><p>{result.doc.user_data['filename']}</p>
+                <div class='text'>{sent_before.text} <span class="text_mark"> {result.text}</span>{sent_after.text}</div>
                 </div>
                 """, unsafe_allow_html=True)
             
-            st.write(results)
             #st.download_button('Download', '', 'text/plain')
 

style.css

@@ -1,12 +1,12 @@
 .card {
   text-align: center;
   border-radius: 10px;
-  background: linear-gradient(
+  /* background: linear-gradient(
     90deg,
     rgba(77, 241, 241, 1) 0%,
     rgba(82, 124, 230, 1) 43%,
     rgba(219, 0, 132, 1) 100%
-  );
+  ); */
   margin-top: 15%;
   margin-bottom: 5%;
   padding-top: 15%;
@@ -70,7 +70,7 @@
 }
 
 .text_mark {
-  background-color: green;
+  background: radial-gradient(circle, rgba(63,94,251,1) 0%, rgba(252,70,107,1) 100%);
   color: white;
   padding: 0.5%;
   border-radius: 5px;