added syntax validator

frontend/package.json

@@ -46,6 +46,7 @@
     "pino-caller": "^3.4.0",
     "pino-pretty": "^11.2.1",
     "pyodide": "^0.26.2",
+    "python-shell": "^5.0.0",
     "rfdc": "^1.3.1",
     "s3-sync-client": "^4.3.1",
     "source-map-support": "^0.5.21",

frontend/resources/compileComputation.mjs

@@ -1,8 +1,12 @@
 import  {parser} from '@lezer/python';
 import { TreeCursor }  from '@lezer/common';
+import { readFileSync } from 'fs';
+import {PythonShell} from 'python-shell'
 
 
-
+async function validateSource(source) {
+await  PythonShell.checkSyntax(source)      
+}
 
 export function compileComputationFunction(source) {
     const result = extractIO(source)
@@ -12,16 +16,20 @@ export function compileComputationFunction(source) {
     }
     return result
 }
-function extractIO(source) {
-    const tree = parser.parse(source)
+
+
+
+async function extractIO(source) {
+    await validateSource(source)
+    const tree = parser.parse(source);
     const cursor = tree.cursor()
+
     const functionInfo = {}
     while(cursor.next()) {
         if(cursor.node.name === 'FunctionDefinition') {
             cursor.firstChild()
             cursor.nextSibling()
             const functionName = source.slice(cursor.node.from, cursor.node.to)
-            console.log(cursor.node.name)
             if(functionName === 'compute') {
 
                 cursor.nextSibling();
@@ -45,23 +53,19 @@ function extractIO(source) {
 
                 const outputs = {}
                 while(cursor.nextSibling()){
-                    console.log(cursor.node.name)
                     if(cursor.node.name === 'Body'){
                         cursor.firstChild()
                         while(cursor.nextSibling()){
-                            console.log(cursor.node.name)
-                            console.log(cursor.node)
+
 
                             if(cursor.node.name === 'ReturnStatement'){
                                 cursor.firstChild()
                                 cursor.nextSibling()
                                 if(cursor.node.name === 'DictionaryExpression') {
                                     cursor.firstChild()
                                     cursor.nextSibling()
-                                    // console.log(cursor.node.name)
                                     while(cursor.node.name !== '}'){
                                         cursor.nextSibling()
-                                        // console.log(cursor.node.name)
                                         if(cursor.node.name === 'VariableName'){
                                             const outputName = source.slice(cursor.node.from, cursor.node.to)
                                             outputs[outputName] = {

frontend/resources/compileComputation.py

@@ -2,21 +2,18 @@
 import json
 import sys
 
-with open("/Users/kaganrua/Desktop/zetaforge-github/zetaforge/frontend/core/blocks/new-python/computations.py") as f:
-    source = f.read()
+# with open("/Users/kaganrua/Desktop/zetaforge-github/zetaforge/frontend/core/blocks/new-python/computations.py") as f:
+#     source = f.read()
 
 
 
 def extract_io(source):
     tree = ast.parse(source)
-    print(tree)
     function_info = {}
     for node in ast.walk(tree):
-        print("NODE", node)
         if isinstance(node, ast.FunctionDef) and node.name == 'compute':
             # Extract docstring
             docstring = ast.get_docstring(node)
-            print(docstring)
             function_info['description'] = docstring
 
             # Extract inputs
@@ -54,10 +51,9 @@ def extract_io(source):
             break
 
     return function_info
-io = extract_io(source)
-print(json.dumps(io))
 
-# if __name__ == '__main__':
-#     source = sys.stdin.read()
-#     io = extract_io(source)
-#     print(json.dumps(io))
+
+if __name__ == '__main__':
+    source = sys.stdin.read()
+    io = extract_io(source)
+    print(json.dumps(io))

frontend/server/blockSerialization.js

@@ -23,8 +23,15 @@ export async function compileComputation(blockPath) {
     );
   }
 
-
-  const io = compileComputationFunction(source)
+  let io;
+  try{
+    io = await compileComputationFunction(source)
+  } catch(err) {
+    throw new ServerError(
+      `Failed to compile your python script. Please check your syntax. ${err.message}`,
+      HttpStatus.INTERNAL_SERVER_ERROR
+    )
+  }
   return io;
 }