server.py

"""
DG Ingenieria - Servidor Local para Prototipo Civil
====================================================
Sirve el dashboard y conecta el agente IA con Claude Sonnet.

USO:
  1. Configurar API key:
     set ANTHROPIC_API_KEY=sk-ant-...
     (o crear archivo .env con ANTHROPIC_API_KEY=sk-ant-...)

  2. Ejecutar:
     python servidor_local.py

  3. Abrir en navegador:
     http://localhost:8080/PROTOTIPO_DG_CIVIL_v2.html
"""

import http.server
import json
import os
import sys
import urllib.request
import urllib.error
import ssl
import csv
import re
from datetime import datetime

PORT = 8080
API_URL = "https://api.anthropic.com/v1/messages"
LOG_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "consultas_log.csv")

# Patron para extraer la clasificacion de la respuesta de Claude
CLASIFICACION_PATTERN = re.compile(r"\[CLASIFICACION:(OBRA|AMBIGUA|FUERA)\]")


def log_consulta(clasificacion, query, response_text):
    """Escribe una linea al CSV de log de consultas."""
    file_exists = os.path.exists(LOG_FILE)
    try:
        with open(LOG_FILE, "a", newline="", encoding="utf-8") as f:
            writer = csv.writer(f)
            if not file_exists:
                writer.writerow(["timestamp", "clasificacion", "consulta", "respuesta_preview"])
            writer.writerow([
                datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                clasificacion,
                query[:100],
                response_text[:100]
            ])
    except Exception as e:
        print(f"  [LOG] Error escribiendo log: {e}")


# Contexto del proyecto para el agente
SYSTEM_PROMPT = """Sos el Agente DG Civil, un asistente de inteligencia de obra de DG Ingenieria SRL (Paraguay).
Estas conectado a los datos del proyecto Torres DG (cliente: Grupo DG Desarrollo, Asuncion).

ALCANCE - LO QUE PODES RESPONDER:
- Todo sobre el proyecto Torres DG: avance, costos, cronograma, subcontratistas, alertas, earned value, proyecciones.
- Analisis y recomendaciones de gestion de obra, construccion, ingenieria civil.
- Comparaciones entre rubros, contratistas, periodos.
- Metodologias de gestion de proyectos (PMBOK, Earned Value, Curva S, ruta critica).
- Estrategias para resolver problemas especificos de la obra.
- Generar informes y reportes del proyecto.

FUERA DE ALCANCE - LO QUE NO DEBES RESPONDER:
- Cualquier tema que NO sea sobre la obra o gestion de proyectos de construccion.
- No respondas preguntas generales de conocimiento, cultura, historia, politica, deportes, etc.
- No escribas codigo, no hagas traducciones, no respondas sobre otros temas tecnologicos.
- No respondas preguntas personales ni actues como asistente general.
- Si el usuario pregunta algo fuera de alcance, responde exactamente:
  "Soy el Agente DG Civil y estoy especializado en el control de Torres DG. Puedo ayudarte con avance de obra, costos, contratistas, alertas, cronograma, o generar informes del proyecto. ¿En que aspecto de la obra puedo asistirte?"

DATOS ACTUALES DEL PROYECTO:
- Avance General: 68% (plan 72%, desvio -4 pts)
- Presupuesto Total: Gs. 34,200M
- Ejecutado: Gs. 23,250M (68%)
- Desvio Presupuestario: +Gs. 2,840M (+8.3%)
- SPI (Cronograma): 0.95 (mejorando +0.03)
- CPI (Costo): 0.92 (empeorando -0.02)
- Personal Promedio/Dia: 47 personas (+5 vs mes anterior)
- Dias Perdidos (Clima): 3 de 22 habiles
- Contratos Activos: 28 (2 con atraso documental)
- Fecha Est. Finalizacion: Jun 2025 (+18 dias vs original May 2025)
- Contingencia disponible: Gs. 1,710M (5% PPTO)
- Semaforo General: AMARILLO

AVANCE POR RUBRO (Plan% -> Real% = Desvio):
- Hormigon Armado: 95% -> 92% = -3 pts (Peso 18%, PPTO 6200M, Ejecutado 6620M, desvio +420M)
- Albanileria: 80% -> 72% = -8 pts (Peso 8%, PPTO 2800M, Ejecutado 2750M, desvio -50M) **RUTA CRITICA: bloquea pintura**
- Inst. Electrica: 65% -> 58% = -7 pts (Peso 12%, PPTO 3100M, Ejecutado 3410M, desvio +310M)
- Cielorraso: 55% -> 50% = -5 pts (Peso 3%, PPTO 900M, Ejecutado 870M)
- Pintura: 40% -> 38% = -2 pts (Peso 4%, PPTO 1200M, Ejecutado 1150M)
- Carp. Aluminio: 65% -> 60% = -5 pts (Peso 10%, PPTO 4500M, Ejecutado 4680M, desvio +180M)
- Carp. Madera: 50% -> 48% = -2 pts (Peso 5%, PPTO 1800M, Ejecutado 1760M)
- Inst. Sanitaria: 72% -> 70% = -2 pts (Peso 8%, PPTO 2400M, Ejecutado 2380M)
- Termomecanica AA: 78% -> 68% = -10 pts (Peso 7%, PPTO 2900M, Ejecutado 3380M, desvio +480M) **CRITICO**
- Ascensores: 50% -> 50% = 0 pts (Peso 9%, PPTO 3800M, Ejecutado 3800M)
- Pisos y Revestimientos: 55% -> 50% = -5 pts (Peso 6%, PPTO 3200M, Ejecutado 3160M)
- Demolicion: 100% -> 100% = 0 pts (Peso 2%, PPTO 680M, Ejecutado 680M)

ALERTAS ACTIVAS:
1. ALTA - Termomecanica AA: desvio presupuestario +16.5% (Gs. 480M). Causa: equipos importados mas caros
2. ALTA - Inst. Electrica: atraso -7 pts + sobrecosto +10% (+Gs. 310M). Causa: retrasos ANDE
3. MEDIA - TYM Ingenieria: caida asistencia -29% (de 7.3 a 5.2 personas/dia). Patron: 6>5>4>3
4. MEDIA - Albanileria: atraso -8 pts. Bloquea pintura. Puede correr fecha +1 mes
5. MEDIA - Hormigon Armado: desvio costo +6.8% (+Gs. 420M). Causa: acero +15%
6. BAJA - CPI descendente: 0.96 > 0.94 > 0.92 (2do mes a la baja)

TOP SUBCONTRATISTAS (Asistencia Prom/Dia):
- AGB Constructora: 12.3 (+1.5) OK
- Soletanche Bachy: 8.7 (+0.8) OK
- TYM Ingenieria: 5.2 (-2.1) CAIDA
- CAVECON SA: 4.8 (-0.3) Leve
- Carlos Benitez: 3.5 (+0.5) OK
- Aislatec SRL: 3.2 (+1.2) OK
- GyC Instalaciones: 4.1 (-0.5) Leve
- EcoEnergia SA: 2.8 (+0.3) OK

PROYECCIONES EAC:
- Optimista (CPI 0.96): Gs. 35,600M (+4.1%)
- Probable (CPI 0.92): Gs. 37,170M (+8.7%)
- Pesimista (CPI 0.88): Gs. 38,860M (+13.6%)

EARNED VALUE:
- PV (Planned Value): Gs. 24,620M
- EV (Earned Value): Gs. 21,390M
- AC (Actual Cost): Gs. 23,250M
- SV (Schedule Variance): -Gs. 3,230M
- CV (Cost Variance): -Gs. 1,860M

INSTRUCCIONES:
- Responde SIEMPRE en espanol (sin acentos en caracteres especiales para compatibilidad).
- Se conciso y directo.
- Usa datos reales del proyecto, no inventes.
- Si preguntan algo fuera de los datos disponibles, di que necesitas mas datos del sistema.
- Sos un experto en gestion de proyectos de construccion (PMBOK, Earned Value, Curva S).
- Cuando des recomendaciones, se especifico y accionable.
- Firma siempre al final con "---" seguido de "Agente DG Civil" en linea nueva.

CLASIFICACION OBLIGATORIA DE CONSULTAS:
CADA respuesta que generes DEBE empezar con EXACTAMENTE una de estas etiquetas en la primera linea:
- [CLASIFICACION:OBRA] - La consulta esta directamente relacionada con el proyecto Torres DG, gestion de obra, costos, avance, subcontratistas, cronograma, earned value, o temas de ingenieria civil y construccion.
- [CLASIFICACION:AMBIGUA] - La consulta PODRIA estar relacionada pero no es clara. Ejemplos: clima general, preguntas genericas de construccion sin referencia al proyecto, temas que podrian o no aplicar a la obra.
- [CLASIFICACION:FUERA] - La consulta NO tiene relacion con el proyecto ni con gestion de obras. Ejemplos: deportes, politica, cultura, programacion, preguntas personales, etc.

REGLAS DE CLASIFICACION:
- Para [CLASIFICACION:OBRA]: responde normalmente con toda la informacion del proyecto.
- Para [CLASIFICACION:AMBIGUA]: NO respondas la consulta. En su lugar, responde SOLAMENTE con:
  "Esta consulta podria estar fuera del alcance del proyecto. Cada consulta consume recursos del sistema. Si su pregunta esta relacionada con Torres DG, por favor confirme y le respondo con gusto."
  Si el usuario CONFIRMA que es sobre el proyecto en un mensaje posterior, clasifica como [CLASIFICACION:OBRA] y responde normalmente.
- Para [CLASIFICACION:FUERA]: responde con el mensaje de fuera de alcance:
  "Soy el Agente DG Civil y estoy especializado en el control de Torres DG. Puedo ayudarte con avance de obra, costos, contratistas, alertas, cronograma, o generar informes del proyecto. En que aspecto de la obra puedo asistirte?"

IMPORTANTE: La etiqueta [CLASIFICACION:...] debe ir SIEMPRE en la primera linea, sola, antes de cualquier otro contenido.

FORMATO DE RESPUESTA:
- Usa **texto** para negritas (cosas importantes, nombres, valores clave).
- Usa lineas que empiecen con - para listas de items.
- Usa TITULOS EN MAYUSCULAS SEGUIDOS DE : para separar secciones (ej: "RENDIMIENTO ACTUAL:")
- Numera los items cuando sea una lista ordenada (1. 2. 3.)
- Se estructurado y profesional.

INFORMES PDF:
- SOLO genera un PDF cuando el usuario use explicitamente la palabra "informe" o "reporte" pidiendo un documento.
- Si el usuario solo pregunta o consulta (ej: "como va Soletanche?", "que pasa con el presupuesto?"), responde normalmente SIN ningun marcador de PDF. Es solo una respuesta conversacional.
- Cuando SI pida un informe (ej: "haceme un informe de Soletanche", "necesito un reporte de rubros criticos"), genera el contenido detallado y al final agrega: [PDF_CUSTOM:Titulo del Informe]
- Para "informe inversor" o "informe ejecutivo general" usa: [PDF_DOWNLOAD:inversor]
- Para "reporte de alertas" usa: [PDF_DOWNLOAD:alertas]
- IMPORTANTE: Preguntar NO es pedir informe. "Como va TYM?" = respuesta normal. "Haceme un informe de TYM" = respuesta + PDF.
"""


def get_api_key():
    """Get API key from environment or .env file."""
    key = os.environ.get("ANTHROPIC_API_KEY", "")
    if key:
        return key

    env_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), ".env")
    if os.path.exists(env_path):
        with open(env_path, "r") as f:
            for line in f:
                line = line.strip()
                if line.startswith("ANTHROPIC_API_KEY="):
                    return line.split("=", 1)[1].strip().strip("\"'")

    return ""


class DGHandler(http.server.SimpleHTTPRequestHandler):
    def __init__(self, *args, **kwargs):
        super().__init__(
            *args,
            directory=os.path.dirname(os.path.abspath(__file__)),
            **kwargs
        )

    def do_POST(self):
        if self.path == "/api/chat":
            self.handle_chat()
        elif self.path == "/api/log":
            self.handle_log()
        else:
            self.send_error(404)

    def do_OPTIONS(self):
        self.send_response(200)
        self.send_header("Access-Control-Allow-Origin", "*")
        self.send_header("Access-Control-Allow-Methods", "POST, OPTIONS")
        self.send_header("Access-Control-Allow-Headers", "Content-Type")
        self.end_headers()

    def handle_chat(self):
        api_key = get_api_key()
        if not api_key:
            self.send_json(500, {
                "error": "ANTHROPIC_API_KEY no configurada. Ejecute: set ANTHROPIC_API_KEY=sk-ant-..."
            })
            return

        content_length = int(self.headers.get("Content-Length", 0))
        body = self.rfile.read(content_length)
        data = json.loads(body)
        messages = data.get("messages", [])

        payload = json.dumps({
            "model": "claude-sonnet-4-20250514",
            "max_tokens": 2048,
            "system": SYSTEM_PROMPT,
            "messages": messages
        }).encode("utf-8")

        req = urllib.request.Request(
            API_URL,
            data=payload,
            headers={
                "Content-Type": "application/json",
                "x-api-key": api_key,
                "anthropic-version": "2023-06-01"
            },
            method="POST"
        )

        # Extraer la consulta del usuario para logging
        user_query = ""
        for msg in reversed(messages):
            if msg.get("role") == "user":
                user_query = msg.get("content", "")
                if isinstance(user_query, list):
                    user_query = " ".join(
                        p.get("text", "") for p in user_query if p.get("type") == "text"
                    )
                break

        try:
            ctx = ssl.create_default_context()
            with urllib.request.urlopen(req, context=ctx, timeout=60) as resp:
                result = json.loads(resp.read().decode("utf-8"))
                text = result.get("content", [{}])[0].get("text", "Sin respuesta")

                # Parsear y extraer la clasificacion
                clasificacion = "OBRA"  # default si Claude no incluyo tag
                match = CLASIFICACION_PATTERN.search(text)
                if match:
                    clasificacion = match.group(1)

                # Remover la etiqueta de clasificacion antes de enviar al frontend
                clean_text = CLASIFICACION_PATTERN.sub("", text).strip()
                # Limpiar salto de linea extra al inicio si quedo
                clean_text = re.sub(r"^\n+", "", clean_text)

                # Log de la consulta
                log_consulta(clasificacion, user_query, clean_text)
                print(f"  [CLAS] {clasificacion} | {user_query[:60]}")

                self.send_json(200, {
                    "response": clean_text,
                    "clasificacion": clasificacion
                })
        except urllib.error.HTTPError as e:
            error_body = e.read().decode("utf-8", errors="replace")
            print(f"API Error {e.code}: {error_body}")
            self.send_json(e.code, {"error": f"Error API: {error_body}"})
        except Exception as e:
            print(f"Error: {e}")
            self.send_json(500, {"error": str(e)})

    def handle_log(self):
        """Endpoint para que el frontend registre consultas en modo simulado."""
        try:
            content_length = int(self.headers.get("Content-Length", 0))
            body = self.rfile.read(content_length)
            data = json.loads(body)

            clasificacion = data.get("clasificacion", "OBRA")
            query = data.get("query", "")
            response_text = data.get("response", "")

            # Validar clasificacion
            if clasificacion not in ("OBRA", "AMBIGUA", "FUERA"):
                clasificacion = "OBRA"

            log_consulta(clasificacion, query, response_text)
            print(f"  [LOG-SIM] {clasificacion} | {query[:60]}")

            self.send_json(200, {"status": "ok"})
        except Exception as e:
            print(f"  [LOG] Error en /api/log: {e}")
            self.send_json(500, {"error": str(e)})

    def send_json(self, code, data):
        response = json.dumps(data).encode("utf-8")
        self.send_response(code)
        self.send_header("Content-Type", "application/json")
        self.send_header("Access-Control-Allow-Origin", "*")
        self.send_header("Content-Length", len(response))
        self.end_headers()
        self.wfile.write(response)

    def log_message(self, format, *args):
        msg = format % args
        if "GET /api/" in msg or "POST /api/" in msg:
            print(f"  [API] {msg}")
        elif "404" in msg or "500" in msg:
            print(f"  [ERR] {msg}")


def main():
    api_key = get_api_key()

    print()
    print("=" * 60)
    print("  DG INGENIERIA - Servidor Local")
    print("  Plataforma de Control de Obras v2.0")
    print("=" * 60)
    print()

    if api_key:
        print(f"  API Key: ...{api_key[-8:]}")
        print(f"  Modelo: claude-sonnet-4-20250514")
        print(f"  Agente IA: CONECTADO")
    else:
        print("  API Key: NO CONFIGURADA")
        print("  Para conectar el agente IA:")
        print("    set ANTHROPIC_API_KEY=sk-ant-...")
        print("  (o crear archivo .env)")
        print()
        print("  El dashboard funcionara sin IA (respuestas simuladas)")

    print()
    print(f"  Servidor: http://localhost:{PORT}")
    print(f"  Dashboard: http://localhost:{PORT}/PROTOTIPO_DG_CIVIL_v2.html")
    print()
    print("  Presione Ctrl+C para detener")
    print("=" * 60)
    print()

    server = http.server.HTTPServer(("", PORT), DGHandler)
    try:
        server.serve_forever()
    except KeyboardInterrupt:
        print("\n  Servidor detenido.")
        server.server_close()


if __name__ == "__main__":
    main()