update

app.py

@@ -1,9 +1,6 @@
 import streamlit as st
 import os
 from PIL import Image
-from googleapiclient.discovery import build
-from googleapiclient.http import MediaFileUpload
-from google.oauth2.credentials import Credentials
 from neo4j import GraphDatabase
 from chainguard.blockchain_logger import BlockchainLogger
 
@@ -18,35 +15,21 @@ class Neo4jHandler:
     def close(self):
         self.driver.close()
 
-    def log_transaction(self, file_name, drive_id, blockchain_hash):
+    def log_transaction(self, file_name, file_path, blockchain_hash):
         """Log transactions to Neo4j."""
         with self.driver.session() as session:
-            session.write_transaction(self._create_transaction_node, file_name, drive_id, blockchain_hash)
+            session.write_transaction(self._create_transaction_node, file_name, file_path, blockchain_hash)
 
     @staticmethod
-    def _create_transaction_node(tx, file_name, drive_id, blockchain_hash):
+    def _create_transaction_node(tx, file_name, file_path, blockchain_hash):
         tx.run(
             """
-            MERGE (t:Transaction {file_name: $file_name, drive_id: $drive_id, blockchain_hash: $blockchain_hash})
+            MERGE (t:Transaction {file_name: $file_name, file_path: $file_path, blockchain_hash: $blockchain_hash})
             RETURN t
             """,
-            file_name=file_name, drive_id=drive_id, blockchain_hash=blockchain_hash
+            file_name=file_name, file_path=file_path, blockchain_hash=blockchain_hash
         )
 
-# Google Drive Integration
-def get_drive_service():
-    creds = Credentials.from_authorized_user_file("credentials.json", scopes=["https://www.googleapis.com/auth/drive.file"])
-    return build('drive', 'v3', credentials=creds)
-
-drive_service = get_drive_service()
-
-def upload_to_google_drive(file_path, folder_id):
-    """Uploads a file to Google Drive."""
-    file_metadata = {'name': os.path.basename(file_path), 'parents': [folder_id]}
-    media = MediaFileUpload(file_path, resumable=True)
-    file = drive_service.files().create(body=file_metadata, media_body=media, fields='id').execute()
-    return file.get('id')
-
 # Metadata Logging
 def log_metadata(file_name, tags, album):
     """Log metadata to Chagu blockchain."""
@@ -58,6 +41,10 @@ def log_metadata(file_name, tags, album):
 st.title("Memora: Secure File Upload with Blockchain & Neo4j")
 st.subheader("Securely upload, organize, and query your files")
 
+# Directory for storing uploaded files
+UPLOAD_DIR = "uploaded_files"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+
 # File Upload
 uploaded_files = st.file_uploader("Upload your files", accept_multiple_files=True, type=['jpg', 'jpeg', 'png', 'mp4', 'avi'])
 
@@ -69,11 +56,9 @@ if uploaded_files:
         password="HwGDOxyGS1-79nLeTiX5bx5ohoFSpvHCmTv8IRgt-lY"
     )
 
-    folder_id = st.text_input("Enter Google Drive Folder ID", "")
     for uploaded_file in uploaded_files:
         # Save file locally
-        file_path = os.path.join("uploads", uploaded_file.name)
-        os.makedirs("uploads", exist_ok=True)
+        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}")
@@ -89,18 +74,10 @@ if uploaded_files:
 
         if st.button(f"Log Metadata for {uploaded_file.name}"):
             metadata = log_metadata(uploaded_file.name, tags.split(','), album)
+            blockchain_hash = metadata.get('block_hash', 'N/A')
+            neo4j_handler.log_transaction(uploaded_file.name, file_path, blockchain_hash)
             st.write(f"Metadata logged successfully! Blockchain Details: {metadata}")
 
-        # Upload to Google Drive
-        if folder_id and st.button(f"Upload {uploaded_file.name} to Google Drive"):
-            try:
-                drive_id = upload_to_google_drive(file_path, folder_id)
-                blockchain_hash = metadata.get('block_hash', 'N/A')
-                neo4j_handler.log_transaction(uploaded_file.name, drive_id, blockchain_hash)
-                st.success(f"File uploaded to Google Drive: File ID {drive_id}")
-            except Exception as e:
-                st.error(f"Failed to upload {uploaded_file.name} to Google Drive: {str(e)}")
-
     neo4j_handler.close()
 
 # Blockchain Integrity Validation