Revert "vv5"

This reverts commit 8e70ec15fc58a92150e416917bbf2e069293057b.

app.py

@@ -247,45 +247,13 @@ 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."
 
-                # 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
+                # Check for the generated plot file
                 if os.path.exists(plot_file_path):
-                    plot_found = True
-                    plot_debug_info.append(f"✅ Plot file found at: {plot_abs_path}")
-                    generated_plot_path = 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
                 else:
-                    # 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."
+                    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."
 
             except Exception as e:
                 traceback_str = traceback.format_exc()