fix: Address PR review comments on code generation pipeline

- Add GroundTruthNode and MetricsNode imports/exports to pytorch/__init__.py
- Add MetricsNode import/export to tensorflow/__init__.py
- Add missing `import torch` to config.py.jinja2 template
- Fix special node exclusion inconsistency in base_orchestrator.py (add loss, metrics, groundtruth to both _generate_code_specs and _generate_forward_pass filters)
- Fix special node exclusion inconsistency in validation.py (add metrics, groundtruth alongside loss)
- Remove torch, torchmetrics, tensorflow from backend requirements.txt (these belong only in generated project requirements)
- Remove output handle from loss nodes in BlockNode.tsx to match LossNode definition (terminal node with no outputs)
- Add GroundTruthNode export to TensorFlow index.ts for consistency with PyTorch

Co-authored-by: Aaditya Jindal <RETR0-OS@users.noreply.github.com>

Author: github-actions[bot]

project/block_manager/services/codegen/base_orchestrator.py

@@ -129,7 +129,7 @@ def _generate_code_specs(
 
         processable_nodes = [
             n for n in sorted_nodes
-            if get_node_type(n) not in ('input', 'dataloader', 'output')
+            if get_node_type(n) not in ('input', 'dataloader', 'output', 'loss', 'metrics', 'groundtruth')
         ]
 
         for node in processable_nodes:
@@ -193,7 +193,7 @@ def _generate_forward_pass(
 
         processable_nodes = [
             n for n in sorted_nodes
-            if get_node_type(n) not in ('output',)
+            if get_node_type(n) not in ('output', 'loss', 'metrics', 'groundtruth')
         ]
 
         for node in processable_nodes:

project/block_manager/services/nodes/pytorch/__init__.py

@@ -18,6 +18,8 @@
 from .concat import ConcatNode
 from .add import AddNode
 from .loss import LossNode
+from .groundtruth import GroundTruthNode
+from .metrics import MetricsNode
 
 __all__ = [
     'LinearNode',
@@ -38,5 +40,7 @@
     'ConcatNode',
     'AddNode',
     'LossNode',
+    'GroundTruthNode',
+    'MetricsNode',
 ]
 

project/block_manager/services/nodes/templates/pytorch/files/config.py.jinja2

@@ -4,6 +4,8 @@ Generated by VisionForge
 Architecture Complexity: {{ complexity }} ({{ layer_count }} layers)
 """
 
+import torch
+
 # Training Configuration
 BATCH_SIZE = {{ batch_size }}  # Adjusted for {{ complexity.lower() }} network
 LEARNING_RATE = {{ learning_rate }}  # {% if has_attention %}Reduced for attention layers{% else %}Standard for architecture{% endif %}

project/block_manager/services/nodes/tensorflow/__init__.py

@@ -18,6 +18,7 @@
 from .concat import ConcatNode
 from .add import AddNode
 from .loss import LossNode
+from .metrics import MetricsNode
 
 __all__ = [
     'LinearNode',
@@ -38,4 +39,5 @@
     'ConcatNode',
     'AddNode',
     'LossNode',
+    'MetricsNode',
 ]

project/block_manager/services/validation.py

@@ -318,7 +318,7 @@ def _validate_shape_compatibility(self):
             config = node.get('data', {}).get('config', {})
 
             # Skip nodes that don't have shape requirements
-            if node_type in ('input', 'output', 'dataloader', 'loss'):
+            if node_type in ('input', 'output', 'dataloader', 'loss', 'metrics', 'groundtruth'):
                 continue
 
             incoming = edge_map.get(node_id, [])

project/frontend/src/components/BlockNode.tsx

@@ -507,22 +507,6 @@ const BlockNode = memo(({ data, selected, id }: BlockNodeProps) => {
             })
           })()}
 
-          {/* Single output handle for loss value */}
-          <Handle
-            type="source"
-            position={Position.Right}
-            className="w-3 h-3 !bg-red-500 transition-all"
-            style={{
-              right: -6,
-              zIndex: 10
-            }}
-          />
-          {selected && (
-            <div
-              className="absolute right-0 top-1/2 -translate-y-1/2 translate-x-1/2 w-6 h-6 rounded-full border-2 border-red-500 bg-red-500/20 animate-pulse pointer-events-none"
-              style={{ right: -6 }}
-            />
-          )}
         </>
       ) : data.blockType === 'metrics' ? (
         <>

project/frontend/src/lib/nodes/definitions/tensorflow/index.ts

@@ -8,6 +8,7 @@
 
 export { InputNode } from '../pytorch/input'
 export { DataLoaderNode } from '../pytorch/dataloader'
+export { GroundTruthNode } from '../pytorch/groundtruth'
 export { OutputNode } from '../pytorch/output'
 export { LossNode } from '../pytorch/loss'
 export { MetricsNode } from './metrics'

project/requirements.txt

@@ -27,9 +27,6 @@ pillow>=11.0.0
 
 # Numerical Computing
 numpy>=2.2.0
-torch>=2.0.0
-torchmetrics>=1.0.0
-tensorflow>=2.14.0
 
 # Security & Rate Limiting
 django-ratelimit>=4.1.0