Update app.py

app.py

@@ -1,17 +1,18 @@
 import logging
 import pandas as pd
 import gradio as gr
-import ipywidgets as widgets
 from pymed import PubMed
-from Bio import Entrez
+import urllib.parse
+import urllib.request
+import ipywidgets as widgets
 
-def search_pubmed(search_term, keywords, max_results):
+def search_pubmed(search_term, keywords, max_results, tool, email):
     # Validate the input
-    if max_results < 1:
+    if max_results is None or max_results < 1:
         raise ValueError("Max Results must be a positive integer")
     
     # Connect to PubMed database
-    pubmed = PubMed(tool="MyTool", email="aalamel@clemson.edu")
+    pubmed = PubMed(tool=tool, email=email)
     results = pubmed.query(search_term, max_results=max_results)
     
     # Prepare the lists to store article information
@@ -48,6 +49,11 @@ def search_pubmed(search_term, keywords, max_results):
 def download_csv(b):
     download_button.description = "Downloading..."
     download_button.disabled = True
+    input_dict = interface.process(raw=False)
+    search_term = input_dict["Search Term"]
+    keywords = input_dict["Keywords"]
+    max_results = input_dict["Max Results"]
+    dataframe = search_pubmed(search_term, keywords, max_results)
     dataframe.to_csv("pubmed_results.csv", index=False)
     download_button.description = "Download CSV"
     download_button.disabled = False
@@ -61,7 +67,6 @@ outputs = [gr.outputs.Dataframe(type="pandas")]
 interface = gr.Interface(search_pubmed, inputs, outputs, title="PubMed Search")
 
 result = interface.launch(share=True)
-dataframe = result[0]
 
 download_button = widgets.Button(description="Download CSV")
 download_button.on_click(download_csv)