Update view_pdf.py

view_pdf.py

@@ -1,70 +1,70 @@
-import streamlit as st
-from pymongo import MongoClient
-import os
-from dotenv import load_dotenv
-from datetime import datetime
-
-# Load environment variables
-load_dotenv()
-MONGO_URI = os.getenv("MONGO_URI")
-DB_NAME = os.getenv("DB_NAME")
-COLLECTION_NAME = os.getenv("COLLECTION_NAME")
-
-mongo_client = MongoClient(MONGO_URI)
-db = mongo_client[DB_NAME]
-collection = db[COLLECTION_NAME]
-
-
-def format_date(timestamp):
-    """Convert timestamp to a readable date format."""
-    return datetime.fromtimestamp(timestamp).strftime("%B %d, %Y")
-
-
-# Custom CSS to control image and expander container width and styling
-
-
-def view_pdfs():
-    if st.button("Back"):
-        st.session_state.page = "upload_main"
-        st.rerun()
-    st.title("Your Uploaded PDFs")
-
-    # Fetch all uploaded images from MongoDB
-    pdfs = list(collection.find({"type": "PDF"}))
-
-    if not pdfs:
-        st.write("You have not uploaded any PDFs yet.")
-        return
-
-    # Display images in a grid (4 images per row)
-    cols = st.columns(4)
-    for idx, pdf in enumerate(pdfs):
-        col = cols[idx % 4]
-
-        with col:
-
-            # Expander for image details
-            filename=pdf.get('name','N/A')
-            with st.expander(f"{filename}"):
-                st.write(f"**File Name:** {pdf.get('name', 'N/A')}")
-                st.write(f"**Date Uploaded:** {format_date(pdf.get('upload_date', datetime.now().timestamp()))}")
-
-                # Display tags if available
-                tags = ", ".join(pdf.get("tags", []))
-                st.write(f"**Tags:** {tags if tags else 'No tags'}")
-
-                # Display categories if available
-                categories = ", ".join(pdf.get("categories", []))
-                st.write(f"**Categories:** {categories if categories else 'No categories'}")
-
-                # Download link
-                st.markdown(
-                    f"<a href='{pdf['object_url']}' class='download-link' download>Download PDF</a>",
-                    unsafe_allow_html=True
-                )
-
-        # Move to a new row after every 4 images
-        if (idx + 1) % 4 == 0:
-            st.write("")  # Line break to move to the next row
-
-
+import streamlit as st
+from pymongo import MongoClient
+import os
+from dotenv import load_dotenv
+from datetime import datetime
+
+# Load environment variables
+load_dotenv()
+MONGO_URI = os.getenv("MONGO_URI")
+DB_NAME = os.getenv("DB_NAME")
+COLLECTION_NAME = os.getenv("COLLECTION_NAME")
+
+mongo_client = MongoClient(MONGO_URI)
+db = mongo_client[DB_NAME]
+collection = db[COLLECTION_NAME]
+
+
+def format_date(timestamp):
+    """Convert timestamp to a readable date format."""
+    return datetime.fromtimestamp(timestamp).strftime("%B %d, %Y")
+
+
+# Custom CSS to control image and expander container width and styling
+
+
+def view_pdfs():
+    if st.button("Back"):
+        st.session_state.page = "upload_main"
+        st.rerun()
+    st.title("Your Uploaded PDFs")
+
+    # Fetch all uploaded images from MongoDB
+    pdfs = list(collection.find({"type": "PDF","status":"processed"}))
+
+    if not pdfs:
+        st.write("You have not uploaded any PDFs yet.")
+        return
+
+    # Display images in a grid (4 images per row)
+    cols = st.columns(4)
+    for idx, pdf in enumerate(pdfs):
+        col = cols[idx % 4]
+
+        with col:
+
+            # Expander for image details
+            filename=pdf.get('name','N/A')
+            with st.expander(f"{filename}"):
+                st.write(f"**File Name:** {pdf.get('name', 'N/A')}")
+                st.write(f"**Date Uploaded:** {format_date(pdf.get('upload_date', datetime.now().timestamp()))}")
+
+                # Display tags if available
+                tags = ", ".join(pdf.get("tags", []))
+                st.write(f"**Tags:** {tags if tags else 'No tags'}")
+
+                # Display categories if available
+                categories = ", ".join(pdf.get("categories", []))
+                st.write(f"**Categories:** {categories if categories else 'No categories'}")
+
+                # Download link
+                st.markdown(
+                    f"<a href='{pdf['object_url']}' class='download-link' download>Download PDF</a>",
+                    unsafe_allow_html=True
+                )
+
+        # Move to a new row after every 4 images
+        if (idx + 1) % 4 == 0:
+            st.write("")  # Line break to move to the next row
+
+