app.py

sys.path.append(os.path.abspath(os.path.dirname(__file__)))
import sys
import os
import json
import logging
import uuid
import streamlit as st
from src.chatbot import generate_response_with_history, COUNTRY_HELPLINES, get_helpline_for_country

# --- Setup ---
sys.path.append(os.path.abspath("src"))
logging.basicConfig(filename="safeguard_logs.txt", level=logging.INFO)
METRICS_FILE = "user_metrics.json"

# --- Custom Color Theme (BAC7D8/D75D72/101920) ---
CUSTOM_CSS = f"""
<style>
    .stApp {{
        background-color: #101920;
        color: #BAC7D8;
    }}
    .stTextInput, .stSelectbox, .stButton {{
        background-color: #1A242E !important;
        border-color: #D75D72 !important;
    }}
    h1, h2, h3 {{
        color: #D75D72 !important;
    }}
    .stMarkdown {{
        color: #BAC7D8;
    }}
    .stAlert {{
        background-color: #1A242E !important;
    }}
</style>
"""
st.markdown(CUSTOM_CSS, unsafe_allow_html=True)

# --- Initialize Session State ---
def init_session():
    if "metrics" not in st.session_state:
        st.session_state.metrics = load_metrics()
    if "session_id" not in st.session_state:
        st.session_state.session_id = str(uuid.uuid4())[:8]
    if "chat_history" not in st.session_state:
        st.session_state.chat_history = []

# --- Data Handling ---
def load_metrics():
    if os.path.exists(METRICS_FILE):
        try:
            with open(METRICS_FILE, "r") as f:
                return json.load(f)
        except Exception as e:
            st.error(f"Error loading metrics: {e}")
    return {"sessions": [], "feedback": {"thumbs_up": 0, "thumbs_down": 0}, "harmful_content": []}

def save_metrics():
    try:
        with open(METRICS_FILE, "w") as f:
            json.dump(st.session_state.metrics, f, indent=4)
    except Exception as e:
        st.error(f"Error saving metrics: {e}")

# --- UI Components ---
def render_chat_message(role, content):
    with st.chat_message(role):
        st.markdown(f"**{'You' if role == 'user' else 'Bot'}:** {content}")

def render_helpline():
    with st.sidebar:
        st.subheader("🌍 Country Support")
        selected_country = st.selectbox(
            "Select your country:", 
            options=list(COUNTRY_HELPLINES.keys()),
            index=0
        )
        st.markdown(f"**📞 {selected_country} Helpline:**  \n{get_helpline_for_country(selected_country)}")

# --- Main App ---
def main():
    st.title("JoyCloud Mental Health Assistant 🧠")
    st.markdown("---")
    
    # Sidebar Components
    with st.sidebar:
        st.markdown("### Session Info")
        st.write(f"Session ID: `{st.session_state.session_id}`")
        st.markdown("---")
        st.markdown("**Disclaimer:**  \nThis is a prototype chatbot. For emergencies, contact a professional.")
    
    # Main Chat Interface
    render_helpline()
    
    # Chat History Display
    for msg in st.session_state.chat_history:
        render_chat_message(msg["role"], msg["content"])
    
    # User Input
    prompt = st.chat_input("Type your message...")
    if prompt:
        try:
            # Append user message
            st.session_state.chat_history.append({"role": "user", "content": prompt})
            render_chat_message("user", prompt)
            
            # Generate response
            with st.spinner("Thinking..."):
                response = generate_response_with_history(
                    st.session_state.chat_history,
                    st.session_state.selected_country
                )
            
            # Append bot response
            st.session_state.chat_history.append({"role": "bot", "content": response})
            render_chat_message("bot", response)
            
        except Exception as e:
            st.error(f"Error generating response: {e}")
            logging.error(f"Session {st.session_state.session_id}: {str(e)}")
    
    # Feedback System (Floating at Bottom)
    st.markdown("---")
    col1, col2 = st.columns(2)
    with col1:
        if st.button("👍 Helpful", use_container_width=True, type="primary"):
            st.session_state.metrics["feedback"]["thumbs_up"] += 1
            save_metrics()
            st.toast("Thank you for your feedback!")
    with col2:
        if st.button("👎 Needs Improvement", use_container_width=True, type="secondary"):
            st.session_state.metrics["feedback"]["thumbs_down"] += 1
            save_metrics()
            st.toast("We'll use this to improve!")

# --- Run App ---
if __name__ == "__main__":
    init_session()
    main()