Update app.py

app.py

@@ -2,9 +2,6 @@ import streamlit as st
 import hashlib
 import json
 import time
-import requests
-
-
 
 # Blockchain Class
 class Blockchain:
@@ -12,6 +9,20 @@ class Blockchain:
         self.chain = []
         self.voters = set()
         self.load_data()
+        if not self.chain:
+            self.create_genesis_block()
+
+    def create_genesis_block(self):
+        """Creates the first block in the blockchain."""
+        genesis_block = {
+            "index": 1,
+            "timestamp": time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),
+            "votes": {},
+            "previous_hash": "0",
+            "hash": self.hash_block({}, "0"),
+        }
+        self.chain.append(genesis_block)
+        self.save_data()
 
     def create_block(self, votes, previous_hash):
         block = {
@@ -21,7 +32,6 @@ class Blockchain:
             "previous_hash": previous_hash,
             "hash": self.hash_block(votes, previous_hash),
         }
-        block["ipfs_hash"] = self.upload_to_ipfs(block)
         self.chain.append(block)
         self.save_data()
         return block
@@ -33,16 +43,6 @@ class Blockchain:
     def get_latest_block(self):
         return self.chain[-1] if self.chain else None
 
-    def upload_to_ipfs(self, block):
-        """Uploads block data to IPFS and returns the IPFS hash."""
-        try:
-            response = requests.post(IPFS_API_URL, headers=IPFS_HEADERS, json={"pinataContent": block})
-            if response.status_code == 200:
-                return response.json().get("IpfsHash", "N/A")
-        except Exception as e:
-            return f"IPFS Error: {str(e)}"
-        return "Upload Failed"
-
     def save_data(self):
         with open("votes.json", "w") as f:
             json.dump({"chain": self.chain, "voters": list(self.voters)}, f, indent=4)
@@ -63,79 +63,16 @@ blockchain = Blockchain()
 # Set page config
 st.set_page_config(page_title="Blockchain Voting", page_icon="🗳", layout="wide")
 
-# Apply Dark Theme CSS Styling
-st.markdown(
-    """
-    <style>
-        /* Dark background */
-        .stApp {
-            background-color: #121212;
-            color: white;
-        }
-
-        /* Header Styling */
-        h1 { 
-            text-align: center; 
-            font-size: 45px;
-            background: linear-gradient(90deg, #ff8a00, #e52e71);
-            -webkit-background-clip: text;
-            color: transparent;
-            font-weight: bold;
-            padding: 10px;
-            border-radius: 10px;
-        }
-
-        /* Sidebar Styling */
-        .sidebar .sidebar-content {
-            background-color: #1e1e1e;
-            border-radius: 10px;
-            padding: 15px;
-            color: white;
-        }
-
-        /* Buttons with Neon Effect */
-        .stButton>button {
-            background-color: #ff6b6b;
-            color: white;
-            font-size: 18px;
-            padding: 10px;
-            width: 100%;
-            border-radius: 10px;
-            transition: 0.3s ease-in-out;
-            box-shadow: 0px 0px 15px rgba(255, 107, 107, 0.8);
-        }
-
-        .stButton>button:hover {
-            background-color: #ff4c4c;
-            box-shadow: 0px 0px 25px rgba(255, 76, 76, 1);
-        }
-
-        /* Vote Count Section */
-        .results {
-            background-color: rgba(30, 30, 30, 0.95);
-            padding: 20px;
-            border-radius: 10px;
-            box-shadow: 2px 2px 10px rgba(255, 255, 255, 0.2);
-        }
-        
-        /* Candidate Selection */
-        label[data-testid="stMarkdownContainer"] > div {
-            font-size: 18px;
-            font-weight: bold;
-            color: #ffcc00;
-        }
-
-    </style>
-    """,
-    unsafe_allow_html=True,
-)
-
 st.markdown("<h1>🗳 Secure Blockchain-Based Voting System</h1>", unsafe_allow_html=True)
 
 # Sidebar for Voter Input
 st.sidebar.header("🔐 Voter Authentication")
 voter_id = st.sidebar.text_input("🔑 Enter your Unique Voter ID:", max_chars=10)
 
+# Hash the voter ID for security
+def hash_voter_id(voter_id):
+    return hashlib.sha256(voter_id.encode()).hexdigest()
+
 # Candidate Selection
 st.sidebar.header("🗳 Vote Now")
 candidates = ["BJP", "Shiv Sena", "NCP", "NOTA"]
@@ -145,18 +82,20 @@ selected_candidate = st.sidebar.radio("Select your candidate:", candidates)
 if st.sidebar.button("✅ Submit Vote"):
     if not voter_id:
         st.sidebar.warning("⚠ Please enter a valid Voter ID!")
-    elif voter_id in blockchain.voters:
-        st.sidebar.error("❌ You have already voted!")
     else:
-        last_block = blockchain.get_latest_block()
-        new_votes = last_block["votes"].copy() if last_block else {}
+        voter_hash = hash_voter_id(voter_id)  # Hash voter ID
+        if voter_hash in blockchain.voters:
+            st.sidebar.error("❌ You have already voted!")
+        else:
+            last_block = blockchain.get_latest_block()
+            new_votes = last_block["votes"].copy() if last_block else {}
 
-        new_votes[selected_candidate] = new_votes.get(selected_candidate, 0) + 1
-        blockchain.create_block(new_votes, last_block["hash"] if last_block else "0")
-        blockchain.voters.add(voter_id)
-        blockchain.save_data()
+            new_votes[selected_candidate] = new_votes.get(selected_candidate, 0) + 1
+            blockchain.create_block(new_votes, last_block["hash"] if last_block else "0")
+            blockchain.voters.add(voter_hash)  # Store hashed ID
+            blockchain.save_data()
 
-        st.sidebar.success(f"✅ Vote cast for {selected_candidate}!")
+            st.sidebar.success(f"✅ Vote cast for {selected_candidate}!")
 
 # Display Vote Count
 st.markdown("<h2 style='text-align: center; color:#ffcc00;'>📊 Live Vote Count</h2>", unsafe_allow_html=True)
@@ -166,7 +105,7 @@ if latest_block:
     with st.container():
         col1, col2, col3, col4 = st.columns(4)
         vote_data = latest_block["votes"]
-        
+
         col1.metric(label="🔹 BJP", value=vote_data.get("BJP", 0))
         col2.metric(label="🔹 Shiv Sena", value=vote_data.get("Shiv Sena", 0))
         col3.metric(label="🔹 NCP", value=vote_data.get("NCP", 0))
@@ -174,4 +113,4 @@ if latest_block:
 
 # Show Blockchain Data
 if st.checkbox("📜 View Blockchain Data"):
-    st.json(blockchain.chain)
+    st.json(blockchain.chain)