uncloseai.com/uncloseai.js

/* 
 * uncloseai.js - Modular AI Integration Library
 * 
 * Main entry point that imports all modules and exports the public API
 * This maintains backward compatibility while enabling better code organization
 */

// External dependencies
import { marked } from "https://cdn.jsdelivr.net/npm/marked/lib/marked.esm.js";
import hljs from 'https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.8.0/es/highlight.min.js';

// Internal modules
import * as Config from './src/config.js';
import * as Models from './src/models.js';
import * as Content from './src/content.js';
import * as TTS from './src/tts.js';
import * as Chat from './src/chat.js';
import * as FileUpload from './src/file-upload.js';
import * as Storage from './src/storage.js';
import * as PageReader from './src/page-reader.js';
import * as UI from './src/ui.js';

// -------------------------
// Public API - Main Functions
// -------------------------

// Chat functionality
export const sendMessage = Chat.sendMessage;
export const handleUserInput = Chat.handleUserInput;

// TTS functionality
export const speakText = TTS.speakText;
export const processContentWithHermes = TTS.processContentWithHermes;
export const extractSpokenTokens = TTS.extractSpokenTokens;
export const generateTitleForTTS = TTS.generateTitleForTTS;

// File upload
export const uploadFile = FileUpload.uploadFile;
export const integrateMegafarceResponse = FileUpload.integrateMegafarceResponse;
export const showProgressIndicator = FileUpload.showProgressIndicator;
export const hideProgressIndicator = FileUpload.hideProgressIndicator;

// Page reading
export const readPageWithHermes = PageReader.readPageWithHermes;
export const extractWebpageContent = Content.extractWebpageContent;

// Model management
export const getSelectedModel = Models.getSelectedModel;
export const getSelectedModelEndpoint = Models.getSelectedModelEndpoint;
export const fetchModelsFromEndpoints = Models.fetchModelsFromEndpoints;
export const createModelSelectionDropdown = Models.createModelSelectionDropdown;
export const addRefreshModelsButton = Models.addRefreshModelsButton;

// Storage
export const saveConversationHistory = Storage.saveConversationHistory;
export const loadConversationHistory = Storage.loadConversationHistory;
export const clearConversationHistory = Storage.clearConversationHistory;
export const getPageSpecificKey = Storage.getPageSpecificKey;

// UI creation
export const createFloatingAIButton = UI.createFloatingAIButton;
export const initializeUncloseaiElements = UI.initializeUncloseaiElements;
export const createFullInterface = UI.createFullInterface;
export const createCustomInterface = UI.createCustomInterface;
export const toggleHermesModal = UI.toggleHermesModal;
export const openHermesModal = UI.openHermesModal;
export const openTTSModal = UI.openTTSModal;

// -------------------------
// Configuration and State
// -------------------------

// Export configuration constants
export const TTS_API_URL = Config.TTS_API_URL;
export const MEGAPARCE_API_URL = Config.MEGAPARCE_API_URL;
export const API_KEY = Config.API_KEY;
export const MODEL = Config.MODEL;
export const VLLM_ENDPOINTS = Config.VLLM_ENDPOINTS;

// Export model registry
export const modelRegistry = Models.modelRegistry;

// Export chat history (for external access)
export { chatHistory } from './src/chat.js';

// -------------------------
// Initialization
// -------------------------

// Initialize on page load
window.addEventListener('load', () => {
  console.log('uncloseai.js: window.onload event fired.');
  UI.initializeSystem();
  Models.createModelSelectionDropdown();
  Models.addRefreshModelsButton();
});

// -------------------------
// Global Scope Exports (for backward compatibility)
// -------------------------

// Export functions to global scope for HTML onclick handlers
window.handleUserInput = Chat.handleUserInput;
window.readPageWithHermes = PageReader.readPageWithHermes;
window.sendMessage = Chat.sendMessage;
window.speakText = TTS.speakText;
window.uploadFile = FileUpload.uploadFile;
window.openTTSModal = UI.openTTSModal;
window.openTranslateModal = UI.openTranslateModal;
window.toggleHermesModal = async () => await UI.toggleHermesModal();
window.extractWebpageContent = Content.extractWebpageContent;
window.getSelectedModel = Models.getSelectedModel;
window.getSelectedModelEndpoint = Models.getSelectedModelEndpoint;
window.showProgressIndicator = FileUpload.showProgressIndicator;
window.hideProgressIndicator = FileUpload.hideProgressIndicator;
window.handleTTSFromElement = UI.handleTTSFromElement;
window.handleUploadFromElement = UI.handleUploadFromElement;

// Export helper functions for class-based integrations
window.handleCustomChat = async function(button) {
  const container = button.parentElement;
  const input = container.querySelector('[data-chat-input]');
  const chatBox = container.querySelector('[data-chat-box]');
  const message = input.value.trim();
  
  if (!message) return;
  
  // Add user message
  chatBox.innerHTML += `<div style="margin: 2px 0; padding: 2px; background: #e3f2fd; border-radius: 2px; font-size: 0.9em;"><strong>You:</strong> ${message}</div>`;
  input.value = '';
  
  // Add AI thinking indicator
  const thinkingDiv = document.createElement('div');
  thinkingDiv.style.cssText = 'margin: 2px 0; padding: 2px; background: #f5f5f5; border-radius: 2px; font-size: 0.9em;';
  thinkingDiv.innerHTML = '<strong>AI:</strong> <em>thinking...</em>';
  chatBox.appendChild(thinkingDiv);
  chatBox.scrollTop = chatBox.scrollHeight;
  
  try {
    let response = '';
    for await (const chunk of Chat.sendMessage(message)) {
      response += chunk;
      thinkingDiv.innerHTML = `<strong>AI:</strong> ${response}`;
      chatBox.scrollTop = chatBox.scrollHeight;
    }
  } catch (error) {
    thinkingDiv.innerHTML = `<strong>Error:</strong> ${error.message}`;
  }
};

console.log('uncloseai.js: Modular version loaded successfully');