Revert "vv4"

This reverts commit 76dd9b4b88e8d9922fa1cdfcc86858cf1bf3bfd6.

app.py

@@ -176,76 +176,22 @@ def run_crewai_process(user_query, model, temperature):
                     f.write(generated_code)
 
                 # Execute the temporary script using subprocess
-                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."
+                # 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'."
 
                 # Check for the generated plot file
                 if os.path.exists(plot_file_path):