Create app.py

app.py

@@ -0,0 +1,164 @@
+import streamlit as st
+import requests
+import xml.etree.ElementTree as ET
+import pandas as pd
+from googletrans import Translator
+
+# 语言翻译字典
+translations = {
+    "en": {
+        "title": "PubMed Literature Search",
+        "search_placeholder": "Enter search query (e.g., cancer treatment)",
+        "search_button": "Search",
+        "language_option": "English",
+        "language_label": "Language",
+        "results_title": "Search Results",
+        "error_message": "An error occurred while fetching data from PubMed.",
+        "no_results": "No articles found for the given query.",
+        "col_index": "Index",
+        "col_title": "Title",
+        "col_abstract": "Abstract",
+        "col_authors": "Authors",
+        "col_link": "Link"
+    },
+    "zh": {
+        "title": "PubMed 文献检索",
+        "search_placeholder": "输入检索词（例如：cancer treatment）",
+        "search_button": "检索",
+        "language_option": "中文",
+        "language_label": "语言",
+        "results_title": "检索结果",
+        "error_message": "从 PubMed 获取数据时出错。",
+        "no_results": "未找到相关文献。",
+        "col_index": "序号",
+        "col_title": "文献标题",
+        "col_abstract": "文献摘要",
+        "col_authors": "文献作者",
+        "col_link": "文献链接"
+    }
+}
+
+def get_translation(lang, key):
+    return translations[lang][key]
+
+def search_pubmed(query, retmax=200):
+    """
+    使用PubMed API进行检索
+    """
+    base_url = "https://eutils.ncbi.nlm.nih.gov/entrez/eutils/"
+    search_url = f"{base_url}esearch.fcgi?db=pubmed&term={query}&retmax={retmax}"
+    try:
+        search_response = requests.get(search_url)
+        search_response.raise_for_status()
+        search_root = ET.fromstring(search_response.content)
+        id_list = [id_elem.text for id_elem in search_root.findall(".//Id")]
+        return id_list
+    except requests.exceptions.RequestException as e:
+        st.error(f"Error during eSearch: {e}")
+        return []
+
+def fetch_articles(id_list):
+    """
+    根据ID列表获取文献详情
+    """
+    base_url = "https://eutils.ncbi.nlm.nih.gov/entrez/eutils/"
+    ids = ",".join(id_list)
+    fetch_url = f"{base_url}efetch.fcgi?db=pubmed&id={ids}&rettype=xml"
+    try:
+        fetch_response = requests.get(fetch_url)
+        fetch_response.raise_for_status()
+        return fetch_response.content
+    except requests.exceptions.RequestException as e:
+        st.error(f"Error during eFetch: {e}")
+        return None
+
+def parse_articles(xml_data):
+    """
+    解析文献的XML数据
+    """
+    articles = []
+    if xml_data:
+        root = ET.fromstring(xml_data)
+        for article in root.findall(".//PubmedArticle"):
+            title_elem = article.find(".//ArticleTitle")
+            title = title_elem.text if title_elem is not None else "No Title"
+
+            abstract_elem = article.find(".//AbstractText")
+            abstract = abstract_elem.text if abstract_elem is not None else "No Abstract"
+
+            author_list = article.findall(".//Author")
+            authors = ", ".join([
+                f"{author.find('LastName').text if author.find('LastName') is not None else ''} {author.find('ForeName').text if author.find('ForeName') is not None else ''}".strip()
+                for author in author_list
+            ]) if author_list else "No Authors"
+
+            pmid_elem = article.find(".//PMID")
+            pmid = pmid_elem.text if pmid_elem is not None else ""
+            link = f"https://pubmed.ncbi.nlm.nih.gov/{pmid}/" if pmid else "No Link"
+
+            articles.append({
+                "Title": title,
+                "Abstract": abstract,
+                "Authors": authors,
+                "Link": link
+            })
+    return articles
+
+def main():
+    st.set_page_config(layout="wide")
+
+    # 语言选择
+    lang = st.sidebar.radio(
+        "Language / 语言",
+        ("en", "zh"),
+        format_func=lambda x: "English" if x == "en" else "中文"
+    )
+
+    st.title(get_translation(lang, "title"))
+
+    # 搜索框
+    search_query = st.text_input(get_translation(lang, "search_placeholder"))
+
+    if st.button(get_translation(lang, "search_button")):
+        if search_query:
+            with st.spinner('Searching...'):
+                id_list = search_pubmed(search_query)
+                if id_list:
+                    xml_data = fetch_articles(id_list)
+                    if xml_data:
+                        articles = parse_articles(xml_data)
+                        if articles:
+                            st.subheader(get_translation(lang, "results_title"))
+
+                            # 翻译标题和摘要
+                            if lang == 'zh':
+                                translator = Translator()
+                                for article in articles:
+                                    try:
+                                        article['Title'] = translator.translate(article['Title'], dest='zh-cn').text
+                                        article['Abstract'] = translator.translate(article['Abstract'], dest='zh-cn').text
+                                    except Exception as e:
+                                        st.warning(f"翻译时出错: {e}")
+
+
+                            df = pd.DataFrame(articles)
+                            df.index = range(1, len(df) + 1)
+                            df.rename(columns={
+                                "Title": get_translation(lang, "col_title"),
+                                "Abstract": get_translation(lang, "col_abstract"),
+                                "Authors": get_translation(lang, "col_authors"),
+                                "Link": get_translation(lang, "col_link")
+                            }, inplace=True)
+                            df.index.name = get_translation(lang, "col_index")
+
+                            st.dataframe(df)
+                        else:
+                            st.warning(get_translation(lang, "no_results"))
+                    else:
+                        st.error(get_translation(lang, "error_message"))
+                else:
+                    st.warning(get_translation(lang, "no_results"))
+
+if __name__ == "__main__":
+    main()
+