v119

app.py

@@ -191,10 +191,24 @@ def run_crewai_process(user_query, model, temperature):
                 with open(temp_script_path, 'r') as f:
                     script_content = f.read()
                 
-                # Update yf.download() calls to include auto_adjust=True
+                # Update yf.download() calls to include auto_adjust parameter
+                def add_auto_adjust(match):
+                    # Check if auto_adjust is already in the arguments
+                    args = match.group(1).strip()
+                    if 'auto_adjust' not in args:
+                        # Add auto_adjust=True to the arguments
+                        if args.endswith(','):
+                            return f'yf.download({args} auto_adjust=True)'
+                        elif args:  # If there are existing arguments
+                            return f'yf.download({args}, auto_adjust=True)'
+                        else:  # If no arguments
+                            return 'yf.download(auto_adjust=True)'
+                    return match.group(0)  # Return unchanged if auto_adjust is already present
+                
+                # This pattern matches yf.download() with any arguments
                 script_content = re.sub(
-                    r'yf\.download\(([^)]+)\)',
-                    r'yf.download(\1, auto_adjust=True)',
+                    r'yf\.download\(([^)]*)\)',
+                    add_auto_adjust,
                     script_content
                 )