Update main_app.py

main_app.py

@@ -415,11 +415,11 @@ def _(mo, template_variant):
 
 
 @app.cell
-def _(function_editor, mo, os):
+def _(function_editor, mo, os, ast):
+    function_name = None
     if function_editor.value:
         # Get the edited code from the function editor
         code = function_editor.value['editor']
-        function_name = None
         # Extract function name using AST without executing the code
 
         try:
@@ -448,6 +448,7 @@ def _(function_editor, mo, os):
             mo.md(f"Syntax error in function code: {str(e)}")
     return (
         code,
+        parsed_ast,
         deployable_function,
         f,
         file_path,