Update app.py

app.py

@@ -2,7 +2,8 @@ import streamlit as st
 import textract
 import tempfile
 import spacy
-from spacy.tokens import DocBin
+from spacy.tokens import DocBin, Doc
+from collections import Counter
 import srsly
 
 st.title('Index and Search a Collection of Documents')
@@ -13,6 +14,7 @@ def download_model(select_model:str):
         spacy.cli.download(select_model)
     return True
 
+
 doc_bin = DocBin()
 models = srsly.read_json('models.json')
 models[''] = [] #require the user to choose a language
@@ -33,14 +35,16 @@ if language:
 
             for uploaded_file in uploaded_files:
                 file_type = uploaded_file.type
-
-                temp = tempfile.NamedTemporaryFile()
+                file_suffix = '.' + uploaded_file.name.split('.')[-1]
+                temp = tempfile.NamedTemporaryFile(suffix=file_suffix,)
                 temp.write(uploaded_file.getvalue())
                 try:
                     text = textract.process(temp.name)
                     text = text.decode('utf-8')
                     doc = nlp(text)
-                    st.write(text)
+                    ent_freq = Counter([ent.label_ for ent in doc.ents])
+                    for key, value in ent_freq.items():
+                        st.write(key, value)
                 except Exception as e:
                     st.error(e)