Add 4 predefined colour for the user

Author: Susmita-Chakrabarty

ZKP_Demo_Tool/levels/base_level.py

@@ -116,6 +116,9 @@ def challenge_edge_once(self):
 
         if color1 == color2:
             result = f"❌ Verifier: Edge {edge} has same colors. Proof fails!"
+            self.update_narration(result)
+            QTimer.singleShot(1500, self.reject_proof)
+            return
         else:
             result = f"✅ Verifier: Edge {edge} looks good."
             self.rounds += 1
@@ -140,6 +143,10 @@ def finish_level(self):
         self.parent_selector.update_trust_points(points_earned=3)
         self.close()
         self.parent_selector.show()
+    def reject_proof(self):
+        QMessageBox.critical(self, "❌ Proof Rejected", "Verifier: The proof failed. I cannot be convinced.")
+        self.close()
+        self.parent_selector.show()
 
     def reset_game(self, preserve_round=False):
         self.committed = False

ZKP_Demo_Tool/ui/node_item.py

@@ -1,10 +1,13 @@
-from PyQt5.QtWidgets import QGraphicsEllipseItem, QGraphicsTextItem, QColorDialog, QToolTip
+from PyQt5.QtWidgets import (
+    QGraphicsEllipseItem, QGraphicsTextItem, QToolTip,
+    QDialog, QVBoxLayout, QPushButton, QHBoxLayout
+)
 from PyQt5.QtGui import QBrush, QColor, QFont
 from PyQt5.QtCore import Qt
 
 class NodeItem(QGraphicsEllipseItem):
     def __init__(self, node_id, pos, parent_level):
-        super().__init__(-20, -20, 40, 40)  # 40x40 circle
+        super().__init__(-20, -20, 40, 40)
         self.node_id = node_id
         self.parent_level = parent_level
         self.setBrush(QBrush(Qt.gray))
@@ -13,24 +16,49 @@ def __init__(self, node_id, pos, parent_level):
         self.setPos(pos)
         self.locked = False
 
-        # Node label
         self.text = QGraphicsTextItem(str(node_id))
         self.text.setDefaultTextColor(Qt.white)
         self.text.setFont(QFont("Arial", 10, QFont.Bold))
         self.text.setParentItem(self)
         self.text.setPos(-8, -10)
 
+        # Define the allowed color palette
+        self.allowed_colors = {
+            "Red": "#E74C3C",
+            "Green": "#2ECC71",
+            " Blue": "#3498DB",
+            "Yellow": "#F1C40F"
+        }
+
     def mousePressEvent(self, event):
         if not self.locked and not self.parent_level.committed:
-            color = QColorDialog.getColor()
-            if color.isValid():
-                self.setBrush(QBrush(color))
-                self.parent_level.node_colors[self.node_id] = color.name()
-                self.parent_level.update_narration(f"🎨 Prover colored node {self.node_id} as a secret.")
+            self.show_color_selection_dialog()
         super().mousePressEvent(event)
 
+    def show_color_selection_dialog(self):
+        dialog = QDialog()
+        dialog.setWindowTitle(f"Select a color for Node {self.node_id}")
+        layout = QVBoxLayout()
+
+        button_row = QHBoxLayout()
+        for label, hex_code in self.allowed_colors.items():
+            btn = QPushButton(label)
+            btn.setStyleSheet(f"background-color: {hex_code}; color: black; font-weight: bold;")
+            btn.clicked.connect(lambda _, c=hex_code: self.select_color(dialog, c))
+            button_row.addWidget(btn)
+
+        layout.addLayout(button_row)
+        dialog.setLayout(layout)
+        dialog.exec_()
+
+    def select_color(self, dialog, color_hex):
+        self.setBrush(QBrush(QColor(color_hex)))
+        self.parent_level.node_colors[self.node_id] = color_hex
+        self.parent_level.update_narration(f"🎨 Prover colored node {self.node_id} with color.")
+        dialog.accept()
+
     def hoverEnterEvent(self, event):
         if self.locked:
             QToolTip.showText(event.screenPos(), "🔒 This node is committed", self.parent_level)
         else:
-            QToolTip.showText(event.screenPos(), f" Node {self.node_id}: Click to color", self.parent_level)
+            QToolTip.showText(event.screenPos(), f" Node {self.node_id}: Click to choose a color", self.parent_level)