# --- PYTHON 3.13  COMPATIBILITY MONKEY-PATCH ---
import sys
import types

if 'audioop' not in sys.modules:
    mock_audioop = types.ModuleType('audioop')
    mock_audioop.error = Exception
    mock_audioop.getsample = lambda data, width, index: 0 
    sys.modules['audioop'] = mock_audioop
# ----------------------------------------------

import os
import sqlite3
import json
import requests
import pandas as pd
from datetime import datetime
from flask import Flask, request, jsonify, render_template_string, redirect, session, url_for
from groq import Groq 

app = Flask(__name__)
app.secret_key = "strides_pharma_domain_secret_encryption_key_2026"

app.config.update(
    SESSION_COOKIE_SECURE=True,
    SESSION_COOKIE_SAMESITE='None',
)

DATA_DIR = "/data" if os.path.exists("/data") else "."
DB_NAME = os.path.join(DATA_DIR, "strides_architecture.db")

def init_db():
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS project_tasks (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
            phase TEXT, task_name TEXT, owner TEXT, timeline TEXT, priority TEXT
        )
    ''')
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS chat_sessions (
            session_id TEXT PRIMARY KEY, user_owner TEXT, title TEXT, updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
        )
    ''')
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS chat_history (
            id INTEGER PRIMARY KEY AUTOINCREMENT, session_id TEXT, role TEXT, content TEXT, timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
        )
    ''')
    cursor.execute("SELECT COUNT(*) FROM project_tasks")
    if cursor.fetchone()[0] == 0:
        cursor.executemany('INSERT INTO project_tasks (phase, task_name, owner, timeline, priority) VALUES (?,?,?,?,?)', [
            ('BRD-Scope', 'Automated Audit Trail Log Consolidation', 'A. Sharma', 'Q3-2026', 'Critical'),
            ('GxP-Compliance', '21 CFR Part 11 Electronic Signature Verification', 'R. Verma', 'Q3-2026', 'Critical')
        ])
    conn.commit()
    conn.close()

init_db()

USER_CREDENTIALS = {"admin": "strides", "compliance_tester": "AuditPass123"}
GROQ_MODELS = ["llama-3.1-8b-instant", "llama-3.1-70b-versatile"]
OPENROUTER_MODELS = ["nvidia/nemotron-3-ultra-550b-a55b:free"]

def get_current_user():
    return session.get("username")

@app.route("/login", methods=["POST"])
def login():
    username = request.form.get("username")
    password = request.form.get("password")
    if USER_CREDENTIALS.get(username) == password:
        session["username"] = username
        return redirect(url_for("index"))
    return render_template_string(HTML_TEMPLATE, error="Invalid credentials profile verification fail.", show_login=True)

@app.route("/logout")
def logout():
    session.clear()
    return redirect(url_for("index"))

@app.route("/api/sessions", methods=["GET"])
def api_get_sessions():
    user = get_current_user()
    if not user: return jsonify({"error": "Unauthorized"}), 401
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute("SELECT session_id, title FROM chat_sessions WHERE user_owner = ? ORDER BY updated_at DESC", (user,))
    rows = cursor.fetchall()
    conn.close()
    if not rows:
        sid = f"session_{user}_initial"
        conn = sqlite3.connect(DB_NAME)
        c = conn.cursor()
        c.execute("INSERT OR IGNORE INTO chat_sessions (session_id, user_owner, title) VALUES (?,?,?)", (sid, user, "Initial Session Thread"))
        conn.commit()
        conn.close()
        rows = [(sid, "Initial Session Thread")]
    return jsonify([{"id": r[0], "title": r[1]} for r in rows])

@app.route("/api/sessions/new", methods=["POST"])
def api_new_session():
    user = get_current_user()
    if not user: return jsonify({"error": "Unauthorized"}), 401
    new_id = f"session_{user}_{int(datetime.now().timestamp())}"
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute("INSERT INTO chat_sessions (session_id, user_owner, title) VALUES (?, ?, ?)", (new_id, user, "New Chat Session"))
    conn.commit()
    conn.close()
    return jsonify({"status": "success", "id": new_id})

@app.route("/api/history/<session_id>", methods=["GET"])
def api_get_history(session_id):
    if not get_current_user(): return jsonify({"error": "Unauthorized"}), 401
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute("SELECT role, content FROM chat_history WHERE session_id = ? ORDER BY timestamp ASC", (session_id,))
    rows = cursor.fetchall()
    conn.close()
    return jsonify([{"role": r[0], "content": r[1]} for r in rows])

@app.route("/api/chat", methods=["POST"])
def api_chat():
    user = get_current_user()
    if not user: return jsonify({"error": "Unauthorized"}), 401
    data = request.json or {}
    sid = data.get("session_id")
    msg = data.get("message", "").strip()
    provider = data.get("provider")
    api_key = data.get("api_key")
    model = data.get("model")

    if not msg or not api_key:
        return jsonify({"error": "Missing message or API key"}), 400

    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute("INSERT INTO chat_history (session_id, role, content) VALUES (?, 'user', ?)", (sid, msg))
    
    cursor.execute("SELECT COUNT(*) FROM chat_history WHERE session_id = ?", (sid,))
    if cursor.fetchone()[0] <= 2:
        cursor.execute("UPDATE chat_sessions SET title = ? WHERE session_id = ?", (msg[:25] + "...", sid))
    conn.commit()

    cursor.execute("SELECT role, content FROM chat_history WHERE session_id = ? ORDER BY timestamp ASC", (sid,))
    history = [{"role": r[0], "content": r[1]} for r in cursor.fetchall()]
    conn.close()

    sys_prompt = "You are a Pharma Business Analyst for Strides. Output specifications and save entries if data schema formats matches."
    messages = [{"role": "system", "content": sys_prompt}] + history

    try:
        if provider == "Groq":
            client = Groq(api_key=api_key)
            resp = client.chat.completions.create(model=model, messages=messages, temperature=0.0)
            ai_out = resp.choices[0].message.content.strip()
        else:
            headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
            payload = {"model": model, "messages": messages, "temperature": 0.0}
            r = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload)
            ai_out = r.json()['choices'][0]['message']['content'].strip()

        if "||JSON_DATA:" in ai_out:
            try:
                payload_str = ai_out.split("||JSON_DATA:")[1].split("||")[0].strip()
                p = json.loads(payload_str)
                conn = sqlite3.connect(DB_NAME)
                conn.cursor().execute("INSERT INTO project_tasks (phase, task_name, owner, timeline, priority) VALUES (?,?,?,?,?)",
                             (p.get('phase'), p.get('task_name'), p.get('owner'), p.get('timeline'), p.get('priority')))
                conn.commit()
                conn.close()
                ai_out += "\n\n⚙️ [SYSTEM LOGGED SUCCESS]"
            except: pass

        conn = sqlite3.connect(DB_NAME)
        conn.cursor().execute("INSERT INTO chat_history (session_id, role, content) VALUES (?, 'assistant', ?)", (sid, ai_out))
        conn.commit()
        conn.close()
        return jsonify({"reply": ai_out})
    except Exception as e:
        return jsonify({"reply": f"❌ Gateway Execution Error: {str(e)}"})

@app.route("/api/query", methods=["POST"])
def api_query():
    if not get_current_user(): return jsonify({"error": "Unauthorized"}), 401
    data = request.json or {}
    q = data.get("query", "")
    api_key = data.get("api_key")
    provider = data.get("provider")
    model = data.get("model")
    
    sys_prompt = "You are a SQLite assistant. Return ONLY a valid executable raw SQLite query targeting table 'project_tasks'."
    try:
        if provider == "Groq":
            client = Groq(api_key=api_key)
            resp = client.chat.completions.create(model=model, messages=[{"role":"system","content":sys_prompt},{"role":"user","content":q}], temperature=0.0)
            sql = resp.choices[0].message.content.strip()
        else:
            r = requests.post("https://openrouter.ai/api/v1/chat/completions", headers={"Authorization": f"Bearer {api_key}"}, json={"model": model, "messages": [{"role":"system","content":sys_prompt},{"role":"user","content":q}]})
            sql = r.json()['choices'][0]['message']['content'].strip()
        
        sql = sql.replace("```sql", "").replace("```", "").replace("`", "").strip()
        conn = sqlite3.connect(DB_NAME)
        box_df = pd.read_sql_query(sql, conn)
        conn.close()
        return jsonify({"sql": sql, "data": box_df.to_dict(orient="records")})
    except Exception as e:
        return jsonify({"error": str(e)})

HTML_TEMPLATE = """
<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Strides Pharma AI</title>
    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/css/bootstrap.min.css">
    <style>
        body { background: #f4f6f9; height: 100vh; display: flex; flex-direction: column; margin: 0; }
        .login-box { max-width: 400px; margin: 100px auto; background: white; padding: 30px; border-radius: 8px; box-shadow: 0 4px 15px rgba(0,0,0,0.1); }
        .app-container { display: flex; flex: 1; overflow: hidden; height: 100vh; }
        .sidebar { width: 300px; background: #2c3e50; color: white; padding: 20px; display: flex; flex-direction: column; justify-content: space-between; }
        .main-content { flex: 1; padding: 20px; display: flex; flex-direction: column; overflow-y: auto; }
        .chat-box { flex: 1; border: 1px solid #ddd; background: white; border-radius: 8px; padding: 15px; overflow-y: auto; margin-bottom: 15px; min-height: 350px; max-height: 500px;}
        .msg { margin-bottom: 12px; padding: 10px; border-radius: 6px; }
        .msg.user { background: #e1f5fe; text-align: right; margin-left: 50px; }
        .msg.assistant { background: #f1f8e9; text-align: left; margin-right: 50px; }
    </style>
</head>
<body>

{% if show_login %}
<div class="container">
    <div class="login-box">
        <h4 class="text-center mb-4">💊 Strides Security Gate</h4>
        <form action="/login" method="POST">
            <div class="mb-3"><label>Username</label><input name="username" class="form-control" required autocomplete="off"></div>
            <div class="mb-3"><label>Password</label><input name="password" type="password" class="form-control" required></div>
            <button type="submit" class="btn btn-primary w-100">Secure Access Login</button>
        </form>
        {% if error %}<div class="text-danger mt-2 text-center">{{ error }}</div>{% endif %}
    </div>
</div>
{% else %}
<div class="app-container">
    <div class="sidebar">
        <div>
            <h5>🕒 Session Explorer</h5>
            <select id="sessionSelect" class="form-select mb-3" onchange="loadHistory()"></select>
            <button onclick="newSession()" class="btn btn-outline-light btn-sm w-100 mb-2">➕ New Chat Thread</button>
            <hr>
            <h5>🔑 Gateway Options</h5>
            <label class="small">API Provider</label>
            <select id="provider" class="form-select mb-2" onchange="updateModels()">
                <option value="Groq">Groq</option>
                <option value="OpenRouter">OpenRouter</option>
            </select>
            <label class="small">API Key</label>
            <input id="apiKey" type="password" class="form-control mb-2" placeholder="sk-...">
            <label class="small">Target Model</label>
            <select id="model" class="form-select"></select>
        </div>
        <div>
            <span class="d-block mb-2 small text-light">Logged in as: <strong>{{ username }}</strong></span>
            <a href="/logout" class="btn btn-danger btn-sm w-100">Log Out</a>
        </div>
    </div>
    
    <div class="main-content">
        <ul class="nav nav-tabs mb-3">
            <li class="nav-item"><a class="nav-link active" data-bs-toggle="tab" href="#chatTab">🤖 Requirement Chat Workspace</a></li>
            <li class="nav-item"><a class="nav-link" data-bs-toggle="tab" href="#queryTab">🔎 Inventory SQL Explorer</a></li>
        </ul>
        
        <div class="tab-content flex-grow-1">
            <div id="chatTab" class="tab-pane fade show active">
                <div id="chatBox" class="chat-box"></div>
                <div class="input-group">
                    <input id="chatInput" class="form-control" placeholder="Type data log requests here...">
                    <button onclick="sendChat()" class="btn btn-primary">Send</button>
                </div>
            </div>
            
            <div id="queryTab" class="tab-pane fade">
                <div class="input-group mb-3">
                    <input id="queryInput" class="form-control" placeholder="Search inventory records using natural language...">
                    <button onclick="runQuery()" class="btn btn-secondary">Run Subsystem Evaluation</button>
                </div>
                <div id="sqlStatus" class="fw-bold text-success mb-2"></div>
                <div class="table-responsive"><table id="dataTable" class="table table-striped table-bordered"></table></div>
            </div>
        </div>
    </div>
</div>

<script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/js/bootstrap.bundle.min.js"></script>
<script>
    const GROQ_MODELS = ["llama-3.1-8b-instant", "llama-3.1-70b-versatile"];
    const OR_MODELS = ["nvidia/nemotron-3-ultra-550b-a55b:free"];

    function init() {
        updateModels();
        loadSessions();
    }

    function updateModels() {
        const p = document.getElementById("provider").value;
        const mSelect = document.getElementById("model");
        if(!mSelect) return;
        mSelect.innerHTML = "";
        const list = (p === "Groq") ? GROQ_MODELS : OR_MODELS;
        list.forEach(m => { mSelect.innerHTML += `<option value="${m}">${m}</option>`; });
    }

    async function loadSessions() {
        const select = document.getElementById("sessionSelect");
        if(!select) return;
        const r = await fetch("/api/sessions");
        if (!r.ok) return;
        const data = await r.json();
        const currentSelected = select.value;
        select.innerHTML = "";
        data.forEach(s => { select.innerHTML += `<option value="${s.id}">${s.title}</option>`; });
        if(currentSelected && data.find(s => s.id === currentSelected)) {
            select.value = currentSelected;
        } else if (data.length > 0) {
            select.value = data[0].id;
        }
        loadHistory();
    }

    async function newSession() {
        const r = await fetch("/api/sessions/new", {method:"POST"});
        if (r.ok) { 
            const res = await r.json();
            await loadSessions();
            document.getElementById("sessionSelect").value = res.id;
            loadHistory();
        }
    }

    async function loadHistory() {
        const select = document.getElementById("sessionSelect");
        if(!select) return;
        const sid = select.value;
        if (!sid) return;
        const r = await fetch(`/api/history/${sid}`);
        const data = await r.json();
        const box = document.getElementById("chatBox");
        box.innerHTML = "";
        data.forEach(m => { box.innerHTML += `<div class="msg ${m.role}"><strong>${m.role}:</strong> ${m.content}</div>`; });
        box.scrollTop = box.scrollHeight;
    }

    async function sendChat() {
        const inp = document.getElementById("chatInput");
        const txt = inp.value.trim();
        if(!txt) return;
        const sid = document.getElementById("sessionSelect").value;
        const box = document.getElementById("chatBox");
        
        box.innerHTML += `<div class="msg user"><strong>user:</strong> ${txt}</div>`;
        inp.value = "";
        box.scrollTop = box.scrollHeight;

        const payload = {
            session_id: sid, message: txt,
            provider: document.getElementById("provider").value,
            api_key: document.getElementById("apiKey").value,
            model: document.getElementById("model").value
        };

        const r = await fetch("/api/chat", {method:"POST", headers:{"Content-Type":"application/json"}, body:JSON.stringify(payload)});
        const res = await r.json();
        box.innerHTML += `<div class="msg assistant"><strong>assistant:</strong> ${res.reply}</div>`;
        box.scrollTop = box.scrollHeight;
        
        const select = document.getElementById("sessionSelect");
        const activeOpt = select.options[select.selectedIndex];
        if (activeOpt && activeOpt.innerText.includes("New Chat Session")) {
             loadSessions();
        }
    }

    async function runQuery() {
        const inp = document.getElementById("queryInput").value;
        const payload = {
            query: inp,
            api_key: document.getElementById("apiKey").value,
            provider: document.getElementById("provider").value,
            model: document.getElementById("model").value
        };
        const r = await fetch("/api/query", {method:"POST", headers:{"Content-Type":"application/json"}, body:JSON.stringify(payload)});
        const res = await r.json();
        if(res.error) {
            document.getElementById("sqlStatus").innerText = `Error: ${res.error}`;
            return;
        }
        document.getElementById("sqlStatus").innerText = `Executed Query: ${res.sql}`;
        const table = document.getElementById("dataTable");
        table.innerHTML = "";
        if(res.data.length > 0) {
            let headers = Object.keys(res.data[0]);
            let tr = "<tr>";
            headers.forEach(h => tr += `<th>${h}</th>`);
            tr += "</tr>";
            res.data.forEach(row => {
                tr += "<tr>";
                headers.forEach(h => tr += `<td>${row[h]}</td>`);
                tr += "</tr>";
            });
            table.innerHTML = tr;
        } else {
            table.innerHTML = "<tr><td>No records matched or returned.</td></tr>";
        }
    }

    window.onload = init;
</script>
{% endif %}
</body>
</html>
"""

@app.route("/")
def index():
    user = get_current_user()
    if not user:
        return render_template_string(HTML_TEMPLATE, show_login=True)
    return render_template_string(HTML_TEMPLATE, show_login=False, username=user)

# Ensure the bottom of app.py looks exactly like this:
if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)