Update main.py

main.py

@@ -44,21 +44,16 @@ def execute_python(code: str) -> str:
     """Execute Python code in an IPython interactiveshell and return the output.
        The returned artifacts (if present) are automatically rendered in the UI and visible to the user. 
     Args:
-        code: The Python code to execute
+        code: Valid Python code with correct indentation and syntax including necessary imports.
+
     Available Libraries:
-    # Use plotly as the default charting library
-    # While using yfinance to pull stock data, Always clean the multiindex columns as this might cause issues in plotting plotly charts
-    # Remove the ticker level from columns if it exists
-    yf_data = yf.download(symbol, start=start_date, end=end_date)
-    if isinstance(yf_data.columns, pd.MultiIndex):
-        yf_data.columns = yf_data.columns.get_level_values(0)
-    matplotlib
-    pandas
+    # Use Plotly for creating visualizations
+
     plotly
+    pandas
     groq
     yfinance
     numpy
-    seaborn
     numpy
     scikit-learn
     statsmodels