vv6

app.py

@@ -177,29 +177,101 @@ def run_crewai_process(user_query, model, temperature):
 
                 # Execute the temporary script using subprocess
                 try:
-                    # Add debug info to the script
+                    # Add debug info and ensure plot is saved with absolute path
                     debug_script = f"""
 import traceback
 import sys
+import os
+import matplotlib
+# Use non-interactive backend to avoid display issues
+matplotlib.use('Agg')
+import matplotlib.pyplot as plt
+
+# Debug info
+print("="*80)
+print("STARTING SCRIPT EXECUTION")
+print("="*80)
+print(f"[DEBUG] Python version: {{sys.version}}")
+print(f"[DEBUG] Working directory: {{os.getcwd()}}")
+print("[DEBUG] Directory contents:")
+for f in os.listdir('.'):
+    print(f"  - {{f}}{' (dir)' if os.path.isdir(f) else ''}")
+print("\n" + "="*80 + "\n")
 
 try:
-    # Original script
+    # Create a test plot first to verify matplotlib is working
+    test_fig, test_ax = plt.subplots()
+    test_ax.plot([1, 2, 3], [1, 4, 9])
+    test_ax.set_title('Test Plot - If you see this, matplotlib is working')
+    test_plot_path = 'test_plot.png'
+    test_fig.savefig(test_plot_path, bbox_inches='tight')
+    print(f"[DEBUG] Test plot saved to: {{os.path.abspath(test_plot_path)}}")
+    print(f"[DEBUG] Test plot size: {{os.path.getsize(test_plot_path)}} bytes")
+    
+    # Execute the original script
+    print("\n" + "="*80)
+    print("EXECUTING USER SCRIPT")
+    print("="*80)
     {generated_code}
     
-    # Ensure the plot is saved
-    import matplotlib.pyplot as plt
-    if plt.get_fignums():
-        plt.savefig('plot.png')
-        print("\n[DEBUG] Plot saved successfully to plot.png")
-    else:
-        print("\n[DEBUG] No figures were created in the script")
+    # Ensure any pending plots are drawn
+    plt.ioff()
+    
+    # Save any open figures
+    print("\n" + "="*80)
+    print("SAVING PLOTS")
+    print("="*80)
+    
+    # Get list of all figure numbers
+    fig_nums = plt.get_fignums()
+    print(f"[DEBUG] Found {{len(fig_nums)}} open figures")
+    
+    if fig_nums:
+        # Save the last figure as plot.png
+        last_fig = plt.figure(fig_nums[-1])
+        plot_path = os.path.abspath('plot.png')
+        last_fig.savefig(plot_path, bbox_inches='tight', dpi=100)
+        print(f"[DEBUG] Saved plot to: {{plot_path}}")
+        print(f"[DEBUG] Plot file size: {{os.path.getsize(plot_path)}} bytes")
         
+        # Save all figures with unique names
+        for i, num in enumerate(fig_nums, 1):
+            fig = plt.figure(num)
+            fig_path = os.path.abspath(f'plot_{{i}}.png')
+            fig.savefig(fig_path, bbox_inches='tight', dpi=100)
+            print(f"[DEBUG] Saved additional plot to: {{fig_path}} ({{os.path.getsize(fig_path)}} bytes)")
+    else:
+        print("[WARNING] No figures were created in the script")
+    
+    # Print final directory contents
+    print("\n" + "="*80)
+    print("FINAL DIRECTORY CONTENTS")
+    print("="*80)
+    for f in os.listdir('.'):
+        fpath = os.path.join('.', f)
+        if os.path.isfile(fpath):
+            print(f"  - {{f}} ({{os.path.getsize(fpath)}} bytes)")
+        else:
+            print(f"  - {{f}}/ (dir)")
+    
+    print("\n" + "="*80)
+    print("SCRIPT EXECUTION COMPLETE")
+    print("="*80)
+    
 except Exception as e:
-    print(f"\n[DEBUG] Error during script execution: {str(e)}")
-    print(f"[DEBUG] Error type: {type(e).__name__}")
-    print("\n[DEBUG] Traceback:")
+    print("\n" + "!"*80)
+    print("ERROR DURING EXECUTION")
+    print("!"*80)
+    print(f"Error type: {{type(e).__name__}}")
+    print(f"Error message: {{str(e)}}")
+    print("\nTraceback:")
     traceback.print_exc()
-    raise  # Re-raise the exception to be caught by the outer try-except
+    print("\n" + "!"*80 + "\n")
+    raise
+
+finally:
+    # Always close all figures to free memory
+    plt.close('all')
 """
                     # Write the debug script to a temporary file
                     with open(temp_script_path, "w") as f: