app.py

import streamlit as st
from datetime import date
from streamlit_js_eval import streamlit_js_eval  # For geolocation
from src.data_acquisition import get_solunar_data
from src.data_processing import process_solunar_data
from src.solunar_calculations import calculate_major_minor_times
from src.recommendation_engine import generate_recommendations
import folium
from streamlit_folium import st_folium

st.set_page_config(page_title="SolunarBass", page_icon="🎣")

st.title("🎣 SolunarBass")
st.subheader("Optimal Bass Fishing Times Based on Solunar Theory")

# Initialize session state variables
if 'use_current_location' not in st.session_state:
    st.session_state.use_current_location = False
if 'latitude' not in st.session_state:
    st.session_state.latitude = 38.8951
if 'longitude' not in st.session_state:
    st.session_state.longitude = -77.0364
if 'loc' not in st.session_state:
    st.session_state.loc = None
if 'data_fetched' not in st.session_state:
    st.session_state.data_fetched = False
if 'solunar_data' not in st.session_state:
    st.session_state.solunar_data = None
if 'recommendations' not in st.session_state:
    st.session_state.recommendations = None

# Sidebar Inputs
st.sidebar.header("Input Parameters")

# Checkbox to use current location
use_current_location = st.sidebar.checkbox(
    "Use my current location",
    value=st.session_state.use_current_location
)
st.session_state.use_current_location = use_current_location

# Button to fetch data
fetch_data = st.sidebar.button("Get Fishing Times")

if st.session_state.use_current_location:
    # Get the user's location using JavaScript
    if st.session_state.loc is None:
        loc = streamlit_js_eval(
            js_expressions="""
            new Promise((resolve, reject) => {
                navigator.geolocation.getCurrentPosition(
                    (position) => {
                        resolve({
                            latitude: position.coords.latitude,
                            longitude: position.coords.longitude
                        });
                    },
                    (error) => {
                        resolve(null);
                    }
                );
            });
            """,
            key="get_location",
        )
        if loc:
            if 'latitude' in loc and 'longitude' in loc:
                st.session_state.latitude = loc["latitude"]
                st.session_state.longitude = loc["longitude"]
                st.session_state.loc = loc
                st.sidebar.success(
                    f"Location acquired: "
                    f"({st.session_state.latitude:.6f}, "
                    f"{st.session_state.longitude:.6f})"
                )
            else:
                st.sidebar.error(
                    "Unable to retrieve location. "
                    "Please allow location access."
                )
        else:
            st.sidebar.warning(
                "Waiting for location... "
                "Make sure to allow location access."
            )
else:
    # Manual input
    st.session_state.latitude = st.sidebar.number_input(
        "Latitude", value=st.session_state.latitude, format="%.6f"
    )
    st.session_state.longitude = st.sidebar.number_input(
        "Longitude", value=st.session_state.longitude, format="%.6f"
    )

selected_date = st.sidebar.date_input("Date", value=date.today())

if fetch_data:
    with st.spinner('Fetching data...'):
        raw_data = get_solunar_data(
            st.session_state.latitude,
            st.session_state.longitude,
            selected_date
        )
        if raw_data is None:
            st.error("Failed to retrieve data. Please try again later.")
        else:
            # Convert date to string
            date_str = selected_date.strftime('%Y-%m-%d')
            st.session_state.solunar_data = process_solunar_data(
                raw_data, date_str
            )
            major_times, minor_times = calculate_major_minor_times(
                st.session_state.solunar_data
            )
            st.session_state.recommendations = generate_recommendations(
                major_times, minor_times
            )
            st.session_state.data_fetched = True

# Display Results if Data Has Been Fetched
if st.session_state.data_fetched and st.session_state.solunar_data:
    st.success("Recommendations Generated!")

    # Display Recommendations
    st.header("Recommended Fishing Times")
    if st.session_state.recommendations:
        for rec in st.session_state.recommendations:
            start_time = rec['start'].strftime('%I:%M %p')
            end_time = rec['end'].strftime('%I:%M %p')
            st.write(f"**{rec['type']} Period:** {start_time} - {end_time}")
    else:
        st.warning(
            "No major or minor times could be calculated. "
            "This may be due to missing moonrise/moonset data."
        )

    # Display Additional Information
    st.header("Additional Information")
    st.write(
        f"**Location:** {st.session_state.latitude:.6f}, "
        f"{st.session_state.longitude:.6f}"
    )
    sunrise_str = st.session_state.solunar_data['sunrise'].strftime('%I:%M %p')
    st.write(f"**Sunrise:** {sunrise_str}")
    sunset_str = st.session_state.solunar_data['sunset'].strftime('%I:%M %p')
    st.write(f"**Sunset:** {sunset_str}")
    moon_phase = st.session_state.solunar_data['moon_phase']
    st.write(f"**Moon Phase:** {moon_phase}")

    # Display Map
    st.header("Map")
    m = folium.Map(
        location=[st.session_state.latitude, st.session_state.longitude],
        zoom_start=12
    )
    folium.Marker(
        [st.session_state.latitude, st.session_state.longitude],
        tooltip="Your Location"
    ).add_to(m)
    st_folium(m, width=700, height=500)
else:
    st.info("Please enter parameters and click 'Get Fishing Times'")

# Footer with social links
st.markdown("---")
st.markdown(
    """
    <div style="text-align: center;">
        <p>
            <a href="https://bassfinity.com"
               target="_blank">Bassfinity.com</a> |
            <a href="https://www.twitter.com/bassfinity"
               target="_blank">Twitter</a> |
            <a href="https://www.tiktok.com/@bassfinity_og"
               target="_blank">TikTok</a> |
            <a href="https://www.instagram.com/bassfinity_com"
               target="_blank">Instagram</a>
        </p>
    </div>
    """,
    unsafe_allow_html=True
)