app.py

import os
import sys
import threading
from flask import Flask, request, Response, stream_with_context, jsonify, render_template_string
from flask_cors import CORS
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer, BitsAndBytesConfig
import torch

print("LOG: Iniciando Motor Hydra 3.0 (Docker API - Hydra Software)...")
sys.stdout.flush()

app = Flask(__name__)
CORS(app)

model_id = "Mution/Hydra-AI-3B-Coder-Pro"
token = os.getenv("HF_TOKEN")

VALID_API_KEYS = {
    "hydra-sk-master-9928374": "Master_Admin",
    "hydra-sk-root-1122334": "Root_System",
    "hydra-sk-dev-5566778": "Developer_Terminal",
    "hydra-sk-bot-alpha-1029384": "Discord_Bot_Alpha",
    "hydra-sk-bot-beta-2039485": "Discord_Bot_Beta",
    "hydra-sk-bot-gamma-3049586": "Telegram_Bot",
    "hydra-sk-bot-delta-4059687": "Twitch_Bot",
    "hydra-sk-panel-main-5647382": "Web_Panel_Main",
    "hydra-sk-panel-sec-6758493": "Web_Panel_Secundario",
    "hydra-sk-panel-event-7869504": "Event_Dashboard",
    "hydra-sk-vip-001-8970615": "VIP_User_1",
    "hydra-sk-vip-002-9081726": "VIP_User_2",
    "hydra-sk-vip-003-0192837": "VIP_User_3",
    "hydra-sk-test-1203948": "Testing_Key_A",
    "hydra-sk-test-2314059": "Testing_Key_B",
    "hydra-sk-node-001-110022": "Node_Server_1",
    "hydra-sk-node-002-334455": "Node_Server_2",
    "hydra-sk-node-003-667788": "Node_Server_3",
    "hydra-sk-node-004-990011": "Node_Server_4",
    "hydra-sk-node-005-223344": "Node_Server_5",
    "hydra-sk-app-mobile-1010": "Mobile_App_iOS",
    "hydra-sk-app-mobile-2020": "Mobile_App_Android",
    "hydra-sk-app-desktop-3030": "Desktop_App_Electron",
    "hydra-sk-app-web-4040": "Web_Client_App",
    "hydra-sk-service-auth-01": "Auth_Service",
    "hydra-sk-service-db-02": "Database_Worker",
    "hydra-sk-service-cache-03": "Redis_Cache_Worker",
    "hydra-sk-service-queue-04": "RabbitMQ_Worker",
    "hydra-sk-service-cron-05": "Cron_Task_Manager",
    "hydra-sk-client-pro-1001": "Client_Pro_Tier_1",
    "hydra-sk-client-pro-1002": "Client_Pro_Tier_2",
    "hydra-sk-client-pro-1003": "Client_Pro_Tier_3",
    "hydra-sk-client-pro-1004": "Client_Pro_Tier_4",
    "hydra-sk-client-pro-1005": "Client_Pro_Tier_5",
    "hydra-sk-client-enterprise-2001": "Enterprise_Client_A",
    "hydra-sk-client-enterprise-2002": "Enterprise_Client_B",
    "hydra-sk-client-enterprise-2003": "Enterprise_Client_C",
    "hydra-sk-internal-tool-01": "Internal_Dev_Tool_1",
    "hydra-sk-internal-tool-02": "Internal_Dev_Tool_2",
    "hydra-sk-internal-tool-03": "Internal_Dev_Tool_3",
    "hydra-sk-bot-v2-01": "Discord_Bot_v2_1",
    "hydra-sk-bot-v2-02": "Discord_Bot_v2_2",
    "hydra-sk-bot-v2-03": "Discord_Bot_v2_3",
    "hydra-sk-bot-v2-04": "Discord_Bot_v2_4",
    "hydra-sk-bot-v2-05": "Discord_Bot_v2_5",
    "hydra-sk-integration-gh-01": "GitHub_Actions_Integration",
    "hydra-sk-integration-gl-02": "GitLab_CI_Integration",
    "hydra-sk-integration-docker-03": "Docker_Hub_Integration",
    "hydra-sk-integration-aws-04": "AWS_Lambda_Integration",
    "hydra-sk-integration-gcf-05": "GCP_Cloud_Functions",
    "hydra-sk-vip-004-1234567": "VIP_User_4",
    "hydra-sk-vip-005-2345678": "VIP_User_5",
    "hydra-sk-vip-006-3456789": "VIP_User_6",
    "hydra-sk-vip-007-4567890": "VIP_User_7",
    "hydra-sk-vip-008-5678901": "VIP_User_8",
    "hydra-sk-vip-009-6789012": "VIP_User_9",
    "hydra-sk-vip-010-7890123": "VIP_User_10",
    "hydra-sk-sec-audit-01": "Security_Audit_Tool_A",
    "hydra-sk-sec-audit-02": "Security_Audit_Tool_B",
    "hydra-sk-monitoring-01": "Prometheus_Monitoring",
    "hydra-sk-monitoring-02": "Grafana_Dashboard_Agent",
    "hydra-sk-ext-partner-01": "External_Partner_Alpha",
    "hydra-sk-ext-partner-02": "External_Partner_Beta",
    "hydra-sk-ext-partner-03": "External_Partner_Gamma",
    "hydra-sk-sandbox-01": "Sandbox_Environment_1",
    "hydra-sk-sandbox-02": "Sandbox_Environment_2",
    "hydra-sk-sandbox-03": "Sandbox_Environment_3",
    "hydra-sk-cluster-east-01": "Cluster_US_East_1",
    "hydra-sk-cluster-east-02": "Cluster_US_East_2",
    "hydra-sk-cluster-west-01": "Cluster_US_West_1",
    "hydra-sk-cluster-eu-01": "Cluster_EU_Central_1",
    "hydra-sk-cluster-sa-01": "Cluster_SA_East_1",
    "hydra-sk-backup-01": "Backup_Worker_Node_1",
    "hydra-sk-backup-02": "Backup_Worker_Node_2",
    "hydra-sk-legacy-01": "Legacy_System_Adapter_1",
    "hydra-sk-legacy-02": "Legacy_System_Adapter_2",
    "hydra-sk-agent-ai-01": "AI_Agent_Code_Reviewer",
    "hydra-sk-agent-ai-02": "AI_Agent_Bug_Fixer",
    "hydra-sk-agent-ai-03": "AI_Agent_Refactorer",
    "hydra-sk-agent-ai-04": "AI_Agent_Doc_Generator",
    "hydra-sk-agent-ai-05": "AI_Agent_Unit_Tester",
    "hydra-sk-gateway-01": "API_Gateway_Primary",
    "hydra-sk-gateway-02": "API_Gateway_Secondary",
    "hydra-sk-gateway-03": "API_Gateway_Failover",
    "hydra-sk-custom-9001": "Custom_Deployment_Key_1",
    "hydra-sk-custom-9002": "Custom_Deployment_Key_2",
    "hydra-sk-custom-9003": "Custom_Deployment_Key_3",
    "hydra-sk-custom-9004": "Custom_Deployment_Key_4",
    "hydra-sk-custom-9005": "Custom_Deployment_Key_5",
    "hydra-sk-custom-9006": "Custom_Deployment_Key_6",
    "hydra-sk-custom-9007": "Custom_Deployment_Key_7",
    "hydra-sk-custom-9008": "Custom_Deployment_Key_8",
    "hydra-sk-custom-9009": "Custom_Deployment_Key_9",
    "hydra-sk-custom-9010": "Custom_Deployment_Key_10"
}

try:
    print("LOG: Cargando Tokenizador Hydra 3B...")
    tokenizer = AutoTokenizer.from_pretrained(model_id, token=token)

    print("LOG: Configurando cuantización 4-bit...")
    bnb_config = BitsAndBytesConfig(
        load_in_4bit=True,
        bnb_4bit_compute_dtype=torch.float16,
        llm_int8_enable_fp32_cpu_offload=True
    )

    print("LOG: Cargando modelo fusionado Hydra 3B...")
    model = AutoModelForCausalLM.from_pretrained(
        model_id,
        token=token,
        quantization_config=bnb_config,
        device_map="auto",
        torch_dtype=torch.float16,
        low_cpu_mem_usage=True,
        offload_folder="offload_dir"
    )

    print("LOG: IA cargada y lista.")
except Exception as e:
    print(f"LOG CRÍTICO: Error al cargar el modelo: {e}")
    sys.exit(1)

HTML_PAGE = """
<!DOCTYPE html>
<html lang="es">
<head>
  <meta charset="UTF-8">
  <meta name="viewport" content="width=device-width, initial-scale=1.0">
  <title>Hydra AI 3B - Private Console</title>
  <link rel="preconnect" href="https://fonts.googleapis.com">
  <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
  <link href="https://fonts.googleapis.com/css2?family=Fira+Code:wght@400;600&family=Inter:wght@400;600;700&display=swap" rel="stylesheet">
  <style>
    :root {
      --bg-main: #0b0f19;
      --bg-card: #111827;
      --bg-chat: #070a12;
      --border-color: #1f293d;
      --accent-cyan: #06b6d4;
      --accent-emerald: #10b981;
      --text-main: #f3f4f6;
      --text-muted: #9ca3af;
    }
    * { box-sizing: border-box; margin: 0; padding: 0; }
    body {
      font-family: 'Inter', sans-serif;
      background-color: var(--bg-main);
      color: var(--text-main);
      display: flex;
      justify-content: center;
      align-items: center;
      min-height: 100vh;
      padding: 20px;
    }
    .wrapper {
      width: 100%;
      max-width: 1100px;
      height: 90vh;
      background: var(--bg-card);
      border: 1px solid var(--border-color);
      border-radius: 16px;
      display: flex;
      flex-direction: column;
      overflow: hidden;
      box-shadow: 0 20px 50px rgba(0,0,0,0.6);
    }
    .top-bar {
      padding: 16px 24px;
      background: #162032;
      border-bottom: 1px solid var(--border-color);
      display: flex;
      justify-content: space-between;
      align-items: center;
      flex-wrap: wrap;
      gap: 12px;
    }
    .brand-title {
      display: flex;
      align-items: center;
      gap: 10px;
      font-weight: 700;
      font-size: 1.15rem;
      color: var(--accent-cyan);
      letter-spacing: 0.5px;
    }
    .status-badge {
      background: rgba(16, 185, 129, 0.15);
      color: var(--accent-emerald);
      border: 1px solid rgba(16, 185, 129, 0.3);
      padding: 4px 10px;
      border-radius: 20px;
      font-size: 0.75rem;
      font-weight: 600;
    }
    .key-auth-box {
      display: flex;
      align-items: center;
      gap: 10px;
    }
    .key-auth-box input {
      background: var(--bg-chat);
      border: 1px solid var(--border-color);
      color: var(--accent-cyan);
      padding: 8px 12px;
      border-radius: 8px;
      font-family: 'Fira Code', monospace;
      font-size: 0.85rem;
      width: 260px;
      outline: none;
    }
    .key-auth-box input:focus {
      border-color: var(--accent-cyan);
    }
    .chat-area {
      flex: 1;
      background: var(--bg-chat);
      padding: 24px;
      overflow-y: auto;
      display: flex;
      flex-direction: column;
      gap: 16px;
    }
    .message {
      padding: 14px 18px;
      border-radius: 10px;
      max-width: 85%;
      font-size: 0.95rem;
      line-height: 1.6;
      white-space: pre-wrap;
      word-break: break-word;
    }
    .message.user {
      background: #1d4ed8;
      color: #ffffff;
      align-self: flex-end;
      border-bottom-right-radius: 2px;
    }
    .message.assistant {
      background: #1e293b;
      color: #e2e8f0;
      align-self: flex-start;
      border-left: 4px solid var(--accent-cyan);
      border-bottom-left-radius: 2px;
      font-family: 'Fira Code', monospace;
    }
    .controls-area {
      padding: 16px 24px;
      background: #162032;
      border-top: 1px solid var(--border-color);
      display: flex;
      gap: 12px;
    }
    textarea {
      flex: 1;
      background: var(--bg-chat);
      border: 1px solid var(--border-color);
      border-radius: 10px;
      color: #f3f4f6;
      padding: 12px;
      font-family: 'Inter', sans-serif;
      font-size: 0.95rem;
      resize: none;
      height: 52px;
      outline: none;
    }
    textarea:focus {
      border-color: var(--accent-cyan);
    }
    button {
      background: var(--accent-cyan);
      color: #070a12;
      border: none;
      border-radius: 10px;
      padding: 0 24px;
      font-weight: 700;
      cursor: pointer;
      transition: all 0.2s ease;
    }
    button:hover {
      background: #0891b2;
      color: #ffffff;
    }
  </style>
</head>
<body>
  <div class="wrapper">
    <div class="top-bar">
      <div class="brand-title">
        HYDRA AI 3B <span class="status-badge">ONLINE</span>
      </div>
      <div class="key-auth-box">
        <input type="password" id="apiKey" placeholder="Ingrese x-api-key para autenticar..." />
      </div>
    </div>
    <div class="chat-area" id="chat">
      <div class="message assistant">Consola privada activa. Ingrese su API Key en la parte superior para consultar al motor de IA.</div>
    </div>
    <div class="controls-area">
      <textarea id="promptInput" placeholder="Escriba su consulta o instrucción de código..."></textarea>
      <button onclick="sendPrompt()">Enviar</button>
    </div>
  </div>

  <script>
    async function sendPrompt() {
      const apiKeyInput = document.getElementById('apiKey').value.trim();
      const promptInput = document.getElementById('promptInput');
      const chat = document.getElementById('chat');
      const userText = promptInput.value.trim();

      if (!apiKeyInput) {
        alert('Ingrese una API Key válida en la barra superior.');
        return;
      }
      if (!userText) return;

      const userMsgDiv = document.createElement('div');
      userMsgDiv.className = 'message user';
      userMsgDiv.textContent = userText;
      chat.appendChild(userMsgDiv);

      promptInput.value = '';
      chat.scrollTop = chat.scrollHeight;

      const assistantMsgDiv = document.createElement('div');
      assistantMsgDiv.className = 'message assistant';
      assistantMsgDiv.textContent = '';
      chat.appendChild(assistantMsgDiv);

      try {
        const response = await fetch('/ask', {
          method: 'POST',
          headers: {
            'Content-Type': 'application/json',
            'x-api-key': apiKeyInput
          },
          body: JSON.stringify({ prompt: userText })
        });

        if (!response.ok) {
          const errData = await response.json().catch(() => ({}));
          assistantMsgDiv.textContent = `Error (${response.status}): ${errData.error || 'Acceso Denegado'}`;
          return;
        }

        const reader = response.body.getReader();
        const decoder = new TextDecoder();

        while (true) {
          const { done, value } = await reader.read();
          if (done) break;
          assistantMsgDiv.textContent += decoder.decode(value, { stream: true });
          chat.scrollTop = chat.scrollHeight;
        }
      } catch (err) {
        assistantMsgDiv.textContent = 'Error de conexión con el servidor.';
      }
    }

    document.getElementById('promptInput').addEventListener('keydown', (e) => {
      if (e.key === 'Enter' && !e.shiftKey) {
        e.preventDefault();
        sendPrompt();
      }
    });
  </script>
</body>
</html>
"""

@app.route('/ask', methods=['POST'])
def ask():
    api_key = request.headers.get("x-api-key")
    if not api_key or api_key not in VALID_API_KEYS:
        return jsonify({"error": "Acceso denegado. Llave invalida o inexistente."}), 401

    try:
        data = request.json
        user_prompt = data.get("prompt", "")
        
        prompt = f"<|im_start|>system\nEres Hydra 3.0, el motor de inteligencia artificial de Hydra Software. Eres un experto absoluto en programación y desarrollo de sistemas backend, bots y paneles web.<|im_end|>\n<|im_start|>user\n{user_prompt}<|im_end|>\n<|im_start|>assistant\n"
        
        inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
        streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)

        def generate_task():
            with torch.no_grad():
                model.generate(
                    **inputs, 
                    max_new_tokens=4096, 
                    do_sample=True, 
                    temperature=0.7,
                    top_p=0.9,
                    repetition_penalty=1.15,
                    streamer=streamer
                )

        thread = threading.Thread(target=generate_task)
        thread.start()

        def generate_stream():
            for text in streamer:
                yield text

        return Response(stream_with_context(generate_stream()), mimetype='text/plain')

    except Exception as e:
        return jsonify({"error": str(e)}), 500

@app.route('/', methods=['GET'])
def health():
    return render_template_string(HTML_PAGE)

if __name__ == '__main__':
    print("LOG: Iniciando servidor Flask en 0.0.0.0:7860")
    app.run(host='0.0.0.0', port=7860, threaded=True)