Update app.py

app.py

@@ -10,8 +10,6 @@ def get_SNP():
         library(dplyr)
         library(data.table)
 
-        setwd("D:/Mli/Project/GWASDB/")
-
         rm(list = ls())
 
         AD <- fread("Lambert/IGAP_summary_statistics/IGAP_stage_1.txt", header = TRUE, stringsAsFactors = F)
@@ -71,15 +69,16 @@ def get_SNP_gene(rsidlist):
 
     return rsid_gene
 
-def todf():
-    rsidlist = get_SNP()
-    rsid_gene = get_SNP_gene(rsidlist)
+def todf(disease1, disease2):
+    if disease1 != "" and disease2 != "":
+        rsidlist = get_SNP()
+        rsid_gene = get_SNP_gene(rsidlist)
 
-    for key, value in rsid_gene.items():
-        if value == []:
-            rsid_gene[key] = ["None"]
-    df = pd.DataFrame(list(rsid_gene.items()), columns=['rsid', 'genes'])
-    df['genes'] = df['genes'].apply(lambda x: x[0] if isinstance(x, list) and len(x) > 0 else x)
+        for key, value in rsid_gene.items():
+            if value == []:
+                rsid_gene[key] = ["None"]
+        df = pd.DataFrame(list(rsid_gene.items()), columns=['rsid', 'genes'])
+        df['genes'] = df['genes'].apply(lambda x: x[0] if isinstance(x, list) and len(x) > 0 else x)
 
     return df
 
@@ -92,40 +91,6 @@ def visible(identifiers):
                 visible=True), button_download.update(visible=True)
 
 
-# def download_all(identifiers, species, required_score, network_type):
-#     # 网络图像
-#     url1 = f"https://string-db.org/api/image/network?identifiers={identifiers}&species={species}"
-#     # 相互作用网络文件
-#     url2 = f"https://string-db.org/api/tsv/network?identifiers={identifiers}&species={species}&required_score={required_score}&network_type={network_type}"
-#     # 富集分析
-#     url3 = f"https://string-db.org/api/tsv/enrichment?identifiers={identifiers}&species={species}"
-
-#     # url 列表
-#     urls = [url1, url2, url3]
-#     # 定义文件名
-#     filenames = [
-#         "network.png", "network_interactions.tsv", "network_enrichment.tsv"
-#     ]
-
-#     # 创建一个空的zip文件对象
-#     z = zipfile.ZipFile("cbd_string_analysis.zip", "w")
-#     for url, filename in zip(urls, filenames):
-#         r = requests.get(url)
-#         # 创建一个类似于文件的对象，用于存储内容
-#         file_obj = io.BytesIO(r.content)
-#         # 将内容写入zip文件
-#         z.writestr(filename, file_obj.read())
-#     # 关闭zip文件对象
-#     z.close()
-#     return "cbd_string_analysis.zip"
-
-
-# def download_visible(identifiers, species, required_score, network_type):
-#     file = download_all(identifiers, species, required_score, network_type)
-#     if file != None:
-#         return file_download.update(visible=True)
-
-
 # 使用gr.Blocks()创建和组合组件
 with gr.Blocks() as demo:
     with gr.Row():
@@ -147,11 +112,6 @@ with gr.Blocks() as demo:
 
     # 按钮监听
     button_input.click(fn=todf, inputs=[disease1, disease2], outputs=rsid_gene_df)
-    # 显示列表
-    # button_input.click(
-    #     fn=visible,
-    #     inputs=protein_input,
-    #     outputs=[topo_para, network_stats, enrichment, button_download])
 
 # 在本地运行demo对象
 demo.launch(share=False)