import streamlit as st
import plotly.express as px
import pandas as pd
import numpy as np
from datetime import datetime, timedelta

# Page configuration
st.set_page_config(
    page_title="Prospira",
    page_icon="📊",
    layout="wide",
    initial_sidebar_state="expanded"
)

# Session state initialization
if 'current_page' not in st.session_state:
    st.session_state['current_page'] = 'Dashboard'

# Sidebar navigation
def sidebar():
    with st.sidebar:
        st.title("Prospira 📊")
        st.subheader("Navigation")
        
        pages = {
            "Dashboard": "📈",
            "Analytics": "📊",
            "Brainstorm": "🧠",
            "Chat": "💬"
        }
        
        for page, emoji in pages.items():
            if st.button(f"{emoji} {page}"):
                st.session_state['current_page'] = page

# Generate sample data for demonstration
def generate_sample_data():
    dates = pd.date_range(start='2024-01-01', end='2024-01-31', freq='D')
    return pd.DataFrame({
        'Date': dates,
        'Revenue': np.random.normal(1000, 100, len(dates)),
        'Users': np.random.randint(100, 200, len(dates)),
        'Engagement': np.random.uniform(0.5, 0.9, len(dates))
    })

# Page components
def show_dashboard():
    st.header("Dashboard")
    col1, col2 = st.columns(2)
    
    data = generate_sample_data()
    
    with col1:
        st.subheader("Revenue Trend")
        fig = px.line(data, x='Date', y='Revenue')
        st.plotly_chart(fig, use_container_width=True)
    
    with col2:
        st.subheader("User Engagement")
        fig = px.scatter(data, x='Date', y='Engagement', size='Users')
        st.plotly_chart(fig, use_container_width=True)

def show_analytics():
    st.header("Analytics")
    st.info("Select data for analysis:")
    
    uploaded_file = st.file_uploader("Upload your data (CSV)", type=['csv'])
    if uploaded_file:
        data = pd.read_csv(uploaded_file)
        st.write("Data Preview:", data.head())
        
        if st.button("Generate Analysis"):
            st.write("Basic Statistics:", data.describe())

def show_brainstorm():
    st.header("Brainstorm")
    
    products = ["Product A", "Product B", "Product C"]
    selected_product = st.selectbox("Select Product", products)
    
    if selected_product:
        st.subheader(f"Analysis for {selected_product}")
        st.write("Product performance metrics will appear here")
        
        if st.button("Add New Product"):
            st.text_input("Product Name")
            st.text_area("Product Description")
            st.button("Submit")

def show_chat():
    st.header("Chat Interface")
    
    user_input = st.text_input("Ask me anything about your business:")
    if user_input:
        st.write("Bot: Thank you for your question! This feature will be implemented with LLAMA integration.")

def main():
    sidebar()
    
    # Page routing
    pages = {
        "Dashboard": show_dashboard,
        "Analytics": show_analytics,
        "Brainstorm": show_brainstorm,
        "Chat": show_chat
    }
    
    current_page = st.session_state['current_page']
    pages[current_page]()

if __name__ == "__main__":
    main()