css/layout.py

import streamlit as st


def custom_css():
    """Add custom CSS for layout"""
    st.markdown("""
        <style>
        .split-container {
            display: flex;
            gap: 2rem;
        }
        .context-container {
            flex: 1;
            padding: 1rem;
            background-color: #f8f9fa;
            border-radius: 0.5rem;
        }
        .questions-container {
            flex: 1;
            padding: 1rem;
        }
        .stButton button {
            width: 100%;
            margin-bottom: 0.5rem;
        }
        .exit-container {
            text-align: center;
            padding: 2rem;
            background-color: #f8f9fa;
            border-radius: 1rem;
            margin: 2rem auto;
            max-width: 600px;
        }
        </style>
    """, unsafe_allow_html=True)