Update code_df_custom.py

code_df_custom.py

@@ -6,29 +6,19 @@ def load_excel(file):
     return file, df
 
 def run_code(file, code):
-    scope = {'pd': pd}
-    if file:
-        print('file ok')
-        df = pd.read_excel(file)
-
-        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, ""
-    else:
+    if not file:
         print("No file provided")
         return pd.DataFrame(), "No file provided"
 
+    try:
+        df = pd.read_excel(file)
+        exec_globals = {'pd': pd, 'df': df}
+        exec(code, exec_globals)
+        new_df = exec_globals.get('df', df)
+        return new_df, ""
+    except Exception as e:
+        return df, f"# ERROR: {str(e)}"
+
 # def run_code_and_update_ui(file, code):
 #     df, error_msg = run_code(file, code)  # This is your updated run_code function.