Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from datetime import date, timedelta
+from datetime import datetime, timedelta
 from langchain.document_loaders import ArxivLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.vectorstores import Chroma
@@ -11,7 +11,7 @@ from langchain.schema import Document
 
 
 LOAD_MAX_DOCS = 100
-
+FORMAT = '%Y%m%d%H%M%S'
 embeddings = HuggingFaceEmbeddings()    
 
 document_prompt = PromptTemplate(
@@ -46,11 +46,22 @@ def process_document(doc: Document):
     metadata["Body"] = doc.page_content
     return Document(page_content=doc.metadata["Summary"], metadata=metadata)
 
+def get_date_range(lookback_days: float):
+    max_date = datetime.today()
+    # Get the current date and time in UTC
+    now_utc = datetime.datetime.utcnow()
+    # Create a new datetime object for today at 18:00 UTC
+    today_1800_utc = datetime.datetime(now_utc.year, now_utc.month, now_utc.day, 18, 0, 0)
+    min_date = (today_1800_utc - timedelta(days=lookback_days+1))
+    # min_date = (max_date - timedelta(days=lookback_days))
+    return min_date, max_date
+
 def get_data(category: str, lookback_days: float, user_query: str):
     print("User query:", user_query)
-    max_date = date.today()
-    min_date = (max_date - timedelta(days=lookback_days))
-    query = f"cat:{category} AND submittedDate:[{min_date.strftime('%Y%m%d')} TO {max_date.strftime('%Y%m%d')}]"
+    
+    min_date, max_date = get_date_range(lookback_days)
+    print(min_date, max_date)
+    query = f"cat:{category} AND submittedDate:[{min_date.strftime(FORMAT)} TO {max_date.strftime(FORMAT)}]"
     loader = ArxivLoader(query=query, load_max_docs=LOAD_MAX_DOCS)
     docs = [process_document(doc) for doc in loader.load()]
     if len(docs) == 0: