Spaces:
Sleeping
Sleeping
| 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_images(): | |
| if st.button("Back"): | |
| st.session_state.page = "upload_main" | |
| st.rerun() | |
| st.title("Your Uploaded Images") | |
| # Fetch all uploaded images from MongoDB | |
| images = list(collection.find({"type": "Image","status":"processed"})) | |
| if not images: | |
| st.write("You have not uploaded any images yet.") | |
| return | |
| # Display images in a grid (4 images per row) | |
| cols = st.columns(4) | |
| for idx, image in enumerate(images): | |
| col = cols[idx % 4] | |
| with col: | |
| # Container for each image and its expander | |
| st.markdown("<div class='image-wrapper'>", unsafe_allow_html=True) | |
| # Display the image using HTML | |
| st.markdown( | |
| f""" | |
| <div style='text-align: center;'> | |
| <img src='{image['object_url']}' alt='{image.get('name','Image')}' style='width:250px; height:250px; object-fit: cover; border-radius: 8px;' /> | |
| </div> | |
| """, | |
| unsafe_allow_html=True | |
| ) | |
| st.markdown("</div>", unsafe_allow_html=True) # Close image container | |
| # Expander for image details | |
| with st.expander("View Image Details"): | |
| st.write(f"**File Name:** {image.get('name', 'N/A')}") | |
| st.write(f"**Date Uploaded:** {format_date(image.get('upload_date', datetime.now().timestamp()))}") | |
| st.write(f"**Description:** {image.get('description', 'No description available')}") | |
| # Display tags if available | |
| tags = ", ".join(image.get("tags", [])) | |
| st.write(f"**Tags:** {tags if tags else 'No tags'}") | |
| # Display categories if available | |
| categories = ", ".join(image.get("categories", [])) | |
| st.write(f"**Categories:** {categories if categories else 'No categories'}") | |
| # Download link | |
| st.markdown( | |
| f"<a href='{image['object_url']}' class='download-link' download>Download Image</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 | |