Update code_df_custom.py

code_df_custom.py

@@ -6,19 +6,30 @@ def load_excel(file):
     return file, df
 
 def run_code(file, code):
-    if not file:
-        print("No file provided")
-        return pd.DataFrame(), "No file provided"
-
-    try:
+    scope = {'pd': pd}
+    if file:
+        print('file ok')
         df = pd.read_excel(file)
-        exec_globals = {'pd': pd, 'df': df}
-        exec(code, exec_globals)
-        new_df = exec_globals.get('df', df)
+
+        scope['df'] = df
+        try:
+            exec(code, scope, scope)
+        except Exception as e:
+            scope['new_df'] = df
+            return scope['new_df'], f"# ERROR: {str(e)}"
+
+        # print(scope.keys())
+        if 'new_df' not in scope:
+            print("new_df not defined")
+            scope['new_df'] = df.copy()
+        new_df = scope['new_df']
+        
         return new_df, ""
-    except Exception as e:
-        return df, f"# ERROR: {str(e)}"
+    else:
+        print("No file provided")
+        return pd.DataFrame(), "No file provided"
 
+        
 # def run_code_and_update_ui(file, code):
 #     df, error_msg = run_code(file, code)  # This is your updated run_code function.