From 2f7cda4b6dcc1feb03d041eb7e41b293d223e4f5 Mon Sep 17 00:00:00 2001
From: Georg Tomitsch <georg@tomitsch.net>
Date: Tue, 1 Apr 2025 23:38:35 +0000
Subject: [PATCH] Refactored everything into modules.

---
 public/index.html           | 26 +---------
 public/js/ai-fiction.js     | 13 ++---
 public/js/input-handler.js  |  5 +-
 public/js/kokoro-handler.js |  6 +--
 public/js/socket-client.js  |  5 +-
 public/js/speech.js         | 54 +++++++++++++++++++++
 public/js/tts-factory.js    | 97 +++++++++++++++----------------------
 public/js/tts-handler.js    |  6 +--
 8 files changed, 107 insertions(+), 105 deletions(-)
 create mode 100644 public/js/speech.js

diff --git a/public/index.html b/public/index.html
index eaf0d85..90a9191 100644
--- a/public/index.html
+++ b/public/index.html
@@ -66,8 +66,7 @@
         window.rstack = [];
     </script>
     
-    <!-- TTS implementation scripts - order matters! -->   
-    <!-- 1. Kokoro TTS library - load as module -->   
+    <!-- Kokoro TTS library - load as module -->   
     <script type="module">
         try {
             // Import KokoroTTS class from the module
@@ -88,28 +87,7 @@
         }
     </script>
     
-    <!-- 2. TTS handlers (kokoro-handler needs to wait for KokoroTTS) -->
-    <script src="js/kokoro-handler.js"></script>
-    <script src="js/tts-handler.js"></script>
-    
-    <!-- 3. TTS Factory for automatic selection -->
-    <script src="js/tts-factory.js"></script>
-    
-    <!-- New Modules for Socket-based Interaction -->
-    <script src="js/input-handler.js"></script>
-    <script src="js/socket-client.js"></script>
-    
-    <!-- Modular layout and animation components -->
-    <script type="module" src="js/animation-queue.js"></script>
-    <script type="module" src="js/text-processor.js"></script>
-    <script type="module" src="js/paragraph-layout.js"></script>
-    <script type="module" src="js/layout-renderer.js"></script>
-    <script type="module" src="js/persistence-manager.js"></script>
-    <script type="module" src="js/ui-controller.js"></script>
-    <script type="module" src="js/audio-manager.js"></script>
-    <script type="module" src="js/tts-player.js"></script>
-    
-    <!-- Main application script - using ES modules -->
+    <!-- Main application script - imports all needed modules -->
     <script type="module" src="js/ai-fiction.js"></script>
   </body>
 </html>
diff --git a/public/js/ai-fiction.js b/public/js/ai-fiction.js
index 015441e..25808ac 100644
--- a/public/js/ai-fiction.js
+++ b/public/js/ai-fiction.js
@@ -9,12 +9,10 @@ import { LayoutRenderer } from './layout-renderer.js';
 import { AudioManager } from './audio-manager.js';
 import { TtsPlayer } from './tts-player.js';
 import { PersistenceManager } from './persistence-manager.js';
-// import { InkStoryPlayer } from './ink-story-player.js'; // Replaced by SocketClient logic
+import { InputHandler } from './input-handler.js';
+import { SocketClient } from './socket-client.js';
 import { UiController } from './ui-controller.js';
-// Assuming InputHandler and SocketClient are loaded globally via <script> tags in HTML
-// If using modules properly, they would need imports:
-// import { InputHandler } from './input-handler.js';
-// import { SocketClient } from './socket-client.js';
+import { ttsFactory } from './tts-factory.js';
 
 export class AnimatedFiction {
     /**
@@ -71,11 +69,6 @@ export class AnimatedFiction {
         this.initializeTextMeasurement();
 
         // 2. Input, Socket, and UI Controller
-        // Ensure InputHandler and SocketClient classes are available globally
-        if (typeof InputHandler === 'undefined' || typeof SocketClient === 'undefined') {
-            console.error("InputHandler or SocketClient class not found. Ensure scripts are loaded correctly.");
-            return; // Stop initialization if core components are missing
-        }
         this.inputHandler = new InputHandler('player_input', 'cursor');
         this.socketClient = new SocketClient(this.config.serverUrl); // Pass server URL if provided
 
diff --git a/public/js/input-handler.js b/public/js/input-handler.js
index ea10d0b..9584af3 100644
--- a/public/js/input-handler.js
+++ b/public/js/input-handler.js
@@ -2,7 +2,7 @@
  * Input Handler Module
  * Manages the multi-line text input field with a custom cursor.
  */
-class InputHandler {
+export class InputHandler {
   constructor(inputId = 'player_input', cursorId = 'cursor') {
     this.playerInput = document.getElementById(inputId);
     this.cursor = document.getElementById(cursorId);
@@ -288,6 +288,3 @@ class InputHandler {
     */
   }
 }
-
-// Export the class if using modules (optional, depends on build setup)
-// export default InputHandler;
diff --git a/public/js/kokoro-handler.js b/public/js/kokoro-handler.js
index 7d147ac..5793041 100644
--- a/public/js/kokoro-handler.js
+++ b/public/js/kokoro-handler.js
@@ -3,7 +3,7 @@
  * Uses the kokoro-js library for high-quality TTS
  */
 
-class KokoroHandler {
+export class KokoroHandler {
   constructor() {
     this.enabled = false;
     this.speaking = false;
@@ -593,5 +593,5 @@ class KokoroHandler {
   }
 }
 
-// Don't create a global instance here - the factory will do this
-// const ttsHandler = new KokoroHandler();
\ No newline at end of file
+// Create and export a singleton for the factory to use
+export const kokoroHandler = new KokoroHandler();
\ No newline at end of file
diff --git a/public/js/socket-client.js b/public/js/socket-client.js
index 88fc635..f522cc6 100644
--- a/public/js/socket-client.js
+++ b/public/js/socket-client.js
@@ -2,7 +2,7 @@
  * Socket Client Module
  * Manages WebSocket communication with the game server.
  */
-class SocketClient {
+export class SocketClient {
   constructor(serverUrl) {
     this.socket = null;
     this.serverUrl = serverUrl || window.location.origin; // Default to current origin
@@ -174,6 +174,3 @@ class SocketClient {
     this.emit('loadGame');
   }
 }
-
-// Export the class if using modules
-// export default SocketClient;
diff --git a/public/js/speech.js b/public/js/speech.js
new file mode 100644
index 0000000..820c60e
--- /dev/null
+++ b/public/js/speech.js
@@ -0,0 +1,54 @@
+const axios = require('axios');
+const fs = require('fs');
+const crypto = require('crypto');
+const player = require('play-sound')(opts = {});
+const { ipcMain } = require('electron');
+
+// Directory where audio files will be cached
+const cacheDirectory = './speech_cache/';
+
+// Create cache directory if it does not exist
+if (!fs.existsSync(cacheDirectory)) {
+  fs.mkdirSync(cacheDirectory);
+}
+
+ipcMain.handle('getSpeech', async (event, text) => {
+  // Create a hash of the text to use as a unique filename
+  const filename = crypto.createHash('md5').update(text).digest('hex') + '.mp3';
+
+  // Full path of the audio file in the cache directory
+  const filepath = cacheDirectory + filename;
+
+  // Check if audio file already exists in the cache
+  if (!fs.existsSync(filepath)) {
+    // If audio file does not exist, make API request
+    try {
+      const response = await axios({
+        method: 'post',
+        url: 'https://api.elevenlabs.io/v1/text-to-speech/8JNqTOY3RaSYcHTVJZ0G',
+        headers: {
+          'Content-Type': 'application/json',
+          'xi-api-key': 'd191e27c2e5b07573b39fe70f0783f48'
+        },
+        data: {
+          text: text,
+          model_id: 'eleven_multilingual_v1',
+          voice_settings: {
+            stability: 0,
+            similarity_boost: 0,
+            style: 0.5,
+            use_speaker_boost: true
+          }
+        },
+        responseType: 'arraybuffer'
+      });
+
+      // Write the audio data to a file in the cache directory
+      fs.writeFileSync(filepath, response.data);
+
+    } catch (error) {
+      console.error(`Error making API request: ${error}`);
+    }
+  }
+  return filepath
+});
\ No newline at end of file
diff --git a/public/js/tts-factory.js b/public/js/tts-factory.js
index 730f904..5c65714 100644
--- a/public/js/tts-factory.js
+++ b/public/js/tts-factory.js
@@ -2,12 +2,12 @@
  * TTS Factory for AI Interactive Fiction
  * Attempts to use Kokoro TTS first, then falls back to browser TTS if needed
  */
+import { kokoroHandler } from './kokoro-handler.js';
+import { browserTtsHandler } from './tts-handler.js';
 
-class TTSFactory {
+export class TTSFactory {
   constructor() {
     this.activeTTSHandler = null;
-    this.kokoroHandler = null;
-    this.browserTTSHandler = null;
     this.initializationAttempted = false;
     this.usingKokoro = false;
     this.initializationPromise = null; // Promise for the factory initialization
@@ -35,54 +35,36 @@ class TTSFactory {
         let kokoroInitialized = false;
         // Try to initialize Kokoro first (preferred option)
         try {
-          // Check if KokoroHandler class is defined (loaded via script tag)
-          if (typeof KokoroHandler !== 'undefined') {
-            console.log('Attempting to initialize Kokoro TTS...');
-            this.kokoroHandler = new KokoroHandler();
-            
-            // --- Increase Timeout for Kokoro Initialization --- 
-            // Wait for KokoroHandler's internal initialization promise
-            // Use Promise.race to add a longer timeout specifically for Kokoro init
-            const kokoroTimeoutPromise = new Promise((_, reject) => 
-                setTimeout(() => reject(new Error('Kokoro initialization timed out in factory')), 60000) // 60 seconds timeout
-            );
+          console.log('Attempting to initialize Kokoro TTS...');
+          
+          // --- Increase Timeout for Kokoro Initialization --- 
+          // Wait for KokoroHandler's internal initialization promise
+          // Use Promise.race to add a longer timeout specifically for Kokoro init
+          const kokoroTimeoutPromise = new Promise((_, reject) => 
+              setTimeout(() => reject(new Error('Kokoro initialization timed out in factory')), 60000) // 60 seconds timeout
+          );
 
-            try {
-                kokoroInitialized = await Promise.race([
-                    this.kokoroHandler.initializationPromise,
-                    kokoroTimeoutPromise
-                ]);
-            } catch (timeoutError) {
-                console.error(timeoutError.message); // Log the timeout error
-                kokoroInitialized = false;
-            }
-            // --- End Increase Timeout --- 
-            
-            if (kokoroInitialized) {
-                 console.log('Kokoro Handler reported successful initialization.');
-            } else {
-                 console.warn('Kokoro Handler reported failed or timed out initialization.');
-            }
+          try {
+              kokoroInitialized = await Promise.race([
+                  kokoroHandler.initializationPromise,
+                  kokoroTimeoutPromise
+              ]);
+          } catch (timeoutError) {
+              console.error(timeoutError.message); // Log the timeout error
+              kokoroInitialized = false;
+          }
+          // --- End Increase Timeout --- 
+          
+          if (kokoroInitialized) {
+               console.log('Kokoro Handler reported successful initialization.');
           } else {
-              console.warn('KokoroHandler class not found when factory initialized.');
+               console.warn('Kokoro Handler reported failed or timed out initialization.');
           }
         } catch (error) {
-          console.error('Error creating or initializing Kokoro Handler:', error);
+          console.error('Error initializing Kokoro Handler:', error);
           kokoroInitialized = false; // Ensure it's marked as failed
         }
 
-        // Initialize browser TTS as fallback (can happen in parallel)
-        try {
-          if (typeof TTSHandler !== 'undefined') {
-            console.log('Initializing browser TTS as fallback...');
-            this.browserTTSHandler = new TTSHandler();
-          } else {
-              console.warn('TTSHandler class not found when factory initialized.');
-          }
-        } catch (error) {
-          console.error('Error initializing browser TTS:', error);
-        }
-
         // Decide which handler to use based on Kokoro's success
         this.selectActiveHandler(kokoroInitialized);
         resolve(); // Resolve the factory's promise
@@ -91,23 +73,21 @@ class TTSFactory {
     return this.initializationPromise;
   }
 
-  // Removed waitForKokoroInitialization as KokoroHandler now manages its own promise
-
   /**
    * Select which TTS handler to use
    * @param {boolean} kokoroInitialized - Whether Kokoro initialization succeeded
    */
   selectActiveHandler(kokoroInitialized) {
     // First choice: Kokoro if it's available and initialized successfully
-    if (kokoroInitialized && this.kokoroHandler && this.kokoroHandler.kokoroReady) {
+    if (kokoroInitialized && kokoroHandler.kokoroReady) {
       console.log('Using Kokoro TTS as primary TTS system');
-      this.activeTTSHandler = this.kokoroHandler;
+      this.activeTTSHandler = kokoroHandler;
       this.usingKokoro = true;
     }
     // Fallback to browser TTS if available
-    else if (this.browserTTSHandler) {
+    else if (browserTtsHandler) {
       console.log('Falling back to browser TTS.');
-      this.activeTTSHandler = this.browserTTSHandler;
+      this.activeTTSHandler = browserTtsHandler;
       this.usingKokoro = false;
     }
     // No TTS available
@@ -117,7 +97,7 @@ class TTSFactory {
       this.usingKokoro = false;
     }
 
-    // Expose the active handler as the global ttsHandler
+    // Expose the active handler as the global ttsHandler for compatibility
     window.ttsHandler = this.activeTTSHandler;
 
     // Log the active TTS system
@@ -160,8 +140,8 @@ class TTSFactory {
    * @param {string} type - Either 'kokoro' or 'browser'
    */
   switchTTS(type) {
-    if (type === 'kokoro' && this.kokoroHandler && this.kokoroHandler.kokoroReady) {
-      this.activeTTSHandler = this.kokoroHandler;
+    if (type === 'kokoro' && kokoroHandler && kokoroHandler.kokoroReady) {
+      this.activeTTSHandler = kokoroHandler;
       this.usingKokoro = true;
       window.ttsHandler = this.activeTTSHandler;
       console.log('Switched to Kokoro TTS');
@@ -169,8 +149,8 @@ class TTSFactory {
       const ttsReadyEvent = new CustomEvent('tts-ready', { detail: { available: true, type: 'kokoro', handler: this.activeTTSHandler } });
       window.dispatchEvent(ttsReadyEvent);
       return true;
-    } else if (type === 'browser' && this.browserTTSHandler) {
-      this.activeTTSHandler = this.browserTTSHandler;
+    } else if (type === 'browser' && browserTtsHandler) {
+      this.activeTTSHandler = browserTtsHandler;
       this.usingKokoro = false;
       window.ttsHandler = this.activeTTSHandler;
       console.log('Switched to browser TTS');
@@ -185,5 +165,8 @@ class TTSFactory {
   }
 }
 
-// Create the global TTS factory instance
-window.ttsFactory = new TTSFactory();
\ No newline at end of file
+// Create and export singleton instance
+export const ttsFactory = new TTSFactory();
+
+// Keep a reference in window for compatibility with existing code
+window.ttsFactory = ttsFactory;
\ No newline at end of file
diff --git a/public/js/tts-handler.js b/public/js/tts-handler.js
index 76a4cb6..cbf251c 100644
--- a/public/js/tts-handler.js
+++ b/public/js/tts-handler.js
@@ -3,7 +3,7 @@
  * Enhanced version with improved voice selection, caching, and playback controls
  */
 
-class TTSHandler {
+export class TTSHandler {
   constructor() {
     this.enabled = false;
     this.speaking = false;
@@ -410,5 +410,5 @@ class TTSHandler {
   }
 }
 
-// Create a global instance
-const ttsHandler = new TTSHandler();
\ No newline at end of file
+// Create and export a singleton instance
+export const browserTtsHandler = new TTSHandler();
\ No newline at end of file