Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from Bio import Entrez
-from datetime import datetime
+from datetime import datetime, timedelta
 import os
 
 # Function to fetch PubMed articles
@@ -38,21 +38,16 @@ def compile_summaries(records):
     return content
 
 # Main function for the Gradio interface
-def pubmed_search(mesh_terms, email, start_date, end_date):
+def pubmed_search(mesh_terms, email):
     if not mesh_terms or not email:
         return "Please provide MeSH terms and email.", None, None
     
-    # Convert float timestamps to datetime objects
-    try:
-        if isinstance(start_date, float):
-            start_date = datetime.fromtimestamp(start_date)
-        if isinstance(end_date, float):
-            end_date = datetime.fromtimestamp(end_date)
-        
-        start_date_str = start_date.strftime("%Y-%m-%d %H:%M:%S")
-        end_date_str = end_date.strftime("%Y-%m-%d %H:%M:%S")
-    except ValueError as e:
-        return f"Error processing dates: {str(e)}", None, None
+    # Set default dates
+    end_date = datetime.now()
+    start_date = end_date - timedelta(days=7)
+
+    start_date_str = start_date.strftime("%Y-%m-%d")
+    end_date_str = end_date.strftime("%Y-%m-%d")
 
     # Join MeSH terms for query
     query = " AND ".join([f'"{term}"[MeSH Terms]' for term in mesh_terms.split(",")])
@@ -112,10 +107,6 @@ with gr.Blocks() as app:
             lines=2
         )
     
-    with gr.Row():
-        start_date = gr.DateTime(label="Start Date", value="2025-01-01 00:00:00")
-        end_date = gr.DateTime(label="End Date", value="2025-01-07 00:00:00")
-        
     email_input = gr.Textbox(
         label="Email",
         placeholder="Your email (required by PubMed API)",
@@ -136,7 +127,7 @@ with gr.Blocks() as app:
     
     search_button.click(
         fn=pubmed_search,
-        inputs=[mesh_terms_box, email_input, start_date, end_date],
+        inputs=[mesh_terms_box, email_input],
         outputs=[status_output, markdown_file, text_file]
     )