Update app.py

app.py

@@ -33,7 +33,7 @@ def load_file(file):
             else:
                 return None
         except Exception as e:
-            print(e)
+            print("Error loading file from dict:", e)
             return None
         return df
     else:
@@ -48,16 +48,16 @@ def load_file(file):
                 df = pd.read_excel(file, engine='xlrd')
             else:
                 return None
-        except Exception:
+        except Exception as e:
+            print("Error loading file from file-like object:", e)
             return None
         return df
 
-
 def preview_file(file):
     """Return the DataFrame for preview."""
     df = load_file(file)
     if df is None:
-        # Return a DataFrame with an error message instead of a string
+        # Return a DataFrame with an error message instead of a plain string
         return pd.DataFrame({"Error": ["Error loading file or unsupported file type."]})
     return df
 
@@ -141,7 +141,6 @@ User Query: "{nl_query}"
     code = response.choices[0].message.content.strip()
     return code
 
-
 def safe_exec_code(code, df, capture_charts=True, interactive=False, extra_globals=None):
     """
     Execute the generated code in a restricted namespace.
@@ -198,8 +197,10 @@ def safe_exec_code(code, df, capture_charts=True, interactive=False, extra_globa
             if output is None:
                 output = safe_locals.get("basic_info", None)
         except Exception as ex:
-            # Capture the full traceback and exception details
             error_details = traceback.format_exc()
+            # Append a hint for KeyError related issues.
+            if "KeyError" in error_details:
+                error_details += "\nHint: The generated code might be referencing columns that do not exist in your DataFrame."
             return f"An error occurred during code execution:\n{error_details}", safe_globals["charts"]
     
     if capture_charts and not safe_globals["charts"]:
@@ -221,7 +222,6 @@ def safe_exec_code(code, df, capture_charts=True, interactive=False, extra_globa
         output = "No output variable ('result' or 'basic_info') was set by the code."
     return output, safe_globals["charts"]
 
-
 def generate_interpretation(analysis_result, nl_query):
     """
     Use OpenAI to generate a detailed interpretation of the analysis result.