vv4

app.py

@@ -176,22 +176,76 @@ def run_crewai_process(user_query, model, temperature):
                     f.write(generated_code)
 
                 # Execute the temporary script using subprocess
-                # Use python3 to ensure correct interpreter in Colab
-                process = subprocess.run(
-                    ["python3", temp_script_path],
-                    capture_output=True,
-                    text=True,  # Capture stdout and stderr as text
-                    check=False  # Don't raise exception for non-zero exit codes
-                )
-                execution_output = process.stdout + process.stderr
-
-                # Check for specific errors in execution output
-                if "KeyError" in execution_output:
-                    execution_output += "\n\nPotential Issue: The generated script encountered a KeyError. This might mean the script tried to access a column or data point that wasn't available for the specified stock or timeframe. Please try a different query or timeframe."
-                elif "FileNotFoundError: [Errno 2] No such file or directory: 'plot.png'" in execution_output and "Figure(" in execution_output:
-                    execution_output += "\n\nPlot Generation Issue: The script seems to have created a plot but did not save it to 'plot.png'. Please ensure the generated code includes `plt.savefig('plot.png')`."
-                elif "FileNotFoundError: [Errno 2] No such file or directory: 'plot.png'" in execution_output:
-                    execution_output += "\n\nPlot Generation Issue: The script ran, but the plot file was not created. Ensure the generated code includes commands to save the plot to 'plot.png'."
+                try:
+                    # Add debug info to the script
+                    debug_script = f"""
+import traceback
+import sys
+
+try:
+    # Original script
+    {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")
+        
+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:")
+    traceback.print_exc()
+    raise  # Re-raise the exception to be caught by the outer try-except
+"""
+                    # Write the debug script to a temporary file
+                    with open(temp_script_path, "w") as f:
+                        f.write(debug_script)
+
+                    # Execute the script
+                    process = subprocess.run(
+                        ["python3", temp_script_path],
+                        capture_output=True,
+                        text=True,
+                        check=False
+                    )
+                    
+                    # Capture both stdout and stderr
+                    execution_output = process.stdout
+                    if process.stderr:
+                        execution_output += "\n\n[ERROR] Script execution errors:\n" + process.stderr
+                    
+                    # Check for common issues in the output
+                    if "KeyError" in execution_output:
+                        execution_output += "\n\n[HELP] The script encountered a KeyError. This typically happens when trying to access a column that doesn't exist in the stock data.\n"
+                        execution_output += "Common causes:\n"
+                        execution_output += "1. The stock symbol might not be recognized by yfinance\n"
+                        execution_output += "2. The requested time period might not have data (e.g., weekends, holidays)\n"
+                        execution_output += "3. The data column names might be different than expected\n\n"
+                        execution_output += "Please try a different stock symbol or time period."
+                        
+                    if "No data" in execution_output or "not found" in execution_output.lower():
+                        execution_output += "\n\n[HELP] No data was returned for the specified stock symbol or time period.\n"
+                        execution_output += "Please check the stock symbol and try a different time period."
+                        
+                    if "Figure(" in execution_output and "plot.png" not in os.listdir():
+                        execution_output += "\n\n[HELP] A plot was created but not saved. Adding save command...\n"
+                        # Try to save the plot if it wasn't saved
+                        try:
+                            import matplotlib.pyplot as plt
+                            if plt.get_fignums():
+                                plt.savefig('plot.png')
+                                execution_output += "Successfully saved plot to plot.png"
+                                generated_plot_path = 'plot.png'
+                            plt.close('all')
+                        except Exception as e:
+                            execution_output += f"Failed to save plot: {str(e)}"
+                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
                 if os.path.exists(plot_file_path):