liang-huggingface commited on
Commit
e73fe73
·
1 Parent(s): 4289cb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -57,12 +57,17 @@ def search_pubmed(query, retmax):
57
  handle.close()
58
  article_list = []
59
  for article in articles:
 
 
 
 
 
60
  article_dict = {
61
  'PMID': str(article['MedlineCitation']['PMID']),
62
  'Authors': ' '.join([author['LastName'] + ' ' + author.get('Initials', '')
63
- for author in article['MedlineCitation']['Article']['AuthorList']]),
64
  'Title': article['MedlineCitation']['Article']['ArticleTitle'],
65
- 'Abstract': article['MedlineCitation']['Article'].get('Abstract', {}).get('AbstractText', [None])[0]
66
  }
67
  article_list.append(article_dict)
68
  return pd.DataFrame(article_list)
 
57
  handle.close()
58
  article_list = []
59
  for article in articles:
60
+ abstract_text = article['MedlineCitation']['Article'].get('Abstract', {}).get('AbstractText', [])
61
+ if isinstance(abstract_text, list):
62
+ # Join the list elements if abstract is a list
63
+ abstract_text = " ".join(abstract_text)
64
+
65
  article_dict = {
66
  'PMID': str(article['MedlineCitation']['PMID']),
67
  'Authors': ' '.join([author['LastName'] + ' ' + author.get('Initials', '')
68
+ for author in article['MedlineCitation']['Article'].get('AuthorList'. [])]),
69
  'Title': article['MedlineCitation']['Article']['ArticleTitle'],
70
+ 'Abstract': abstract_text,
71
  }
72
  article_list.append(article_dict)
73
  return pd.DataFrame(article_list)