added write to file button

Author: Aaron J Clifft

gui/astral_nexus_drafter.html

@@ -99,8 +99,13 @@ <h1 class="text-2xl font-bold text-white">Astral Drafter</h1>
                         <circle class="opacity-25" cx="12" cy="12" r="10" stroke="currentColor" stroke-width="4"></circle>
                         <path class="opacity-75" fill="currentColor" d="M4 12a8 8 0 018-8V0C5.373 0 0 5.373 0 12h4zm2 5.291A7.962 7.962 0 014 12H0c0 3.042 1.135 5.824 3 7.938l3-2.647z"></path>
                     </svg>
-                    <span>Generate & Save Prose</span>
+                    <span>Generate Prose</span>
                 </button>
+
+                <button id="save-button" class="mt-2 w-full bg-green-600 text-white font-bold py-3 px-4 rounded-md hover:bg-green-700 disabled:bg-green-800 disabled:cursor-not-allowed transition" disabled>
+                    <span>Save to File</span>
+                </button>
+
                  <div id="error-message" class="mt-3 text-center text-red-400 text-sm"></div>
             </div>
         </div>
@@ -128,6 +133,7 @@ <h1 class="text-2xl font-bold text-white">Astral Drafter</h1>
     <script>
         // --- Element Selectors ---
 const generateButton = document.getElementById('generate-button');
+const saveButton = document.getElementById('save-button');
 const newSceneButton = document.getElementById('new-scene-button');
 const spinner = document.getElementById('spinner');
 const prevSceneInput = document.getElementById('prev-scene-summary');
@@ -161,6 +167,7 @@ <h1 class="text-2xl font-bold text-white">Astral Drafter</h1>
 outlineInput.addEventListener('input', updateContextUsage);
 characterInput.addEventListener('input', updateContextUsage);
 generateButton.addEventListener('click', generateProse);
+saveButton.addEventListener('click', saveProseToFile);
 newSceneButton.addEventListener('click', resetInterface);
 sendButton.addEventListener('click', sendChatMessage);
 chatInput.addEventListener('keypress', (e) => {
@@ -180,9 +187,10 @@ <h1 class="text-2xl font-bold text-white">Astral Drafter</h1>
     conversationHistory = [];
     currentOutputPath = '';
     lastGeneratedContent = '';
-    chatInput.value = '';
+    chatInput.value = true;
     chatInput.disabled = true;
     sendButton.disabled = true;
+    saveButton.disabled = true;
     updateContextUsage();
 }
 
@@ -211,6 +219,47 @@ <h1 class="text-2xl font-bold text-white">Astral Drafter</h1>
     }
 }
 
+async function saveProseToFile() {
+    if (!lastGeneratedContent || !currentOutputPath) {
+        errorMessage.textContent = "No content to save or output path is missing.";
+        return;
+    }
+
+    saveButton.disabled = true;
+    errorMessage.textContent = '';
+
+    try {
+        const response = await fetch(MCP_SERVER_URL, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                save_content: lastGeneratedContent,
+                output_path: currentOutputPath
+            })
+        });
+
+        if (!response.ok) {
+            const errorData = await response.json();
+            throw new Error(errorData.error || `Save request failed: ${response.status}`);
+        }
+        
+        const result = await response.json();
+        // You can add a success message here if you want
+        console.log(result.message);
+        // Maybe flash the button green or show a small confirmation message
+        const originalText = saveButton.textContent;
+        saveButton.textContent = 'Saved!';
+        setTimeout(() => { saveButton.textContent = originalText; }, 2000);
+
+
+    } catch (error) {
+        console.error("Error saving file:", error);
+        errorMessage.textContent = `Error saving file: ${error.message}`;
+    } finally {
+        saveButton.disabled = false;
+    }
+}
+
 async function generateProse() {
     const prevScene = prevSceneInput.value.trim();
     const outline = outlineInput.value.trim();
@@ -268,6 +317,7 @@ <h1 class="text-2xl font-bold text-white">Astral Drafter</h1>
     generateButton.disabled = true;
     sendButton.disabled = true;
     chatInput.disabled = true;
+    saveButton.disabled = true;
     spinner.classList.remove('hidden');
 
     const aiResponseContainer = addAiMessageContainer();
@@ -280,7 +330,6 @@ <h1 class="text-2xl font-bold text-white">Astral Drafter</h1>
             body: JSON.stringify({
                 conversation_history: conversationHistory,
                 stream: true,
-                output_path: currentOutputPath 
             })
         });
 
@@ -318,8 +367,8 @@ <h1 class="text-2xl font-bold text-white">Astral Drafter</h1>
         });
 
         addCopyToClipboardButton(aiResponseContainer, fullResponse);
-        addSaveConfirmation(aiResponseContainer, currentOutputPath);
-
+        saveButton.disabled = false;
+        
     } catch (error) {
         console.error("Error querying Bridge server:", error);
         aiResponseContent.innerText = `Error connecting to Bridge server. Make sure it's running at ${MCP_SERVER_URL}.\n\nDetails: ${error.message}`;
@@ -409,7 +458,9 @@ <h1 class="text-2xl font-bold text-white">Astral Drafter</h1>
     container.appendChild(confirmation);
 }
 
-resetInterface();
+function resetInterface() {
+    saveButton.disabled = true;
+};
     </script>
 </body>
 </html>

llama_cpp_server_bridge.py

@@ -15,31 +15,57 @@ def do_POST(self):
             post_data = self.rfile.read(content_length)
             data = json.loads(post_data)
 
-            conversation_history = data.get('conversation_history', [])
-            output_path = data.get('output_path')
-            
-            if not conversation_history or not output_path:
-                self._send_error("Missing 'conversation_history' or 'output_path'.", 400)
-                return
+            # NEW: Check if this is a save request or a generation request
+            if 'save_content' in data:
+                self._handle_save_request(data)
+            else:
+                self._handle_generation_request(data)
 
         except (json.JSONDecodeError, TypeError, KeyError) as e:
             self._send_error(f"Invalid JSON request: {e}", 400)
             return
 
+    # NEW: Function to handle saving the file
+    def _handle_save_request(self, data):
+        content = data.get('save_content')
+        output_path = data.get('output_path')
+
+        if not content or not output_path:
+            self._send_error("Missing 'save_content' or 'output_path' for save request.", 400)
+            return
+        
         try:
             output_dir = os.path.dirname(output_path)
             if output_dir:
                 os.makedirs(output_dir, exist_ok=True)
 
+            # Write the final content to the file
             with open(output_path, 'w', encoding='utf-8') as outfile:
-                self._stream_to_llama_cpp(conversation_history, outfile)
+                outfile.write(content)
+            
+            self.send_response(200)
+            self.send_header('Content-Type', 'application/json')
+            self.send_header('Access-Control-Allow-Origin', '*')
+            self.end_headers()
+            response = json.dumps({'message': f'Successfully saved to {output_path}'})
+            self.wfile.write(response.encode('utf-8'))
 
         except IOError as e:
             self._send_error(f"Could not write to file: {e}", 500)
         except Exception as e:
             self._send_error(f"An unexpected server error occurred: {e}", 500)
-    
-    def _stream_to_llama_cpp(self, conversation_history, outfile):
+
+    # MODIFIED: This function now only handles generation and streaming
+    def _handle_generation_request(self, data):
+        conversation_history = data.get('conversation_history', [])
+        if not conversation_history:
+            self._send_error("Missing 'conversation_history' for generation request.", 400)
+            return
+        
+        self._stream_to_llama_cpp(conversation_history)
+
+    # MODIFIED: This function no longer writes to a file
+    def _stream_to_llama_cpp(self, conversation_history):
         try:
             llama_cpp_payload = {
                 "messages": conversation_history,
@@ -64,7 +90,7 @@ def _stream_to_llama_cpp(self, conversation_history, outfile):
                             chunk = json.loads(line)
                             content = chunk.get('choices', [{}])[0].get('delta', {}).get('content', '')
                             if content:
-                                outfile.write(content)
+                                # REMOVED: outfile.write(content)
                                 gui_response = json.dumps({"response": content})
                                 self.wfile.write(gui_response.encode('utf-8') + b'\n')
                                 self.wfile.flush()