Update code_df_custom.py

code_df_custom.py

@@ -34,6 +34,21 @@ def run_code(file, code):
         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.
+    
+    # Now, check if there's an error message and handle it appropriately.
+    if error_msg:
+        # You can update some error display component in Gradio here to show the error_msg.
+        # Assuming you have a gr.Textbox to display errors:
+        error_display.update(value=f"Error in code execution: {error_msg}")
+        # Ensure you still return the original DataFrame or an empty one to satisfy the expected output type.
+        return df
+    else:
+        # If no error, clear the error display and return the DataFrame as usual.
+        error_display.update(value="")
+        return df
+
 
 def export_df(df, filename):
     filename = filename.replace('.xlsx', '_coded.xlsx')