Update utils.py

utils.py

@@ -6,7 +6,7 @@ from nltk.corpus import stopwords
 stop_words = stopwords.words('english')
 import arxiv
 
-def get_md_text_abstract(rag_answer, source = ['Arxiv Search', 'Semantic Search'][1], return_prompt_formatting = False):
+def get_md_text_abstract(rag_answer, source = ['Arxiv Search', 'Semantic Search','BeWeb'][1], return_prompt_formatting = False):
     if 'Semantic Search' in source:
       title = rag_answer['document_metadata']['title'].replace('\n','')
       #score = round(rag_answer['score'], 2)
@@ -24,7 +24,13 @@ def get_md_text_abstract(rag_answer, source = ['Arxiv Search', 'Semantic Search'
       authors = ', '.join([author.name for author in rag_answer.authors])
       paper_link = rag_answer.links[0].href
       download_link = rag_answer.links[1].href
-      
+    elif 'BeWeb' in source:
+      title = rag_answer.title
+      date = rag_answer.updated.strftime('%d %b %Y')
+      paper_abs = rag_answer.summary.replace('\n',' ') + '\n'
+      authors = ', '.join([author.name for author in rag_answer.authors])
+      paper_link = rag_answer.links[0].href
+      download_link = rag_answer.links[1].href      
     else:
       raise Exception