vv5

app.py

@@ -247,13 +247,45 @@ except Exception as e:
                     execution_output = f"Error during script execution: {str(e)}\n\n"
                     execution_output += "Please check the generated code for issues or try a different query."
 
-                # Check for the generated plot file
+                # Enhanced plot file checking with more detailed debugging
+                plot_debug_info = []
+                plot_found = False
+                
+                # Check in current directory first
+                current_dir = os.getcwd()
+                plot_abs_path = os.path.abspath(plot_file_path)
+                
+                # Log directory contents for debugging
+                plot_debug_info.append(f"Current directory: {current_dir}")
+                plot_debug_info.append("Directory contents:" + "\n- " + "\n- ".join(os.listdir('.')))
+                
+                # Check if plot exists in current directory
                 if os.path.exists(plot_file_path):
-                    print(f"Plot file found at: {os.path.abspath(plot_file_path)}")  # Log file path
-                    generated_plot_path = plot_file_path  # Set the path to be returned
+                    plot_found = True
+                    plot_debug_info.append(f"✅ Plot file found at: {plot_abs_path}")
+                    generated_plot_path = plot_file_path
                 else:
-                    print(f"Plot file not found at expected path: {os.path.abspath(plot_file_path)}")  # Log missing file path
-                    execution_output += f"\nPlot file '{plot_file_path}' not found after execution."
+                    # Try to find the plot file in subdirectories
+                    for root, _, files in os.walk('.'):
+                        if plot_file_path in files:
+                            found_path = os.path.join(root, plot_file_path)
+                            plot_found = True
+                            plot_debug_info.append(f"✅ Plot file found at: {os.path.abspath(found_path)}")
+                            generated_plot_path = found_path
+                            break
+                    
+                    if not plot_found:
+                        plot_debug_info.append(f"❌ Plot file not found at: {plot_abs_path}")
+                        plot_debug_info.append("Troubleshooting tips:")
+                        plot_debug_info.append("1. Ensure the script calls plt.savefig('plot.png')")
+                        plot_debug_info.append("2. Check for any errors in the execution output")
+                        plot_debug_info.append("3. Verify the script has write permissions in the current directory")
+                
+                # Add debug info to execution output
+                execution_output += "\n\n[PLOT DEBUG] " + "\n[PLOT DEBUG] ".join(plot_debug_info)
+                
+                if not plot_found:
+                    execution_output += f"\n\n[ERROR] Plot file '{plot_file_path}' was not generated. Check the debug information above for details."
 
             except Exception as e:
                 traceback_str = traceback.format_exc()