Update code_df_custom.py

code_df_custom.py

@@ -1,6 +1,11 @@
 import pandas as pd
 import traceback
 
+
+def load_excel(file):
+    df = pd.read_excel(file)
+    return file, df
+
 def run_code(file, code):
     scope = {'pd': pd}
     if file:
@@ -28,3 +33,9 @@ def run_code(file, code):
     else:
         print("No file provided")
         return pd.DataFrame(), "No file provided"
+
+
+def export_df(df, filename):
+    filename = filename.replace('.xlsx', '_coded.xlsx')
+    df.to_excel(filename, index=False)
+    return filename