app.py

# app.py
# Main entry point for the Streamlit application.
import streamlit as st
import logging
from datetime import datetime, timedelta
import os
from dotenv import load_dotenv

from src.stocks import is_valid_stock_code
from src.config import (
    MAX_LOGIN_ATTEMPTS,
    LOCKOUT_MINUTES,
    DEFAULT_INITIAL_CAPITAL,
    DEFAULT_RISK_TOLERANCE,
)
from src.workers.queue_worker import start_worker
from stock_analysis import analyse_stock

load_dotenv()

logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(name)s: %(message)s")
logger = logging.getLogger(__name__)

start_worker()

st.set_page_config(layout="wide")

# --- Authentication ---
PASSKEY = os.getenv("PASSKEY")
hide_streamlit_style = """
    <style>
    #MainMenu {visibility: hidden;}
    header {visibility: hidden;}
    footer {visibility: hidden;}
    </style>
    """
st.markdown(hide_streamlit_style, unsafe_allow_html=True)

if "authenticated" not in st.session_state:
    st.session_state.authenticated = False

if "incorrect_count" not in st.session_state:
    st.session_state.incorrect_count = 0

if "pending_stock" not in st.session_state:
    st.session_state.pending_stock = None

if st.session_state.get("locked_until"):
    if datetime.now() < st.session_state.locked_until:
        remaining = int((st.session_state.locked_until - datetime.now()).total_seconds())
        st.error(f"Too many incorrect password attempts. Please wait {remaining} seconds before trying again.")
        st.stop()
    else:
        st.session_state.incorrect_count = 0
        st.session_state.locked_until = None

if not st.session_state.authenticated:
    st.title("Login")
    password = st.text_input("Enter Passkey", type="password")
    if st.button("Login"):
        if password == PASSKEY:
            st.session_state.authenticated = True
            st.session_state.incorrect_count = 0
            st.success("Access Granted!")
            st.rerun()
        else:
            st.session_state.incorrect_count += 1
            if st.session_state.incorrect_count >= MAX_LOGIN_ATTEMPTS:
                st.session_state.locked_until = datetime.now() + timedelta(minutes=LOCKOUT_MINUTES)
                st.error(f"Too many incorrect password attempts. Locked out for {LOCKOUT_MINUTES} minutes.")
                st.stop()
            remaining = MAX_LOGIN_ATTEMPTS - st.session_state.incorrect_count
            st.error(f"Incorrect passkey. Attempt {st.session_state.incorrect_count} of {MAX_LOGIN_ATTEMPTS}.")
    st.stop()

# --- Poll for pending stock result ---
if st.session_state.pending_stock:
    from src.storage.cache import get_stock_analysis
    pending = st.session_state.pending_stock
    cached = get_stock_analysis(pending)
    if cached:
        st.session_state.pending_stock = None
        st.success(f"Analysis for {pending} is ready!")
        st.markdown(cached)
        st.stop()
    else:
        st.info(f"Analysis for {pending} is still processing...")
        st.info("Click the button again to check if results are ready.")
        st.session_state.pending_stock = pending

# --- Main Application UI ---
st.title("All India Stock Analysis Agent")
st.write("Enter the stock code below to generate analysis for it")

stock = st.text_input("Stock analysis", placeholder="STOCK SYMBOL e.g., TATAMOTORS")
initial_capital = st.number_input("Initial capital:", value=DEFAULT_INITIAL_CAPITAL, step=1, format="%d")

button_placeholder = st.empty()

if button_placeholder.button("Generate Analysis"):
    if stock and is_valid_stock_code(stock) is not None:
        stock = stock.strip()
        button_placeholder.empty()
        logger.info("Query for %s", stock)
        result = analyse_stock(stock, initial_capital, DEFAULT_RISK_TOLERANCE)

        if "queued" in result.lower() or "already in the queue" in result.lower():
            st.session_state.pending_stock = stock
            st.info(result)
            st.info("The page will re-check automatically. You can also click the button again to check manually.")
        else:
            st.session_state.pending_stock = None
            st.subheader("Generated Stock Analysis")
            st.markdown(result)
    else:
        st.error("Please enter a valid stock code.")

st.markdown(
    """
    <div style="text-align: center; font-size: small; color: gray; margin-top: 2rem;">
      <p><strong>Disclaimer:</strong> This content is generated by AI and provided for informational purposes only. It is not intended as financial, legal, or professional advice by any actual human.</p>
      <p>Please consult with a qualified professional before making any decisions based on this information.</p>
    </div>
    """,
    unsafe_allow_html=True
)