main.py

import os
import pytz
import streamlit as st
from datetime import datetime
from streamlit_autorefresh import st_autorefresh
from modules.search_logic import start_search, stop_search, update_timer, sanitize_filename
from modules.ui import apply_styles, display_sidebar, display_timer, display_status, display_logs, display_sidebar_buttons

arg_tz = pytz.timezone("America/Argentina/Buenos_Aires")
URL = "https://psicologia.unc.edu.ar/egreso/practicas-supervisadas/orden-de-merito-inicio-ps/#collapse-1-3170"

st.set_page_config(
    page_title="PS-Mailer",
    page_icon="assets/icon.ico",
    layout="wide",
)

if "searching" not in st.session_state:
    st.session_state.searching = False
if "configured" not in st.session_state:
    st.session_state.configured = False
if "start_time" not in st.session_state:
    st.session_state.start_time = None
if "logs" not in st.session_state:
    st.session_state.logs = []
if "timer_text" not in st.session_state:
    st.session_state.timer_text = ""
if "institution" not in st.session_state:
    st.session_state.institution = ""
if "email" not in st.session_state:
    st.session_state.email = ""
if "interval" not in st.session_state:
    st.session_state.interval = 1
if "search_triggered" not in st.session_state:
    st.session_state.search_triggered = False

apply_styles()

def add_log(message):
    timestamp = datetime.now(arg_tz).strftime("%H:%M:%S")
    st.session_state.logs.append(f"[{timestamp}] {message}")

selected_date, selected_time, institution, email, interval = display_sidebar(
    st.session_state.institution, 
    st.session_state.email, 
    st.session_state.interval,
    st.session_state.configured,
    st.session_state.searching
)

configure_button = display_sidebar_buttons(
    st.session_state.configured,
    st.session_state.searching,
    lambda: stop_search(add_log)
)

if configure_button and not st.session_state.configured:
    if not selected_date or not selected_time or institution == "" or email == "" or interval == 0:
        st.error("⚠️ Completa todos los campos")
    else:
        try:
            naive_start = datetime.combine(selected_date, selected_time)
            st.session_state.start_time = arg_tz.localize(naive_start)
            st.session_state.institution = institution
            st.session_state.email = email
            st.session_state.interval = int(interval)
            st.session_state.configured = True
            st.session_state.search_triggered = False
            
            add_log(f"✅ Configuración guardada. La búsqueda se iniciará a las {st.session_state.start_time.strftime('%H:%M:%S')}")
            
            current_time = datetime.now(arg_tz)
            if current_time >= st.session_state.start_time:
                add_log("⏰ La hora configurada ya pasó - iniciando búsqueda inmediatamente")
                st.session_state.search_triggered = True
                st.session_state.searching = True
                st.rerun()
                
        except Exception as e:
            st.error(f"⚠️ Error en el formato: {str(e)}")

PDF_SAVE_NAME = os.path.join(os.path.dirname(__file__), f"{sanitize_filename(st.session_state.institution)}.pdf")

timer_title, timer_value = update_timer(add_log)
display_timer(timer_title, timer_value)

start_now = display_status(
    st.session_state.configured, 
    st.session_state.searching, 
    st.session_state.start_time
)

if start_now:
    st.session_state.searching = True
    st.session_state.search_triggered = True
    st.rerun()

display_logs(st.session_state.logs)

st_autorefresh(interval=1000, key="autorefresh")
if st.session_state.searching and st.session_state.search_triggered and not st.session_state.get('search_thread_running', False):
    start_search(
        st.session_state.institution,
        st.session_state.email,
        st.session_state.interval,
        st.session_state.start_time,
        add_log,
        URL,
        PDF_SAVE_NAME
    )