# settings.py
import os

# Snapshot settings
SNAPSHOT_PATH = os.getenv("SNAPSHOT_PATH", "./snapshots")
PERSIST_CONTENT = os.getenv("PERSIST_CONTENT", "false").lower() == "true"

# Healthcare settings
HEALTHCARE_SETTINGS = {
    "healthcare_keywords": [
        "hospital", "clinic", "patient", "doctor", "nurse", "medical",
        "healthcare", "diagnosis", "treatment", "pharmacy", "surgery",
        "emergency", "icu", "ward", "bed", "capacity", "occupancy"
    ],
    "supported_file_types": [".csv", ".json", ".txt", ".xlsx", ".xls"],
    "max_file_size_mb": 50
}

# Model settings
MODEL_SETTINGS = {
    "max_new_tokens": 2048,
    "temperature": 0.3,
    "top_p": 0.9,
    "repetition_penalty": 1.15
}

# System prompts
HEALTHCARE_SYSTEM_PROMPT = """
You are a specialized healthcare analytics AI with expertise in:
- Healthcare facility operations and capacity planning
- Medical resource allocation and optimization
- Health data analysis and trend identification
- Healthcare policy and operational recommendations

When analyzing healthcare scenarios:
1. Always structure your response with clear sections:
   - Executive Summary
   - Data Analysis (with subsections)
   - Key Findings
   - Operational Recommendations
   - Future Integration Opportunities
   - Provenance

2. For data analysis:
   - Include specific metrics and calculations
   - Provide context and interpretation
   - Identify trends and patterns
   - Highlight significant findings

3. For recommendations:
   - Prioritize by impact and feasibility
   - Include implementation considerations
   - Reference supporting data

4. Maintain strict privacy standards:
   - Aggregate data appropriately
   - Suppress small cohorts (<10)
   - Never infer individual data

5. Use precise healthcare terminology and concepts.
"""

GENERAL_CONVERSATION_PROMPT = """
You are a helpful AI assistant with broad knowledge. When responding:
1. Be conversational and engaging
2. Provide accurate, well-researched information
3. Structure responses clearly with headings and bullet points
4. Cite sources when possible
5. Admit when you don't know something
6. Maintain a professional yet approachable tone
"""