app.py

import streamlit as st
import os
from openai import OpenAI
from PIL import Image
import io
import base64
import pandas as pd
import plotly.express as px
import plotly.graph_objects as go
import numpy as np

st.set_page_config(
    page_title="Baseball Analytics Hub",
    layout="wide",
    initial_sidebar_state="expanded"
)

# Custom CSS for baseball theme
def apply_custom_css():
    st.markdown("""
        <style>
        .stApp {
            max-width: 1400px;
            margin: 0 auto;
        }
        .upload-box {
            border: 2px dashed #1E4C96;
            border-radius: 10px;
            padding: 20px;
            text-align: center;
            background-color: #f8f9fa;
        }
        .sidebar-content {
            padding: 20px;
            background-color: #1E4C96;
            color: white;
            border-radius: 10px;
        }
        .response-box {
            background-color: #ffffff;
            padding: 20px;
            border-radius: 10px;
            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
        }
        .title-text {
            color: #1E4C96;
            text-align: center;
            font-size: 2.5rem;
            margin-bottom: 2rem;
        }
        .stats-card {
            background-color: #ffffff;
            padding: 20px;
            border-radius: 10px;
            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
            margin: 10px;
        }
        </style>
    """, unsafe_allow_html=True)

def initialize_session_state():
    if 'history' not in st.session_state:
        st.session_state.history = []
    if 'player_stats' not in st.session_state:
        st.session_state.player_stats = {
            'batting_avg': [],
            'hits': [],
            'home_runs': [],
            'games': []
        }

def encode_image_to_base64(image):
    buffered = io.BytesIO()
    image.save(buffered, format="PNG")
    return base64.b64encode(buffered.getvalue()).decode('utf-8')

def analyze_image(image, question, api_key):
    try:
        client = OpenAI(api_key=api_key)
        base64_image = encode_image_to_base64(image)
        
        # Add baseball-specific context to the prompt
        baseball_context = "Analyze this baseball-related image. Consider aspects like player stance, pitch mechanics, field positioning, and game situation. "
        enhanced_question = baseball_context + question
        
        completion = client.chat.completions.create(
            model="gpt-4o-mini",
            messages=[
                {
                    "role": "user",
                    "content": [
                        {"type": "text", "text": enhanced_question},
                        {
                            "type": "image_url",
                            "image_url": {
                                "url": f"data:image/png;base64,{base64_image}"
                            }
                        },
                    ],
                }
            ],
            max_tokens=500
        )
        return completion.choices[0].message.content
    except Exception as e:
        return f"Error: {str(e)}"

def plot_strike_zone():
    fig = go.Figure()
    fig.add_shape(
        type="rect",
        x0=-0.85, x1=0.85,
        y0=1.5, y1=3.5,
        line=dict(color="red", width=2),
        fillcolor="rgba(255, 0, 0, 0.1)"
    )
    fig.add_trace(go.Scatter(
        x=np.random.uniform(-1, 1, 10),
        y=np.random.uniform(1, 4, 10),
        mode='markers',
        marker=dict(size=10, color='blue'),
        name='Pitches'
    ))
    fig.update_layout(
        title="Strike Zone Analysis",
        xaxis_title="Horizontal Location (ft)",
        yaxis_title="Height (ft)",
        template="plotly_white"
    )
    return fig

def main():
    apply_custom_css()
    initialize_session_state()

    # Sidebar Configuration
    with st.sidebar:
        st.markdown('<div class="sidebar-content">', unsafe_allow_html=True)
        st.header("⚾ Baseball Analytics")
        api_key = st.text_input("OpenAI API Key:", type="password")
        
        st.markdown("### 🎯 Baseball Image Analysis")
        quick_prompts = [
            "Analyze the batting stance and suggest improvements",
            "Evaluate the pitcher's mechanics",
            "Analyze the fielding position and alignment",
            "Identify potential batting technique issues",
            "Assess the base running form",
            "Analyze the catcher's positioning",
            "Evaluate the pitch grip technique"
        ]
        selected_prompt = st.selectbox("Select analysis type:", quick_prompts)
        st.markdown("</div>", unsafe_allow_html=True)

    # Main Content
    st.markdown('<h1 class="title-text">⚾ Baseball Vision Analytics</h1>', unsafe_allow_html=True)

    # Create tabs for different features
    tab1, tab2 = st.tabs(["Image Analysis", "Performance Analytics"])

    with tab1:
        # Image Upload Section
        col1, col2, col3 = st.columns([1, 2, 1])
        with col2:
            st.markdown('<div class="upload-box">', unsafe_allow_html=True)
            uploaded_file = st.file_uploader(
                "Upload baseball image for analysis",
                type=['png', 'jpg', 'jpeg'],
                help="Upload images of batting, pitching, or fielding"
            )
            st.markdown('</div>', unsafe_allow_html=True)

        if uploaded_file:
            image = Image.open(uploaded_file)
            
            # Image Display and Analysis Section
            col1, col2 = st.columns([1, 1])
            
            with col1:
                st.image(image, use_container_width=True, caption="Uploaded Baseball Image")
                
            with col2:
                st.markdown('<div class="response-box">', unsafe_allow_html=True)
                question = st.text_input(
                    "Analysis Focus:",
                    value=selected_prompt,
                    key="question_input"
                )
                
                if st.button("🔍 Analyze", use_container_width=True):
                    if not api_key:
                        st.error("⚠️ Please enter your OpenAI API key in the sidebar.")
                    else:
                        with st.spinner("🔄 Analyzing baseball technique..."):
                            response = analyze_image(image, question, api_key)
                            st.markdown("### 📝 Analysis Result:")
                            st.write(response)
                            
                            st.session_state.history.append({
                                "question": question,
                                "response": response,
                                "timestamp": pd.Timestamp.now().strftime("%Y-%m-%d %H:%M:%S")
                            })
                st.markdown('</div>', unsafe_allow_html=True)

    with tab2:
        # Performance Analytics Section
        col1, col2 = st.columns(2)
        
        with col1:
            st.markdown('<div class="stats-card">', unsafe_allow_html=True)
            st.subheader("Strike Zone Analysis")
            fig = plot_strike_zone()
            st.plotly_chart(fig, use_container_width=True)
            st.markdown('</div>', unsafe_allow_html=True)
        
        with col2:
            st.markdown('<div class="stats-card">', unsafe_allow_html=True)
            st.subheader("Key Metrics")
            metrics_col1, metrics_col2 = st.columns(2)
            with metrics_col1:
                st.metric("Pitch Accuracy", "68%", "+2%")
                st.metric("Avg Exit Velocity", "92 mph", "-1 mph")
            with metrics_col2:
                st.metric("Strike Rate", "65%", "+5%")
                st.metric("Spin Rate", "2400 rpm", "+100 rpm")
            st.markdown('</div>', unsafe_allow_html=True)

    # History Section
    if st.session_state.history:
        st.markdown("### 📜 Previous Analyses")
        for item in reversed(st.session_state.history[-5:]):
            with st.expander(f"Q: {item['question'][:50]}..."):
                st.write(item['response'])
                st.caption(f"Analyzed on: {item['timestamp']}")

    # Footer
    st.markdown("---")
    st.markdown(
        """
        <div style='text-align: center'>
            <p>Powered by Advanced Baseball Analytics and OpenAI Vision API</p>
            <p>© 2024 Baseball Vision Analytics</p>
        </div>
        """,
        unsafe_allow_html=True
    )

if __name__ == "__main__":
    main()