Update app.py

app.py

@@ -1,3 +1,4 @@
+import datetime
 import json
 
 from langchain_community.retrievers import BM25Retriever
@@ -11,18 +12,18 @@ def load_docs_from_json(json_path):
 
     docs = []
     for paper in papers:
-        page_content = f"{paper['title']}\n\n{paper['abstract']}"
+        page_content = f"Title: {paper['title']}\n\nAbstract: {paper['abstract']}"
         doc = Document(
-        page_content=page_content,
-        metadata={
-            'title': paper['title'],
-            'link': paper['link'],
-            'authors': paper['authors'],
-            'submitter': paper['submitter'],
-            'date': paper['date'],
-            }
-        )
-    docs.append(doc)
+            page_content=page_content,
+            metadata={
+                'title': paper['title'],
+                'link': paper['link'],
+                'authors': paper['authors'],
+                'submitter': paper['submitter'],
+                'date': paper['date'],
+                }
+            )
+        docs.append(doc)
     
     return docs
 
@@ -33,9 +34,13 @@ docs = load_docs_from_json(json_path)
 retriever = BM25Retriever.from_documents(docs)
 retriever.k = 10
 
+dates = [datetime.datetime.strptime(doc.metadata['date'], '%Y-%m-%d') for doc in docs]
+oldest_date = min(dates)
+newest_date = max(dates)
+
 # streamlit
 st.title("HF Daily Papers Search")
-st.markdown(f"Search papers from [HF daily papers](https://huggingface.co/papers).\nNmber of documents: {len(docs)}")
+st.markdown(f"Search papers from [HF daily papers](https://huggingface.co/papers).\nNmber of documents: {len(docs)}\nfrom {oldest_date.strftime('%Y-%m-%d')} to {newest_date.strftime('%Y-%m-%d')}")
 
 user_query = st.text_input("Search anything...")
 if st.button('→'):