Update app.py

added publication-parsing

app.py

@@ -54,6 +54,16 @@ from tqdm import tqdm
 
 
 
+
+
+
+
+
+
+
+
+
+
 def query_records(search_term):
     def invert_abstract(inv_index):
         if inv_index is not None:
@@ -62,6 +72,16 @@ def query_records(search_term):
         else:
             return ' '
 
+    def get_pub(x):
+        try: 
+            source = x['source']['display_name']
+            if source not in ['parsed_publication','Deleted Journal']:
+                return source
+            else: 
+                return ' '
+        except:
+            return ' '
+
     # Fetch records based on the search term
     query = Works().search_filter(abstract=search_term)
 
@@ -71,6 +91,8 @@ def query_records(search_term):
 
     records_df = pd.DataFrame(records)
     records_df['abstract'] = [invert_abstract(t) for t in records_df['abstract_inverted_index']]
+    records_df['parsed_publication'] = [get_pub(x) for x in records_df['primary_location']]
+
     
     return records_df