app.py

import streamlit as st
import os
from utils.logging_utils import setup_logging
from typing import Dict, List
from datetime import datetime
from dotenv import load_dotenv
from openai import OpenAI
import json
import time
import hashlib
import base64

# Configure page settings and styling
svg_content = '''
<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 502 512">
<path fill="#c5203e" d="M157.1 91.6l-5.5 32.2H135.5c-9 0-13.6.7-13.6 7v15.6h10.5c9.3 0 10.5.2 10.5 1.7c0 1.8-4.5 27.8-5.5 32.1c-.5 2.2-.9 2.3-12 2.3H114v15.6 15.6h8.4c4.7 0 8.7.3 8.9.7c.3.5-.3 5.3-1.3 10.8s-2.3 13.4-3 17.3l-1.2 7.3h-13-13v15.1 15.1H241.5 383v-15.1-15.1h-78.3-78.3v-2.1c0-2.8 5-32.1 5.6-33.2c.3-.4 37.3-.8 82.3-.8h81.7v-15.6-15.6h-79.3c-57 0-79.3-.3-79.3-1.1c0-1 4.4-26.2 5.6-32.3l.6-2.7h80.7 80.8l-.3-15.4-.3-15.3-9.3-.3c-8.9-.3-9.2-.3-8.7-2.5c.6-3 5.5-31.3 5.5-32.1c0-.4-7.2-.7-16.1-.7s-16.1.4-16.1.8c0 .9-4.8 28.8-5.5 32c-.5 2.3-.7 2.3-18.1 2.3c-9.6 0-17.6-.2-17.6-.5c0-.7 5-29.8 5.5-32.4l.5-2.2h-16.1-16.1l-1.5 2.2c-.5 2.6-5.5 31.7-5.5 32.4c0 .3-8.5.5-18.9.5l-19 0 2.8-15.9c1.6-8.6 2.8-16.6 2.9-17.6c0-1.5-1.4-1.7-16.1-1.7c-8.8 0-16.1.2-16.1.5c0 .7-5 29.8-5.5 32.3c-.5 2.3-.7 2.3-18.2 2.3H180l.5-2.3c.3-1.2 1.5-7.8 2.7-14.8c1.1-6.9 2.3-13.8 2.6-15.4l.6-2.7h-16.2c-15.9 0-16.2 0-16.7 2.2zm56.7 64.7c0 1.7-4.6 30.1-5.5 34.2c-.5 2.3-.7 2.3-18.6 2.3c-12.6 0-18.1-.3-18.1-1.1c0-1.2 4.6-29.5 5.6-34.3l.6-2.7h18c15.9 0 18 .2 18 1.6zm-12 67.3c0 .9-4.9 30.7-5.5 33.2c-.5 2.3-.7 2.3-18.1 2.3h-17.6l.5-2.3c.2-1.2 1.6-8.9 3-17.3c1.5-8.2 2.9-15.4 3.1-15.8c.6-.9 34.5-1.1 34.5-.2zM93.4 341.3v15.1h9c7.4 0 9 .3 9 1.5c0 .9-1.1 8.1-2.5 16.2s-2.5 15.7-2.5 17.1l0 2.4H122c15.3 0 15.6 0 16.1-2.3c.3-1.2 1.8-9.4 3.3-18.3l2.8-16.1 17.3-.3 17.3-.2v2.2c0 1.3-1.1 8.8-2.5 16.9s-2.5 15.4-2.5 16.3c0 1.6 1.4 1.8 15.9 1.8h15.9l3-18.4 2.9-18.3 19.4-.3c12.7-.2 19.3.1 19.3.8c0 1.2-4.8 30.5-5.5 33.8l-.5 2.3H260c8.8 0 16.1-.1 16.1-.3c.1-.1 1.5-8.3 3.3-18.3l3.2-18.1 17.4-.3c16-.2 17.3-.1 17.3 1.5c0 1-1.1 8.3-2.5 16.4s-2.5 15.6-2.5 16.9l0 2.3 15.8-.3 15.8-.3 3.1-18.1 3-18.1 13.4-.3 13.3-.3v-15-15.1l-141.6 0-141.6 0v15.1z"/>
</svg>
'''


def get_svg_base64(svg_content):
    b64 = base64.b64encode(svg_content.encode('utf-8')).decode('utf-8')
    return f"data:image/svg+xml;base64,{b64}"


# Configure the page with your brand colors
st.set_page_config(
    page_title="VetsAI: Vets Who Code Assistant",
    page_icon=get_svg_base64(svg_content),
    layout="wide"
)

# Define VWC brand colors and add custom CSS
st.markdown("""
    <style>
    /* VWC Brand Colors */
    :root {
        --navy-blue: #091f40;
        --red: #c5203e;
        --white: #ffffff;
    }
    
    /* Sidebar styles */
    [data-testid="stSidebarUserContent"] {
        color: var(--white) !important;
    }
    
    [data-testid="stSidebarUserContent"] .stMarkdown {
        color: var(--white) !important;
    }
    
    /* Make sidebar headers and text white */
    [data-testid="stSidebarUserContent"] h1,
    [data-testid="stSidebarUserContent"] h2,
    [data-testid="stSidebarUserContent"] h3,
    [data-testid="stSidebarUserContent"] h4,
    [data-testid="stSidebarUserContent"] h5,
    [data-testid="stSidebarUserContent"] h6,
    [data-testid="stSidebarUserContent"] p,
    [data-testid="stSidebarUserContent"] li {
        color: var(--white) !important;
    }
    
    /* Style strong tags in sidebar */
    [data-testid="stSidebarUserContent"] strong {
        color: var(--white) !important;
    }

        code {
        font-family: "Source Code Pro", monospace;
        padding: 0.2em 0.4em;
        margin: 0px;
        border-radius: 0.25rem;
        background: var(--navy-blue);
        color: var(--white) !important;
        overflow-wrap: break-word;
    }

 /* Target both placeholder and actual text */
    .stChatInput textarea,
    .stChatInput textarea::placeholder,
    .stChatInput textarea:not(:placeholder-shown) {
        color: white !important;
    }
    
    /* Force text color for active input */
    .stChatInput textarea:focus {
        color: white !important;
    }
    
    /* Target the wrapper to ensure color inheritance */
    [data-testid="stChatInput"] > div {
        color: white !important;
    }
    </style>
""", unsafe_allow_html=True)

logger = setup_logging()

# Load environment variables
load_dotenv()

# Optionally, load the model name if needed
client = OpenAI(api_key=st.secrets["openai"]["OPENAI_API_KEY"])

# Test the setup
if not client.api_key:  # Changed from openai.api_key to client.api_key
    raise ValueError("OpenAI API key not found in Streamlit secrets.")


def parse_mos_file(file_content: str) -> dict:
    """Parse military job code text file content into a structured dictionary."""
    lines = file_content.strip().split('\n')
    job_code, title, description = "", "", []
    parsing_description = False

    for line in lines:
        line = line.strip()
        if not line:
            continue
        if line.startswith("Job Code:"):
            job_code = line.replace("Job Code:", "").strip()
        elif line.startswith("Description:"):
            parsing_description = True
        elif parsing_description:
            description.append(line)

    for line in description:
        if line:
            title = line
            break

    full_text = ' '.join(description).lower()
    category = "general"
    category_keywords = {
        "information_technology": ["technology", "computer", "network", "data", "software", "hardware", "system",
                                   "database"],
        "communications": ["communications", "signal", "radio", "transmission", "telecom"],
        "intelligence": ["intelligence", "analysis", "surveillance", "reconnaissance"],
        "maintenance": ["maintenance", "repair", "technical", "equipment"],
        "cyber": ["cyber", "security", "information assurance", "cryptographic"]
    }

    for cat, keywords in category_keywords.items():
        if any(keyword in full_text for keyword in keywords):
            category = cat
            break

    return {
        "title": title or "Military Professional",
        "category": category,
        "skills": [line for line in description if line and len(line) > 10]
    }


def load_military_job_codes() -> dict:
    base_path = "data/employment_transitions/job_codes"
    job_codes = {}
    branches = {
        "army": {"path": "army", "prefix": "MOS"},
        "air_force": {"path": "air_force", "prefix": "AFSC"},
        "coast_guard": {"path": "coast_guard", "prefix": "RATE"},
        "navy": {"path": "navy", "prefix": "RATE"},
        "marine_corps": {"path": "marine_corps", "prefix": "MOS"}
    }

    for branch, info in branches.items():
        branch_path = os.path.join(base_path, info["path"])
        if os.path.exists(branch_path):
            for file in os.listdir(branch_path):
                if file.endswith('.txt'):
                    try:
                        with open(os.path.join(branch_path, file), 'r') as f:
                            content = f.read()
                            code = file.replace('.txt', '')
                            details = parse_mos_file(content)
                            vwc_mapping = map_to_vwc_path(details.get('category', ''), details.get('skills', []))
                            details.update({
                                'vwc_path': vwc_mapping['path'],
                                'tech_focus': vwc_mapping['tech_focus'],
                                'branch': branch,
                                'code_type': info['prefix']
                            })
                            job_codes[f"{info['prefix']}_{code}"] = details
                    except Exception as e:
                        logger.error(f"Error loading {file}: {e}")
                        continue
    return job_codes


def map_to_vwc_path(category: str, skills: List[str]) -> dict:
    """Map military job categories and skills to VWC tech stack paths."""
    default_path = {
        "path": "Full Stack Development",
        "tech_focus": [
            "JavaScript/TypeScript fundamentals",
            "Next.js and Tailwind for frontend",
            "Python with FastAPI/Django for backend"
        ]
    }

    tech_paths = {
        "information_technology": {
            "path": "Full Stack Development",
            "tech_focus": [
                "JavaScript/TypeScript with focus on system architecture",
                "Next.js for complex web applications",
                "Python backend services with FastAPI"
            ]
        },
        "cyber": {
            "path": "Security-Focused Development",
            "tech_focus": [
                "TypeScript for type-safe applications",
                "Secure API development with FastAPI/Django",
                "AI/ML for security applications"
            ]
        },
        "communications": {
            "path": "Frontend Development",
            "tech_focus": [
                "JavaScript/TypeScript specialization",
                "Advanced Next.js and Tailwind",
                "API integration with Python backends"
            ]
        },
        "intelligence": {
            "path": "AI/ML Development",
            "tech_focus": [
                "Python for data processing",
                "ML model deployment with FastAPI",
                "Next.js for ML application frontends"
            ]
        },
        "maintenance": {
            "path": "Backend Development",
            "tech_focus": [
                "Python backend development",
                "API design with FastAPI/Django",
                "Basic frontend with Next.js"
            ]
        }
    }

    skill_keywords = {
        "programming": "software",
        "database": "data",
        "network": "communications",
        "security": "cyber",
        "analysis": "intelligence"
    }

    if category.lower() in tech_paths:
        return tech_paths[category.lower()]

    for skill in skills:
        skill_lower = skill.lower()
        for keyword, category in skill_keywords.items():
            if keyword in skill_lower and category in tech_paths:
                return tech_paths[category]

    return default_path


def translate_military_code(code: str, job_codes: dict) -> dict:
    """Translate military code to VWC development path."""
    code = code.upper().strip()
    prefixes = ["MOS", "AFSC", "RATE"]
    for prefix in prefixes:
        if code.startswith(prefix):
            code = code.replace(prefix, "").strip()

    possible_codes = [f"MOS_{code}", f"AFSC_{code}", f"RATE_{code}"]

    for possible_code in possible_codes:
        if possible_code in job_codes:
            job_data = job_codes[possible_code]
            return {
                "found": True,
                "data": {
                    "title": job_data.get('title', 'Military Professional'),
                    "branch": job_data.get('branch', 'Military'),
                    "dev_path": job_data.get('vwc_path', 'Full Stack Development'),
                    "tech_focus": job_data.get('tech_focus', []),
                    "skills": job_data.get('skills', [])
                }
            }

    return {
        "found": False,
        "data": {
            "title": "Military Professional",
            "branch": "Military",
            "dev_path": "Full Stack Development",
            "tech_focus": [
                "Start with JavaScript/TypeScript fundamentals",
                "Build projects with Next.js and Tailwind",
                "Learn Python backend development with FastAPI"
            ],
            "skills": [
                "Leadership and team coordination",
                "Problem-solving and adaptation",
                "Project planning and execution"
            ]
        }
    }


def get_chat_response(messages: List[Dict]) -> str:
    """Get response from OpenAI chat completion."""
    try:
        response = client.chat.completions.create(
            model="gpt-4",
            messages=messages,
            temperature=0.7,
        )
        return response.choices[0].message.content
    except Exception as e:
        logger.error(f"OpenAI API error: {e}")
        raise


def export_chat_history(chat_history: List[Dict]) -> str:
    """Export chat history to JSON."""
    export_data = {
        "timestamp": datetime.now().isoformat(),
        "messages": chat_history
    }
    return json.dumps(export_data, indent=2)


def save_feedback(feedback: Dict):
    """Save user feedback to file."""
    feedback_dir = "feedback"
    os.makedirs(feedback_dir, exist_ok=True)
    feedback_file = os.path.join(
        feedback_dir,
        f"feedback_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
    )
    with open(feedback_file, 'w') as f:
        json.dump(feedback, f, indent=2)


def handle_command(command: str) -> str:
    """Handle special commands including MOS translation."""
    parts = command.lower().split()
    if not parts:
        return None

    cmd = parts[0]
    if cmd in ['/mos', '/afsc', '/rate']:
        if len(parts) < 2:
            return "Please provide a military job code. Example: `/mos 25B`"

        code = parts[1]
        translation = translate_military_code(code, st.session_state.job_codes)
        if translation['found']:
            return (
                    f"🎖️ **{translation['data']['title']}** ({translation['data']['branch']})\n\n"
                    f"💻 **VWC Development Path**: {translation['data']['dev_path']}\n\n"
                    "🔧 **Military Skills**:\n" +
                    "\n".join(f"- {skill}" for skill in translation['data']['skills']) +
                    "\n\n📚 **VWC Tech Focus**:\n" +
                    "\n".join(f"{i + 1}. {focus}" for i, focus in enumerate(translation['data']['tech_focus']))
            )
        else:
            return (
                    "I don't have that specific code in my database, but here's a recommended "
                    "VWC learning path based on general military experience:\n\n" +
                    "\n".join(f"{i + 1}. {focus}" for i, focus in enumerate(translation['data']['tech_focus']))
            )

    return None


def initialize_chat():
    """Initialize the chat with a VWC-focused welcome message."""
    welcome_message = {
        "role": "assistant",
        "content": (
            "Welcome to VetsAI - Your Vets Who Code Assistant! 👨‍💻\n\n"
            "I'm here to help you with:\n\n"
            "🔹 VWC Tech Stack:\n"
            "- JavaScript/TypeScript\n"
            "- Python (FastAPI, Streamlit)\n"
            "- Next.js & Tailwind CSS\n"
            "- AI/ML Integration\n\n"
            "🔹 Commands:\n"
            "- `/mos [code]` - Translate your MOS to dev path\n"
            "- `/afsc [code]` - Translate your AFSC to dev path\n"
            "- `/rate [code]` - Translate your Rate to dev path\n"
            "- `/frontend` - Help with JS/TS/Next.js\n"
            "- `/backend` - Help with Python frameworks\n"
            "- `/ai` - AI/ML guidance\n\n"
            "Let's start by checking how your military experience "
            "aligns with software development! Share your MOS/AFSC/Rate, "
            "or ask about any part of our tech stack."
        )
    }
    return [welcome_message]


def main():
    """Main application function."""
    st.title("🇺🇸 VetsAI: Vets Who Code Assistant")

    # Initialize session
    if 'session_id' not in st.session_state:
        st.session_state.session_id = hashlib.md5(str(time.time()).encode()).hexdigest()

    if 'job_codes' not in st.session_state:
        try:
            st.session_state.job_codes = load_military_job_codes()
        except Exception as e:
            logger.error(f"Error loading job codes: {e}")
            st.session_state.job_codes = {}

    if 'messages' not in st.session_state:
        st.session_state.messages = initialize_chat()

    # Sidebar with VWC tech stack resources
    with st.sidebar:
        st.markdown("""
### VWC Tech Stack

#### 🌐 Frontend
- **JavaScript/TypeScript**: For building scalable, maintainable codebases.
- **CSS & Tailwind CSS**: For styling and responsive design.
- **Next.js**: React framework for server-side rendering and static web applications.

#### ⚙️ Backend
- **Python**: Core language for backend development and data processing.
- **FastAPI**: For building fast, asynchronous APIs with Python.
- **Streamlit**: Framework for creating data-driven apps and interactive visualizations in Python.

#### 🤖 AI/ML Integration
- **Machine Learning (ML)**: Tools and frameworks for model development.
- **AI Applications**: Integrating AI models and APIs in applications, including OpenAI GPT models.

#### 📊 Data & Visualization
- **Pandas**: Data manipulation and analysis.
- **Matplotlib & Plotly**: Data visualization libraries for generating plots and charts.

#### 🛠️ DevOps and Tooling
- **Git & GitHub**: Version control and collaborative code management.
- **Docker**: Containerization for consistent development and deployment.
- **VS Code**: Development environment setup with extensions for Python, JavaScript, etc.

#### 🧪 Testing
- **Jest**: JavaScript testing framework for frontend applications.
- **Pytest**: Testing framework for Python applications, with extensive plugin support.
- **Coverage.py**: For measuring code coverage in Python projects.

""", unsafe_allow_html=True)

    # Display chat messages
    for message in st.session_state.messages:
        with st.chat_message(message["role"]):
            st.markdown(message["content"])

    # Chat input
    if prompt := st.chat_input():
        st.session_state.messages.append({"role": "user", "content": prompt})
        with st.chat_message("user"):
            st.markdown(prompt)

        # Check for commands
        if prompt.startswith('/'):
            command_response = handle_command(prompt)
            if command_response:
                with st.chat_message("assistant"):
                    st.markdown(command_response)
                st.session_state.messages.append({
                    "role": "assistant",
                    "content": command_response
                })
                return

        # Generate and display assistant response
        with st.chat_message("assistant"):
            try:
                messages = st.session_state.messages.copy()
                messages.insert(0, {
                    "role": "system",
                    "content": (
                        "You are a specialized AI assistant for Vets Who Code members, designed to provide clear, practical technical guidance "
                        "to veterans transitioning into software development careers.\n\n"

                        "CORE TECH STACK:\n"
                        "- Frontend: JavaScript, TypeScript, React, Next.js\n"
                        "- Styling: CSS, Tailwind CSS\n"
                        "- Backend: Python, FastAPI\n"
                        "- Data & Visualization: Streamlit\n"
                        "- Advanced: AI/ML fundamentals\n"
                        "- Development Tools: Git, GitHub, VS Code\n"
                        "- Testing: Jest, Pytest\n\n"

                        "CAREER TRANSITION GUIDANCE:\n"
                        "1. Resume Development:\n"
                        "   - Technical Skills: Programming Languages, Frameworks, Tools, Cloud, Testing\n"
                        "   - Military Experience Translation: Leadership, Problem-solving, Team Collaboration\n\n"

                        "2. Portfolio Development:\n"
                        "   - Clean code and documentation\n"
                        "   - Version control and API integration\n"
                        "   - Responsive design and performance\n"
                        "   - Testing and TypeScript implementation\n"
                        "   - Security and accessibility standards\n\n"

                        "LEARNING PATHS:\n"
                        "1. Fundamentals: HTML, CSS, JavaScript, Git\n"
                        "2. Intermediate: TypeScript, React, Python\n"
                        "3. Advanced: Next.js, FastAPI, Streamlit, AI/ML\n\n"

                        "PROJECT FOCUS:\n"
                        "1. Portfolio Projects: Personal website, APIs, Data visualization\n"
                        "2. Technical Skills: Code quality, Testing, Security, Performance\n"
                        "3. Career Materials: GitHub profile, Technical blog, Documentation\n\n"

                        "Remember: Provide practical guidance for building technical skills and transitioning to software development careers. "
                        "Focus on concrete examples and best practices."
                    )
                })
                with st.spinner("Building a response..."):
                    response = get_chat_response(messages)
                    st.markdown(response)
                    st.session_state.messages.append({
                        "role": "assistant",
                        "content": response
                    })
            except Exception as e:
                st.error(f"Error generating response: {str(e)}")

    # Export chat history
    if st.button("Export Chat History"):
        chat_export = export_chat_history(st.session_state.messages)
        st.download_button("Download Chat History", chat_export, "vetsai_chat_history.json", "application/json")

    # Feedback mechanism
    with st.expander("Provide Feedback"):
        feedback_rating = st.slider("Rate your experience (1-5)", min_value=1, max_value=5, value=5)
        feedback_text = st.text_area("Additional feedback")

        if st.button("Submit Feedback"):
            feedback = {
                "timestamp": datetime.now().isoformat(),
                "session_id": st.session_state.session_id,
                "rating": feedback_rating,
                "feedback": feedback_text
            }
            save_feedback(feedback)
            st.success("Thank you for your feedback!")


if __name__ == "__main__":
    main()