v111

app.py

@@ -4,6 +4,7 @@ import sys
 import re
 import traceback
 import subprocess
+import warnings
 import gradio as gr
 import pandas as pd
 from dotenv import load_dotenv
@@ -11,6 +12,9 @@ from crewai import Crew, Agent, Task, Process, LLM
 from crewai_tools import FileReadTool
 from pydantic import BaseModel, Field
 
+# Filter out specific warnings
+warnings.filterwarnings('ignore', category=FutureWarning, module='yfinance.*')
+
 # Load environment variables
 load_dotenv()
 
@@ -175,6 +179,21 @@ def run_crewai_process(user_query, model, temperature):
                 with open(temp_script_path, "w") as f:
                     f.write(generated_code)
 
+                # Add auto_adjust=True to yf.download() calls in the generated script
+                with open(temp_script_path, 'r') as f:
+                    script_content = f.read()
+                
+                # Update yf.download() calls to include auto_adjust=True
+                script_content = re.sub(
+                    r'yf\.download\(([^)]+)\)',
+                    r'yf.download(\1, auto_adjust=True)',
+                    script_content
+                )
+                
+                # Write the updated script back
+                with open(temp_script_path, 'w') as f:
+                    f.write(script_content)
+                
                 # Execute the temporary script using subprocess
                 # Use python3 to ensure correct interpreter in Colab
                 process = subprocess.run(