Make smarter market decisions with AI

import yfinance as yf
from anthropic import Anthropic
import pandas as pd

client = Anthropic(api_key="your-key")

def fetch_data(ticker, period="1y"):
    stock = yf.Ticker(ticker)
    df = stock.history(period=period)
    return df

def analyze_patterns(df):
    df["returns"] = df["Close"].pct_change()
    df["ma20"] = df["Close"].rolling(20).mean()
    df["volatility"] = df["returns"].rolling(20).std()
    return df

def generate_signal(df):
    latest = df.iloc[-1]
    prompt = f"""
    Analyze this stock data:
    - Price: ${latest["Close"]:.2f}
    - 20-day MA: ${latest["ma20"]:.2f}
    - Volatility: {latest["volatility"]:.4f}
    
    Should we go long, short, or hold? Provide reasoning.
    """
    
    response = client.messages.create(
        model="claude-3-sonnet-20240229",
        max_tokens=200,
        messages=[{"role": "user", "content": prompt}]
    )
    return response.content[0].text

# Run daily
df = fetch_data("AAPL")
df = analyze_patterns(df)
signal = generate_signal(df)
print(signal)