Update app.py

app.py

@@ -42,12 +42,12 @@ def pubmed_search(mesh_terms, email, start_date, end_date):
     if not mesh_terms or not email:
         return "Please provide MeSH terms and email.", None, None
     
-    # Ensure time defaults to midnight if not provided
+    # Convert float timestamps to datetime objects
     try:
-        if isinstance(start_date, str):
-            start_date = datetime.strptime(start_date, "%Y-%m-%d %H:%M:%S")
-        if isinstance(end_date, str):
-            end_date = datetime.strptime(end_date, "%Y-%m-%d %H:%M:%S")
+        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")
@@ -113,8 +113,8 @@ with gr.Blocks() as app:
         )
     
     with gr.Row():
-        start_date = gr.DateTime(label="Start Date", value="2025-01-01 00:00:00")  # Ensure midnight time
-        end_date = gr.DateTime(label="End Date", value="2025-01-07 00:00:00")      # Ensure midnight time
+        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",