update

app.py

@@ -6,10 +6,15 @@ import pandas as pd
 import os
 from rag_sec.document_search_system import DocumentSearchSystem
 from chainguard.blockchain_logger import BlockchainLogger
+from PIL import Image
 
 # Blockchain Logger
 blockchain_logger = BlockchainLogger()
 
+# Directory for storing uploaded files
+UPLOAD_DIR = "uploaded_files"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+
 # Initialize DocumentSearchSystem
 @st.cache_resource
 def initialize_system():
@@ -25,9 +30,45 @@ def initialize_system():
 # Initialize the system
 system = initialize_system()
 
-# Streamlit Layout
-st.title("Memora: Advanced Search and News Insights")
+st.title("Memora: Secure File Upload and Search with Blockchain & Neo4j")
 st.subheader("Personalized news and global updates at your fingertips")
+# File Upload Section
+uploaded_files = st.file_uploader("Upload your files", accept_multiple_files=True, type=['jpg', 'jpeg', 'png', 'mp4', 'avi'])
+
+if uploaded_files:
+    for uploaded_file in uploaded_files:
+        # Save file locally
+        file_path = os.path.join(UPLOAD_DIR, uploaded_file.name)
+        with open(file_path, "wb") as f:
+            f.write(uploaded_file.getbuffer())
+        st.success(f"File saved locally: {file_path}")
+
+        # Display uploaded file details
+        if uploaded_file.type.startswith('image'):
+            image = Image.open(uploaded_file)
+            st.image(image, caption=uploaded_file.name, use_column_width=True)
+
+        # Metadata Input
+        album = st.text_input(f"Album for {uploaded_file.name}", "Default Album")
+        tags = st.text_input(f"Tags for {uploaded_file.name} (comma-separated)", "")
+
+        # Log Metadata and Transaction
+        if st.button(f"Log Metadata for {uploaded_file.name}"):
+            metadata = {"file_name": uploaded_file.name, "tags": tags.split(','), "album": album}
+            blockchain_details = blockchain_logger.log_data(metadata)
+            blockchain_hash = blockchain_details.get("block_hash", "N/A")
+
+            # Use Neo4jHandler from DocumentSearchSystem to log the transaction
+            system.neo4j_handler.log_relationships(uploaded_file.name, tags, blockchain_hash, [album])
+            st.write(f"Metadata logged successfully! Blockchain Details: {blockchain_details}")
+
+# Blockchain Integrity Validation
+if st.button("Validate Blockchain Integrity"):
+    is_valid = blockchain_logger.is_blockchain_valid()
+    st.write("Blockchain Integrity:", "Valid ✅" if is_valid else "Invalid ❌")
+
+# Document Search Section
+st.subheader("Search Documents")
 
 # Google Search: User-Specific News
 st.subheader("1. Latest News About You")