Merge pull request #7 from bhowiebkr/feature/node_improvements

Adding custom GUI code embedded into the node graph

Author: bhowiebkr

code_editor_dialog.py

@@ -1,44 +1,72 @@
 # code_editor_dialog.py
-# A dedicated dialog window that now uses the advanced PythonCodeEditor widget.
+# A dedicated dialog window that now uses a QTabWidget to host
+# separate editors for execution code, GUI layout, and GUI logic.
 
-from PySide6.QtWidgets import QDialog, QVBoxLayout, QDialogButtonBox
+from PySide6.QtWidgets import QDialog, QVBoxLayout, QDialogButtonBox, QTabWidget
 from PySide6.QtGui import QFont
 from python_code_editor import PythonCodeEditor
 
+
 class CodeEditorDialog(QDialog):
     """
-    A dialog that hosts the advanced PythonCodeEditor widget.
+    A dialog that hosts a tabbed interface for editing all of a node's code.
     """
-    def __init__(self, code, parent=None):
+
+    def __init__(self, code, gui_code, gui_logic_code, parent=None):
         super().__init__(parent)
-        self.setWindowTitle("Node Code Editor")
-        self.setMinimumSize(700, 500)
+        self.setWindowTitle("Unified Code Editor")
+        self.setMinimumSize(750, 600)
 
         layout = QVBoxLayout(self)
+        tab_widget = QTabWidget()
+        layout.addWidget(tab_widget)
 
-        # --- Use the new advanced code editor ---
+        # --- Execution Code Editor ---
         self.code_editor = PythonCodeEditor()
         self.code_editor.setFont(QFont("Monospace", 11))
-        
-        # Set placeholder text if the initial code is empty
-        if not code:
-            code = ("from typing import Tuple\n\n"
-                    "def process_data(input_1: str) -> Tuple[str, int]:\n"
-                    "    # Your code here\n"
-                    "    # Use parameters for inputs and return for outputs.\n"
-                    "    # Type hints define the pins.\n"
-                    "    return 'processed', len(input_1)")
-        
-        self.code_editor.setPlainText(code)
-        
-        layout.addWidget(self.code_editor)
-
-        # --- OK and Cancel Buttons ---
+        exec_placeholder = "from typing import Tuple\n\n" "@node_entry\n" "def node_function(input_1: str) -> Tuple[str, int]:\n" "    return 'hello', len(input_1)"
+        # Only add placeholder if the code is None (new node)
+        self.code_editor.setPlainText(code if code is not None else exec_placeholder)
+        tab_widget.addTab(self.code_editor, "Execution Code")
+
+        # --- GUI Layout Code Editor ---
+        self.gui_editor = PythonCodeEditor()
+        self.gui_editor.setFont(QFont("Monospace", 11))
+        gui_placeholder = (
+            "# This script builds the node's custom GUI.\n"
+            "# Use 'parent', 'layout', 'widgets', and 'QtWidgets' variables.\n\n"
+            "label = QtWidgets.QLabel('Multiplier:', parent)\n"
+            "spinbox = QtWidgets.QSpinBox(parent)\n"
+            "spinbox.setValue(2)\n"
+            "layout.addWidget(label)\n"
+            "layout.addWidget(spinbox)\n"
+            "widgets['multiplier'] = spinbox\n"
+        )
+        self.gui_editor.setPlainText(gui_code if gui_code is not None else gui_placeholder)
+        tab_widget.addTab(self.gui_editor, "GUI Layout")
+
+        # --- GUI Logic Code Editor ---
+        self.gui_logic_editor = PythonCodeEditor()
+        self.gui_logic_editor.setFont(QFont("Monospace", 11))
+        gui_logic_placeholder = (
+            "# This script defines how the GUI interacts with the execution code.\n\n"
+            "def get_values(widgets):\n"
+            "    return {'multiplier': widgets['multiplier'].value()}\n\n"
+            "def set_values(widgets, outputs):\n"
+            "    # result = outputs.get('output_1', 'N/A')\n"
+            "    # widgets['result_label'].setText(f'Result: {result}')\n\n"
+            "def set_initial_state(widgets, state):\n"
+            "    if 'multiplier' in state:\n"
+            "        widgets['multiplier'].setValue(state['multiplier'])\n"
+        )
+        self.gui_logic_editor.setPlainText(gui_logic_code if gui_logic_code is not None else gui_logic_placeholder)
+        tab_widget.addTab(self.gui_logic_editor, "GUI Logic")
+
         button_box = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
         button_box.accepted.connect(self.accept)
         button_box.rejected.connect(self.reject)
         layout.addWidget(button_box)
 
-    def get_code(self):
-        """Returns the code currently in the editor."""
-        return self.code_editor.toPlainText()
+    def get_results(self):
+        """Returns the code from all three editors in a dictionary."""
+        return {"code": self.code_editor.toPlainText(), "gui_code": self.gui_editor.toPlainText(), "gui_logic_code": self.gui_logic_editor.toPlainText()}