Update main_app.py

main_app.py

@@ -425,7 +425,7 @@ def _(function_editor, mo, os):
             try:
                 return __import__(name, *args, **kwargs)
             except ImportError:
-                # Return a dummy module
+                # Return a dummy module with singleton pattern
                 class DummyModule:
                     _instance = None
                     
@@ -440,8 +440,19 @@ def _(function_editor, mo, os):
                     def __call__(self, *args, **kwargs):
                         return self
                         
+                # Add to sys.modules to handle nested imports
+                import sys
+                if '.' in name:
+                    parts = name.split('.')
+                    for i in range(1, len(parts) + 1):
+                        submod_name = '.'.join(parts[:i])
+                        if submod_name not in sys.modules:
+                            sys.modules[submod_name] = DummyModule()
+                else:
+                    sys.modules[name] = DummyModule()
+                        
                 return DummyModule()
-                
+                    
         # Create a namespace to execute the code in
         namespace = {'__builtins__': __builtins__, '__import__': safe_import}
         # Execute the code
@@ -796,7 +807,7 @@ def _(
                 # If JSON parsing fails, try Python literal evaluation as fallback
                 function_meta[deployment_client.repository.FunctionMetaNames.SAMPLE_SCORING_INPUT] = ast.literal_eval(sample_input_editor.value)
 
-    def upload_function(function_meta, use_function_object=False):
+    def upload_function(function_meta, use_function_object=True):
         """
         Uploads a Python function to watsonx.ai as a deployable asset.
         Parameters:
@@ -863,7 +874,7 @@ def _(
 
     upload_button = mo.ui.button(
         label="Upload Function",
-        on_click=lambda _: upload_function(function_meta, use_function_object=False),
+        on_click=lambda _: upload_function(function_meta, use_function_object=True),
         kind="success",
         tooltip="Click to upload function to watsonx.ai"
     )