liang-huggingface commited on
Commit
3c318e6
·
1 Parent(s): e8629da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -74,14 +74,16 @@ def search_pubmed(query, retmax=5, mindate=None, maxdate=None, datetype="pdat"):
74
  abstracts = []
75
  for article in articles:
76
  article_id = article['MedlineCitation']['PMID']
 
 
 
77
  article_title = article['MedlineCitation']['Article']['ArticleTitle']
78
- print(len(article['MedlineCitation']['Article'].get('Abstract', {}).get('AbstractText', [None])))
79
  abstract_text = article['MedlineCitation']['Article'].get('Abstract', {}).get('AbstractText', [None])
80
 
81
  if isinstance(abstract_text, list):
82
  # Join the list elements if abstract is a list
83
  abstract_text = " ".join(abstract_text)
84
- abstracts.append((article_id, article_title, abstract_text))
85
 
86
  return pd.DataFrame(abstracts)
87
 
@@ -105,7 +107,7 @@ def search_pubmed_old(query, retmax):
105
  article_dict = {
106
  'PMID': str(article['MedlineCitation']['PMID']),
107
  'Authors': ' '.join([author['LastName'] + ' ' + author.get('Initials', '')
108
- for author in article['MedlineCitation']['Article'].get('AuthorList'. [])]),
109
  'Title': article['MedlineCitation']['Article']['ArticleTitle'],
110
  'Abstract': abstract_text,
111
  }
 
74
  abstracts = []
75
  for article in articles:
76
  article_id = article['MedlineCitation']['PMID']
77
+ authors = ' '.join([author['LastName'] + ' ' + author.get('Initials', '')
78
+ for author in article['MedlineCitation']['Article'].get('AuthorList', [])]),
79
+
80
  article_title = article['MedlineCitation']['Article']['ArticleTitle']
 
81
  abstract_text = article['MedlineCitation']['Article'].get('Abstract', {}).get('AbstractText', [None])
82
 
83
  if isinstance(abstract_text, list):
84
  # Join the list elements if abstract is a list
85
  abstract_text = " ".join(abstract_text)
86
+ abstracts.append((article_id, authors, article_title, abstract_text))
87
 
88
  return pd.DataFrame(abstracts)
89
 
 
107
  article_dict = {
108
  'PMID': str(article['MedlineCitation']['PMID']),
109
  'Authors': ' '.join([author['LastName'] + ' ' + author.get('Initials', '')
110
+ for author in article['MedlineCitation']['Article'].get('AuthorList', [])]),
111
  'Title': article['MedlineCitation']['Article']['ArticleTitle'],
112
  'Abstract': abstract_text,
113
  }