Update app.py

app.py

@@ -45,7 +45,7 @@ def generate_response(prompt):
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return response
 
-def search_pubmed(query, retmax=5, mindate=None, maxdate=None, datetype="pdat"):
+def search_pubmed_v2(query, retmax=5, mindate=None, maxdate=None, datetype="pdat"):
     Entrez.email = '[email protected]'  # Always set the Entrez.email to tell NCBI who you are
     search_kwargs = {
         "db": "pubmed",
@@ -88,9 +88,24 @@ def search_pubmed(query, retmax=5, mindate=None, maxdate=None, datetype="pdat"):
     return pd.DataFrame(abstracts)
     
 # Function to search PubMed for articles
-def search_pubmed_old(query, retmax):
+def search_pubmed(query, retmax=5, mindate=None, maxdate=None, datetype="pdat"):
     Entrez.email = '[email protected]'
-    handle = Entrez.esearch(db="pubmed", term=query, retmax=retmax, sort = 'relevance')
+
+    search_kwargs = {
+        "db": "pubmed",
+        "term": query,
+        "retmax": retmax,
+        "sort": 'relevance',
+        "datetype": datetype
+    }
+    
+    # If dates are provided, add them to the search arguments
+    if mindate:
+        search_kwargs["mindate"] = mindate
+    if maxdate:
+        search_kwargs["maxdate"] = maxdate
+    
+    handle = Entrez.esearch(search_kwargs)
     record = Entrez.read(handle)
     handle.close()
     idlist = record['IdList']