Update app.py

app.py

@@ -1,35 +1,40 @@
 import streamlit as st
 import requests
-from urllib.parse import quote
+from bs4 import BeautifulSoup
 
-def search_papers(user_input, max_results=3):
-    url = "https://api.openaire.eu/search/publications?title=" + quote(user_input)
-    try:
-        with requests.Session() as session:
-            response = session.get(url)
-            if response.status_code == 200:
-                data = response.json()
-                if 'response' in data.keys() and 'results' in data['response']:
-                    search_results = data["response"]["results"][:max_results]
-                    return search_results
-    except requests.exceptions.RequestException as e:
-        st.error(f"An error occurred: {str(e)}")
-    return None
+def search_arxiv(search_query: str, start: int = 0):
+    base_url = 'http://export.arxiv.org/api/query?'
+    query = f'search_query={search_query}&start={start}&max_results=10'
+    response = requests.get(base_url + query)
+    feed = BeautifulSoup(response.content, 'html.parser')
+    entries = feed.find_all('entry')
+    articles = []
+    for entry in entries:
+        article = {}
+        article['title'] = entry.title.text
+        article['authors'] = [author.find('name').text for author in entry.find_all('author')]
+        article['abstract'] = entry.summary.text
+        articles.append(article)
+    return articles
 
-st.title('The Substantiator')
+def get_paper_info(paper_id: str):
+    base_url = 'https://api.semanticscholar.org/v1/paper/'
+    response = requests.get(base_url + paper_id)
+    paper = response.json()
+    paper_info = {}
+    paper_info['title'] = paper['title']
+    paper_info['authors'] = [author['name'] for author in paper['authors']]
+    paper_info['abstract'] = paper['abstract']
+    paper_info['fieldsOfStudy'] = paper['fieldsOfStudy']
+    return paper_info
 
-user_input = st.text_input('Input your claim')
+st.title('Scientific Data Substantiator')
+search_query = st.text_input("Enter your search term")
 
-if st.button('Substantiate'):
-    search_results = search_papers(user_input)
-    if search_results is not None and len(search_results) > 0:
-        for result in search_results:
-            if 'title' in result.keys() and 'link' in result.keys() and 'url' in result['link']:
-                st.markdown(f"[{result['title']}]({result['link']['url']})", unsafe_allow_html=True)
-                if 'creator' in result.keys():
-                    st.markdown(f"**Author(s):** {result['creator']}")
-                if 'dateofacceptance' in result.keys():
-                    st.markdown(f"**Date of Acceptance:** {result['dateofacceptance']}")
-                st.markdown("---")
-    else:
-        st.write("No results found.")
+if search_query:
+    articles = search_arxiv(search_query)
+    for article in articles:
+        st.write("Title: ", article['title'])
+        st.write("Authors: ", ", ".join(article['authors']))
+        st.write("Abstract: ", article['abstract'])
+        st.write("-----")