#!/bin/bash

# Configuración inicial
OLLAMA_URL="http://localhost:11434/api/generate"
MODELS=("gemma3:4b" "llama3.2:3b")  # LLM1 (Escritor), LLM2 (Editor) 
SYSTEM_PROMPT1="Eres un escritor de libros de ficción. Escribe una página (aproximadamente 100 palabras) para un libro. Usa el contexto previo para continuar la historia de manera coherente, manteniendo los personajes, el tono y la trama. No incluyas títulos ni explicaciones, solo el texto de la página."
SYSTEM_PROMPT2="Eres un editor de libros. Recibe una página escrita y revísala para mejorar la coherencia, estilo y continuidad con el contexto previo. Asegúrate de que los personajes, el tono y la trama se mantengan consistentes. Devuelve la página editada (aproximadamente 100 palabras). Si no hay cambios necesarios, devuelve la página tal cual. No incluyas explicaciones, solo el texto editado."
LOG_DIR="RUTA/DE/TU/DIRECTORIO/PARA/GUARDAR"
LOG_FILE="$LOG_DIR/book_ollama.log"
BOOK_FILE="$LOG_DIR/book.txt"
PAGE_COUNT=0
MAX_PAGES=5  # Reducido a 5 páginas

# Verificar dependencias
for cmd in curl jq; do
    if ! command -v "$cmd" &> /dev/null; then
        echo "Error: $cmd no está instalado. Instálalo para continuar."
        exit 1
    fi
done

# Crear directorio de logs
if [ ! -d "$LOG_DIR" ]; then
    echo "Creando directorio $LOG_DIR..."
    mkdir -p "$LOG_DIR" 2>/tmp/mkdir_error.log
    if [ $? -ne 0 ]; then
        echo "Error al crear $LOG_DIR. Detalles:"
        cat /tmp/mkdir_error.log
        exit 1
    fi
else
    if [ ! -w "$LOG_DIR" ]; then
        echo "Error: No tienes permisos de escritura en $LOG_DIR"
        exit 1
    fi
fi

for log in "$LOG_FILE" "$BOOK_FILE"; do
    touch "$log" || { echo "Error al crear $log"; exit 1; }
done

# Función para registrar mensajes
log_message() {
    echo "$(date '+%Y-%m-%d %H:%M:%S') - $1" >> "$LOG_FILE"
}

# Función para enviar solicitud a Ollama
send_request() {
    local prompt="$1"
    local model="$2"

    json_payload=$(jq -n --arg model "$model" --arg prompt "$prompt" '{model: $model, prompt: $prompt, stream: false}')
    response=$(curl -s -X POST "$OLLAMA_URL" -H "Content-Type: application/json" -d "$json_payload")
    if [ $? -ne 0 ]; then
        echo "Error: No se pudo conectar con Ollama"
        log_message "Error: No se pudo conectar con Ollama para $model"
        exit 1
    fi

    answer=$(echo "$response" | jq -r '.response')
    if [ -z "$answer" ] || [ "$answer" = "null" ]; then
        echo "Error: Respuesta inválida de $model"
        log_message "Error: Respuesta inválida de $model"
        exit 1
    fi

    echo "$answer"
}

# Función para obtener el contexto reciente (última página)
get_recent_context() {
    if [ -s "$BOOK_FILE" ]; then
        # Extraer la última página (ignorando las líneas de "Página X" y los separadores)
        tac "$BOOK_FILE" | awk '/^Página [0-9]+$/ {exit} {print}' | tac | grep -v '^$' | tail -n 100
    else
        echo ""
    fi
}

# Inicializar el libro
> "$BOOK_FILE"
log_message "Inicio de la creación del libro"

# Bucle para generar las 5 páginas
while [ $PAGE_COUNT -lt $MAX_PAGES ]; do
    PAGE_COUNT=$((PAGE_COUNT + 1))
    log_message "Generando página $PAGE_COUNT"

    # Obtener el contexto reciente (solo la última página)
    CONTEXT=$(get_recent_context)
    if [ -z "$CONTEXT" ]; then
        log_message "Contexto vacío para página $PAGE_COUNT (inicio del libro)"
    else
        log_message "Contexto para página $PAGE_COUNT:\n$CONTEXT"
    fi

    # LLM1: Generar página inicial
    log_message "LLM1 (${MODELS[0]}) escribiendo página $PAGE_COUNT"
    WRITER_PROMPT="$SYSTEM_PROMPT1\nContexto previo:\n$CONTEXT"
    INITIAL_PAGE=$(send_request "$WRITER_PROMPT" "${MODELS[0]}")
    log_message "Página inicial generada:\n$INITIAL_PAGE"

    # LLM2: Revisar y editar página
    log_message "LLM2 (${MODELS[1]}) editando página $PAGE_COUNT"
    EDITOR_PROMPT="$SYSTEM_PROMPT2\nContexto previo:\n$CONTEXT\nPágina a editar:\n$INITIAL_PAGE"
    EDITED_PAGE=$(send_request "$EDITOR_PROMPT" "${MODELS[1]}")
    log_message "Página editada:\n$EDITED_PAGE"

    # Validar continuidad (simple: comparar si hay palabras clave comunes)
    # Por ejemplo, buscamos si hay al menos 3 palabras comunes entre el contexto y la página editada
    if [ -n "$CONTEXT" ]; then
        COMMON_WORDS=$(echo "$CONTEXT" | tr ' ' '\n' | sort | uniq | grep -F -f - <(echo "$EDITED_PAGE" | tr ' ' '\n' | sort | uniq) | wc -l)
        if [ "$COMMON_WORDS" -lt 3 ]; then
            log_message "Advertencia: La página $PAGE_COUNT parece no seguir el contexto (solo $COMMON_WORDS palabras comunes). Reintentando..."
            PAGE_COUNT=$((PAGE_COUNT - 1))
            continue
        fi
    fi

    # Agregar la página al libro
    echo "Página $PAGE_COUNT" >> "$BOOK_FILE"
    echo "$EDITED_PAGE" >> "$BOOK_FILE"
    echo "" >> "$BOOK_FILE"  # Separador entre páginas
    log_message "Página $PAGE_COUNT agregada al libro"
done

log_message "Libro completado con $PAGE_COUNT páginas. Guardado en $BOOK_FILE"
echo "Libro completado. Revisa $BOOK_FILE para ver el resultado."

ollama serve

ollama run llama3.2:3b

>>>Send a message (/? for help)

>>>/?
Available Commands:
  /set            Set session variables
  /show           Show model information
  /load <model>   Load a session or model
  /save <model>   Save your current session
  /clear          Clear session context
  /bye            Exit
  /?, /help       Help for a command
  /? shortcuts    Help for keyboard shortcuts

Use """ to begin a multi-line message.

>>> Send a message (/? for help)

>>>/set
Available Commands:
  /set parameter ...     Set a parameter
  /set system <string>   Set system message
  /set history           Enable history
  /set nohistory         Disable history
  /set wordwrap          Enable wordwrap
  /set nowordwrap        Disable wordwrap
  /set format json       Enable JSON mode
  /set noformat          Disable formatting
  /set verbose           Show LLM stats
  /set quiet             Disable LLM stats

>>> Send a message (/? for help)

/set parameter

>>> /set parameter
Available Parameters:
  /set parameter seed <int>             Random number seed
  /set parameter num_predict <int>      Max number of tokens to predict
  /set parameter top_k <int>            Pick from top k num of tokens
  /set parameter top_p <float>          Pick token based on sum of probabilities
  /set parameter min_p <float>          Pick token based on top token probability * min_p
  /set parameter num_ctx <int>          Set the context size
  /set parameter temperature <float>    Set creativity level
  /set parameter repeat_penalty <float> How strongly to penalize repetitions
  /set parameter repeat_last_n <int>    Set how far back to look for repetitions
  /set parameter num_gpu <int>          The number of layers to send to the GPU
  /set parameter stop <string> <string> ...   Set the stop parameters

>>> Send a message (/? for help)

/set parameter temperature 0.4

/set system

>>> /set system Eres agenteIA, un asistente de IA para operaciones de analisis de datos.

>>>/bye

/set history

/set wordwrap

/set verbose

/save llama3.2:3b

/set parameter num_ctx 6144