Add application file

Dockerfile

@@ -22,8 +22,9 @@ RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir --upgrade -r requirements.txt && \
     pip install --no-cache-dir --upgrade g4f[all]
 
-# Copy application code
+# Copy application code and .env file
 COPY --chown=user . .
+COPY --chown=user .env .env
 
 # Expose port
 EXPOSE 7860

app.py

@@ -1,4 +1,4 @@
-from flask import Flask
+from flask import Flask, url_for
 from config import Config
 from routes.api import api_bp
 from routes.views import views_bp
@@ -8,18 +8,7 @@ app.config.from_object(Config)
 
 # Register blueprints
 app.register_blueprint(views_bp)
-app.register_blueprint(api_bp)
-# app.run(debug=True)
-def create_app():
-    app = Flask(__name__)
-    app.config.from_object(Config)
-    
-    # Register blueprints
-    app.register_blueprint(views_bp)
-    app.register_blueprint(api_bp, url_prefix='/api')
-    
-    return app
+app.register_blueprint(api_bp, url_prefix='/api')
 
-# if __name__ == '__main__':
-#     app = create_app()
-#     app.run(debug=True)
+if __name__ == '__main__':
+    app.run(debug=True)

config.py

@@ -1,4 +1,8 @@
 import os
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
 
 class Config:
     UPLOAD_FOLDER = 'output'

docker-compose.yml

@@ -0,0 +1,8 @@
+version: '3'
+services:
+  app:
+    build: .
+    ports:
+      - "7860:7860"
+    env_file:
+      - .env 

routes/api.py

@@ -1,5 +1,5 @@
 from typing import List, Tuple
-from flask import Blueprint, jsonify, request, Response, send_from_directory, copy_current_request_context
+from flask import Blueprint, jsonify, request, Response, send_from_directory, copy_current_request_context, abort, g
 from functools import wraps
 import time
 import json
@@ -11,10 +11,12 @@ from services.model_provider import ModelProvider
 from services.document_generator import DocumentGenerator
 from config import Config
 from utils.retry_decorator import retry
+import threading
 
 api_bp = Blueprint('api', __name__)
 model_provider = ModelProvider()
 doc_generator = DocumentGenerator(Config.UPLOAD_FOLDER)
+_generation_tasks = {}
 
 def sse_stream_required(f):
     """Decorator to ensure SSE stream has request context"""
@@ -133,8 +135,15 @@ def stream():
     include_references = request.args.get('includeReferences') == 'true'
     citation_style = request.args.get('citationStyle')
 
+    # Generate unique task ID
+    task_id = str(uuid.uuid4())
+    _generation_tasks[task_id] = {'abort': False}
+
     def generate():
         try:
+            # Send task ID to client
+            yield "data: " + json.dumps({"task_id": task_id}) + "\n\n"
+
             if not research_subject:
                 yield "data: " + json.dumps({"error": "Research subject is required"}) + "\n\n"
                 return
@@ -475,8 +484,18 @@ def stream():
                     "md_file": md_filename
                 }) + "\n\n"
             
+            # Clean up task when done
+            if task_id in _generation_tasks:
+                del _generation_tasks[task_id]
+
         except Exception as e:
-            yield "data: " + json.dumps({"error": f"Failed to generate paper: {str(e)}"}) + "\n\n"
+            # Clean up task on error
+            if task_id in _generation_tasks:
+                del _generation_tasks[task_id]
+            if str(e) == "Generation aborted by user":
+                yield "data: " + json.dumps({"status": "aborted"}) + "\n\n"
+            else:
+                yield "data: " + json.dumps({"error": f"Failed to generate paper: {str(e)}"}) + "\n\n"
     
     return Response(generate(), mimetype="text/event-stream")
 
@@ -488,4 +507,12 @@ def download(filename):
         safe_filename,
         as_attachment=True
     )
+
+@api_bp.route('/abort/<task_id>', methods=['POST'])
+def abort_generation(task_id):
+    """Abort an ongoing generation task"""
+    if task_id in _generation_tasks:
+        _generation_tasks[task_id]['abort'] = True
+        return jsonify({'status': 'aborted'})
+    return jsonify({'status': 'not_found'}), 404
     

static/js/main.js

@@ -0,0 +1,381 @@
+let currentTaskId = null;
+let currentEventSource = null;
+
+document.addEventListener('DOMContentLoaded', function() {
+    // Form submission handler
+    const form = document.getElementById('paperForm');
+    form.addEventListener('submit', startGeneration);
+
+    // Abort button handler
+    const abortBtn = document.getElementById('abortBtn');
+    abortBtn.addEventListener('click', abortGeneration);
+
+    // Structure controls
+    const structureRadios = document.querySelectorAll('input[name="structure"]');
+    const automaticOptions = document.getElementById('automaticOptions');
+
+    // References controls
+    const includeReferences = document.getElementById('includeReferences');
+    const citationStyle = document.getElementById('citationStyle');
+    const citationStyleContainer = citationStyle.parentElement;
+
+    // Chapter count controls
+    const chapterCountTypeRadios = document.querySelectorAll('input[name="chapterCountType"]');
+    const chapterCountSelect = document.getElementById('chapterCount');
+    const customChapterCount = document.getElementById('customChapterCount');
+    const chapterCountContainer = chapterCountSelect.parentElement;
+
+    // Word count controls
+    const wordCountTypeRadios = document.querySelectorAll('input[name="wordCountType"]');
+    const wordCountSelect = document.getElementById('wordCount');
+    const customWordCount = document.getElementById('customWordCount');
+    const wordCountContainer = wordCountSelect.parentElement;
+
+    // Initialize UI state
+    initializeControls();
+
+    // Set up event listeners
+    setupEventListeners();
+
+    function initializeControls() {
+        // Structure options
+        automaticOptions.style.display = 
+            document.querySelector('input[name="structure"]:checked').value === 'automatic' 
+                ? 'block' 
+                : 'none';
+
+        // Citation style
+        citationStyleContainer.style.display = includeReferences.checked ? 'block' : 'none';
+        citationStyle.disabled = !includeReferences.checked;
+
+        // Chapter count
+        const chapterCountType = document.querySelector('input[name="chapterCountType"]:checked').value;
+        chapterCountContainer.style.display = chapterCountType === 'manual' ? 'flex' : 'none';
+        chapterCountSelect.disabled = chapterCountType !== 'manual';
+
+        // Word count
+        const wordCountType = document.querySelector('input[name="wordCountType"]:checked').value;
+        wordCountContainer.style.display = wordCountType === 'manual' ? 'flex' : 'none';
+        wordCountSelect.disabled = wordCountType !== 'manual';
+
+        // Add custom options
+        if (!chapterCountSelect.querySelector('option[value="custom"]')) {
+            chapterCountSelect.innerHTML += '<option value="custom">Custom...</option>';
+        }
+        if (!wordCountSelect.querySelector('option[value="custom"]')) {
+            wordCountSelect.innerHTML += '<option value="custom">Custom...</option>';
+        }
+    }
+
+    function setupEventListeners() {
+        // Structure radio changes
+        structureRadios.forEach(radio => {
+            radio.addEventListener('change', function() {
+                automaticOptions.style.display = this.value === 'automatic' ? 'block' : 'none';
+            });
+        });
+
+        // References checkbox
+        includeReferences.addEventListener('change', function() {
+            citationStyleContainer.style.display = this.checked ? 'block' : 'none';
+            citationStyle.disabled = !this.checked;
+        });
+
+        // Chapter count type
+        chapterCountTypeRadios.forEach(radio => {
+            radio.addEventListener('change', function() {
+                const isManual = this.value === 'manual';
+                chapterCountContainer.style.display = isManual ? 'flex' : 'none';
+                chapterCountSelect.disabled = !isManual;
+                customChapterCount.classList.add('hidden');
+                chapterCountSelect.classList.remove('hidden');
+            });
+        });
+
+        // Chapter count select
+        chapterCountSelect.addEventListener('change', function() {
+            if (this.value === 'custom') {
+                this.classList.add('hidden');
+                customChapterCount.classList.remove('hidden');
+            }
+        });
+
+        // Word count type
+        wordCountTypeRadios.forEach(radio => {
+            radio.addEventListener('change', function() {
+                const isManual = this.value === 'manual';
+                wordCountContainer.style.display = isManual ? 'flex' : 'none';
+                wordCountSelect.disabled = !isManual;
+                customWordCount.classList.add('hidden');
+                wordCountSelect.classList.remove('hidden');
+            });
+        });
+
+        // Word count select
+        wordCountSelect.addEventListener('change', function() {
+            if (this.value === 'custom') {
+                this.classList.add('hidden');
+                customWordCount.classList.remove('hidden');
+            }
+        });
+    }
+});
+
+function startGeneration(event) {
+    event.preventDefault();
+    
+    const form = event.target;
+    const abortBtn = document.getElementById('abortBtn');
+    const startBtn = document.getElementById('startBtn');
+    const progressContainer = document.getElementById('progressContainer');
+    const resultContainer = document.getElementById('resultContainer');
+    const errorContainer = document.getElementById('errorContainer');
+    
+    // Show/hide appropriate elements
+    startBtn.style.display = 'none';
+    abortBtn.style.display = 'block';
+    progressContainer.classList.remove('hidden');
+    resultContainer.classList.add('hidden');
+    errorContainer.classList.add('hidden');
+
+    // Get form data
+    const formData = new FormData(form);
+    const queryParams = new URLSearchParams(formData);
+
+    // Create SSE connection
+    currentEventSource = new EventSource(`/api/stream?${queryParams.toString()}`);
+    
+    currentEventSource.onmessage = function(event) {
+        const data = JSON.parse(event.data);
+        
+        if (data.task_id) {
+            currentTaskId = data.task_id;
+        }
+        
+        if (data.status === 'aborted') {
+            handleAbort();
+        }
+        
+        if (data.error) {
+            handleError(data.error);
+        }
+
+        updateProgress(data);
+    };
+    
+    currentEventSource.onerror = function() {
+        handleError('Connection error occurred');
+    };
+}
+
+function abortGeneration() {
+    if (currentTaskId) {
+        fetch(`/api/abort/${currentTaskId}`, {
+            method: 'POST'
+        }).catch(error => {
+            console.error('Failed to abort generation:', error);
+        });
+        
+        if (currentEventSource) {
+            currentEventSource.close();
+        }
+        handleAbort();
+    }
+}
+
+function handleAbort() {
+    currentEventSource?.close();
+    document.getElementById('abortBtn').style.display = 'none';
+    document.getElementById('startBtn').style.display = 'block';
+    document.getElementById('progressContainer').classList.add('hidden');
+    showError('Generation aborted');
+}
+
+function handleError(message) {
+    currentEventSource?.close();
+    document.getElementById('abortBtn').style.display = 'none';
+    document.getElementById('startBtn').style.display = 'block';
+    document.getElementById('progressContainer').classList.add('hidden');
+    showError(message);
+}
+
+function showError(message) {
+    const errorContainer = document.getElementById('errorContainer');
+    const errorMessage = document.getElementById('errorMessage');
+    errorMessage.textContent = message;
+    errorContainer.classList.remove('hidden');
+}
+
+function updateProgress(data) {
+    // Update progress bar
+    if (data.progress) {
+        const progressBar = document.getElementById('progressBar');
+        const progressText = document.getElementById('progressText');
+        progressBar.style.width = `${data.progress}%`;
+        progressText.textContent = `${Math.round(data.progress)}%`;
+    }
+
+    // Update steps
+    if (data.steps) {
+        // First render if steps don't exist
+        if (!document.getElementById('step-0')) {
+            renderSteps(data.steps);
+        }
+        updateSteps(data.steps);
+    }
+
+    // Update chapter progress if available
+    if (data.chapter_progress) {
+        updateChapterProgress(data.chapter_progress);
+    }
+
+    // Handle completion
+    if (data.status === 'complete' || data.status === 'partial_success') {
+        handleCompletion(data);
+    }
+}
+
+function renderSteps(steps) {
+    const progressSteps = document.getElementById('progressSteps');
+    progressSteps.innerHTML = ''; // Clear existing steps
+    
+    steps.forEach(step => {
+        const stepElement = document.createElement('div');
+        stepElement.className = 'flex items-start';
+        stepElement.id = `step-${step.id}`;
+        
+        let stepContent = `
+            <div class="flex-shrink-0 h-5 w-5 text-gray-400 mt-1">
+                <i class="far fa-circle" id="step-icon-${step.id}"></i>
+            </div>
+            <div class="ml-3">
+                <p class="text-sm font-medium text-gray-700" id="step-text-${step.id}">${step.text}</p>
+                <p class="text-xs text-gray-500 hidden" id="step-message-${step.id}"></p>
+        `;
+        
+        // Add sub-steps if they exist
+        if (step.subSteps && step.subSteps.length > 0) {
+            stepContent += `<div class="ml-4 mt-2 space-y-2" id="substeps-${step.id}">`;
+            step.subSteps.forEach(subStep => {
+                stepContent += `
+                    <div class="flex items-center">
+                        <div class="flex-shrink-0 h-4 w-4 text-gray-400">
+                            <i class="far fa-circle" id="substep-icon-${subStep.id}"></i>
+                        </div>
+                        <div class="ml-2">
+                            <p class="text-xs font-medium text-gray-700" id="substep-text-${subStep.id}">${subStep.text}</p>
+                            <p class="text-xs text-gray-500 hidden" id="substep-message-${subStep.id}"></p>
+                        </div>
+                    </div>
+                `;
+            });
+            stepContent += `</div>`;
+        }
+        
+        stepContent += `</div>`;
+        stepElement.innerHTML = stepContent;
+        progressSteps.appendChild(stepElement);
+    });
+}
+
+function updateSteps(steps) {
+    steps.forEach(step => {
+        const icon = document.getElementById(`step-icon-${step.id}`);
+        const message = document.getElementById(`step-message-${step.id}`);
+        
+        if (icon) {
+            icon.className = getStepIconClass(step.status);
+        }
+        
+        if (message && step.message) {
+            message.textContent = step.message;
+            message.classList.remove('hidden');
+        }
+
+        // Update sub-steps if they exist
+        if (step.subSteps) {
+            step.subSteps.forEach(subStep => {
+                const subIcon = document.getElementById(`substep-icon-${subStep.id}`);
+                const subMessage = document.getElementById(`substep-message-${subStep.id}`);
+                
+                if (subIcon) {
+                    subIcon.className = getStepIconClass(subStep.status);
+                }
+                
+                if (subMessage && subStep.message) {
+                    subMessage.textContent = subStep.message;
+                    subMessage.classList.remove('hidden');
+                }
+            });
+        }
+    });
+}
+
+function getStepIconClass(status) {
+    switch (status) {
+        case 'pending': return 'far fa-circle text-gray-400';
+        case 'in-progress': return 'fas fa-spinner fa-spin text-indigo-500';
+        case 'complete': return 'fas fa-check-circle text-green-500';
+        case 'error': return 'fas fa-exclamation-circle text-red-500';
+        default: return 'far fa-circle text-gray-400';
+    }
+}
+
+function handleCompletion(data) {
+    const resultContainer = document.getElementById('resultContainer');
+    const abortBtn = document.getElementById('abortBtn');
+    const startBtn = document.getElementById('startBtn');
+
+    if (data.docx_file) {
+        document.getElementById('downloadDocx').href = `/api/download/${data.docx_file}`;
+    }
+    if (data.md_file) {
+        document.getElementById('downloadMd').href = `/api/download/${data.md_file}`;
+    }
+    
+    resultContainer.classList.remove('hidden');
+    abortBtn.style.display = 'none';
+    startBtn.style.display = 'block';
+}
+
+// Handle page unload
+window.addEventListener('beforeunload', function() {
+    if (currentTaskId) {
+        abortGeneration();
+    }
+});
+
+function updateChapterProgress(chapterProgress) {
+    const container = document.getElementById('chapterProgressContainer');
+    const currentChapter = document.getElementById('currentChapter');
+    const chapterPercent = document.getElementById('chapterPercent');
+    const chapterProgressBar = document.getElementById('chapterProgressBar');
+    const chapterTime = document.getElementById('chapterTime');
+    const chapterStatus = document.getElementById('chapterStatus');
+    const chapterError = document.getElementById('chapterError');
+
+    container.classList.remove('hidden');
+
+    if (chapterProgress.chapter) {
+        currentChapter.textContent = `Chapter ${chapterProgress.current}/${chapterProgress.total}: ${chapterProgress.chapter}`;
+        chapterPercent.textContent = `${Math.round(chapterProgress.percent)}%`;
+        chapterProgressBar.style.width = `${chapterProgress.percent}%`;
+        
+        if (chapterProgress.duration) {
+            chapterTime.textContent = `Time taken: ${chapterProgress.duration}`;
+        }
+        
+        if (chapterProgress.error) {
+            chapterStatus.textContent = 'Error';
+            chapterError.textContent = chapterProgress.error;
+            chapterError.classList.remove('hidden');
+        } else {
+            chapterStatus.textContent = 'In progress';
+            chapterError.classList.add('hidden');
+        }
+    }
+
+    if (chapterProgress.complete) {
+        chapterStatus.textContent = `Completed ${chapterProgress.total_chapters} chapters`;
+    }
+} 

templates/index.html

@@ -6,6 +6,7 @@
     <title>Research Paper Generator</title>
     <script src="https://cdn.tailwindcss.com"></script>
     <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css">
+    <script src="{{ url_for('static', filename='js/main.js') }}" defer></script>
 </head>
 <body class="bg-gray-50 min-h-screen">
     <div class="container mx-auto px-4 py-8">
@@ -140,9 +141,11 @@
                 </div>
 
                 <div class="pt-4">
-                    <button type="submit" id="generateBtn"
-                        class="w-full flex justify-center items-center py-3 px-4 border border-transparent rounded-md shadow-sm text-sm font-medium text-white bg-indigo-600 hover:bg-indigo-700 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-indigo-500 disabled:opacity-50 disabled:cursor-not-allowed">
-                        <i class="fas fa-magic mr-2"></i> Generate Paper
+                    <button type="submit" id="startBtn" class="inline-flex items-center px-4 py-2 border border-transparent text-sm font-medium rounded-md shadow-sm text-white bg-indigo-600 hover:bg-indigo-700">
+                        <i class="fas fa-play mr-2"></i> Generate Paper
+                    </button>
+                    <button type="button" id="abortBtn" style="display: none;" class="inline-flex items-center px-4 py-2 border border-transparent text-sm font-medium rounded-md shadow-sm text-white bg-red-600 hover:bg-red-700">
+                        <i class="fas fa-stop mr-2"></i> Stop Generation
                     </button>
                 </div>
             </form>
@@ -217,321 +220,5 @@
             </div>
         </div>
     </div>
-
-    <script>
-        document.getElementById('paperForm').addEventListener('submit', function(e) {
-            e.preventDefault();
-            
-            const form = e.target;
-            const generateBtn = document.getElementById('generateBtn');
-            const progressContainer = document.getElementById('progressContainer');
-            const chapterProgressContainer = document.getElementById('chapterProgressContainer');
-            const resultContainer = document.getElementById('resultContainer');
-            const errorContainer = document.getElementById('errorContainer');
-            const progressSteps = document.getElementById('progressSteps');
-            const progressBar = document.getElementById('progressBar');
-            const progressText = document.getElementById('progressText');
-            
-            // Chapter progress elements
-            const currentChapterEl = document.getElementById('currentChapter');
-            const chapterPercentEl = document.getElementById('chapterPercent');
-            const chapterProgressBar = document.getElementById('chapterProgressBar');
-            const chapterTimeEl = document.getElementById('chapterTime');
-            const chapterStatusEl = document.getElementById('chapterStatus');
-            const chapterErrorEl = document.getElementById('chapterError');
-            
-            // Reset UI
-            generateBtn.disabled = true;
-            progressContainer.classList.remove('hidden');
-            chapterProgressContainer.classList.add('hidden');
-            resultContainer.classList.add('hidden');
-            errorContainer.classList.add('hidden');
-            progressSteps.innerHTML = '';
-            progressBar.style.width = '0%';
-            progressText.textContent = '0%';
-            chapterProgressBar.style.width = '0%';
-            chapterPercentEl.textContent = '0%';
-            chapterErrorEl.classList.add('hidden');
-            
-            // Initial steps (will be updated if automatic structure)
-            let steps = [
-                {id: 0, text: 'Preparing document structure...', status: 'pending'},
-                {id: 1, text: 'Generating index/table of contents...', status: 'pending'},
-                {id: 2, text: 'Determining chapters...', status: 'pending'},
-                {id: 3, text: 'Writing content...', status: 'pending', subSteps: []},
-                {id: 4, text: 'Finalizing document...', status: 'pending'},
-                {id: 5, text: 'Converting to Word format...', status: 'pending'}
-            ];
-            
-            // Function to render steps
-            function renderSteps() {
-                progressSteps.innerHTML = '';
-                steps.forEach(step => {
-                    const stepElement = document.createElement('div');
-                    stepElement.className = 'flex items-start';
-                    
-                    let stepContent = `
-                        <div class="flex-shrink-0 h-5 w-5 text-gray-400 mt-1">
-                            <i class="far fa-circle" id="step-icon-${step.id}"></i>
-                        </div>
-                        <div class="ml-3">
-                            <p class="text-sm font-medium text-gray-700" id="step-text-${step.id}">${step.text}</p>
-                            <p class="text-xs text-gray-500 hidden" id="step-message-${step.id}"></p>
-                    `;
-                    
-                    // Add sub-steps if they exist
-                    if (step.subSteps && step.subSteps.length > 0) {
-                        stepContent += `<div class="ml-4 mt-2 space-y-2" id="substeps-${step.id}">`;
-                        step.subSteps.forEach(subStep => {
-                            stepContent += `
-                                <div class="flex items-center">
-                                    <div class="flex-shrink-0 h-4 w-4 text-gray-400">
-                                        <i class="far fa-circle" id="substep-icon-${subStep.id}"></i>
-                                    </div>
-                                    <div class="ml-2">
-                                        <p class="text-xs font-medium text-gray-700" id="substep-text-${subStep.id}">${subStep.text}</p>
-                                        <p class="text-xs text-gray-500 hidden" id="substep-message-${subStep.id}"></p>
-                                    </div>
-                                </div>
-                            `;
-                        });
-                        stepContent += `</div>`;
-                    }
-                    
-                    stepContent += `</div>`;
-                    stepElement.innerHTML = stepContent;
-                    stepElement.id = `step-${step.id}`;
-                    progressSteps.appendChild(stepElement);
-                });
-            }
-            
-            // Initial render
-            renderSteps();
-            
-            // Get form data
-            const formData = {
-                subject: form.subject.value,
-                model: form.model.value,
-                structure: form.structure.value
-            };
-            
-            // Create SSE connection
-            const eventSource = new EventSource(`/stream?subject=${encodeURIComponent(formData.subject)}&model=${encodeURIComponent(formData.model)}&structure=${encodeURIComponent(formData.structure)}`);
-            
-            eventSource.onmessage = function(event) {
-                const data = JSON.parse(event.data);
-                
-                if (data.error) {
-                    document.getElementById('errorMessage').textContent = data.error;
-                    errorContainer.classList.remove('hidden');
-                    eventSource.close();
-                    generateBtn.disabled = false;
-                    return;
-                }
-                
-                // Handle chapter progress updates
-                if (data.chapter_progress) {
-                    chapterProgressContainer.classList.remove('hidden');
-                    const cp = data.chapter_progress;
-                    
-                    if (cp.chapter) {
-                        currentChapterEl.textContent = `Chapter ${cp.current}/${cp.total}: ${cp.chapter}`;
-                        chapterPercentEl.textContent = `${Math.round(cp.percent)}%`;
-                        chapterProgressBar.style.width = `${cp.percent}%`;
-                        chapterStatusEl.textContent = cp.error ? 'Error' : 'In progress';
-                        
-                        if (cp.duration) {
-                            chapterTimeEl.textContent = `Time taken: ${cp.duration}`;
-                        }
-                        
-                        if (cp.error) {
-                            chapterErrorEl.textContent = cp.error;
-                            chapterErrorEl.classList.remove('hidden');
-                        } else {
-                            chapterErrorEl.classList.add('hidden');
-                        }
-                    }
-                    
-                    if (cp.complete) {
-                        chapterStatusEl.textContent = `Completed ${cp.total_chapters} chapters`;
-                    }
-                }
-                
-                if (data.update_steps) {
-                    // Update steps if structure changed (automatic mode)
-                    steps = data.steps;
-                    renderSteps();
-                }
-                
-                if (data.status === 'complete' || data.status === 'partial_success') {
-                    // Show download links
-                    if (data.docx_file) {
-                        document.getElementById('downloadDocx').href = `/download/${data.docx_file}`;
-                    }
-                    if (data.md_file) {
-                        document.getElementById('downloadMd').href = `/download/${data.md_file}`;
-                    }
-                    
-                    resultContainer.classList.remove('hidden');
-                    eventSource.close();
-                    generateBtn.disabled = false;
-                }
-                
-                // Update progress
-                if (data.progress) {
-                    progressBar.style.width = `${data.progress}%`;
-                    progressText.textContent = `${Math.round(data.progress)}%`;
-                }
-                
-                // Update steps
-                if (data.steps) {
-                    data.steps.forEach(step => {
-                        const icon = document.getElementById(`step-icon-${step.id}`);
-                        const text = document.getElementById(`step-text-${step.id}`);
-                        const message = document.getElementById(`step-message-${step.id}`);
-                        
-                        if (icon) {
-                            if (step.status === 'pending') {
-                                icon.className = 'far fa-circle text-gray-400';
-                            } else if (step.status === 'in-progress') {
-                                icon.className = 'fas fa-spinner fa-spin text-indigo-500';
-                            } else if (step.status === 'complete') {
-                                icon.className = 'fas fa-check-circle text-green-500';
-                            } else if (step.status === 'error') {
-                                icon.className = 'fas fa-exclamation-circle text-red-500';
-                            }
-                        }
-                        
-                        if (message && step.message) {
-                            message.textContent = step.message;
-                            message.classList.remove('hidden');
-                        }
-                        
-                        // Update sub-steps
-                        if (step.subSteps) {
-                            step.subSteps.forEach(subStep => {
-                                const subIcon = document.getElementById(`substep-icon-${subStep.id}`);
-                                const subText = document.getElementById(`substep-text-${subStep.id}`);
-                                const subMessage = document.getElementById(`substep-message-${subStep.id}`);
-                                
-                                if (subIcon) {
-                                    if (subStep.status === 'pending') {
-                                        subIcon.className = 'far fa-circle text-gray-400';
-                                    } else if (subStep.status === 'in-progress') {
-                                        subIcon.className = 'fas fa-spinner fa-spin text-indigo-500';
-                                    } else if (subStep.status === 'complete') {
-                                        subIcon.className = 'fas fa-check-circle text-green-500';
-                                    } else if (subStep.status === 'error') {
-                                        subIcon.className = 'fas fa-exclamation-circle text-red-500';
-                                    }
-                                }
-                                
-                                if (subMessage && subStep.message) {
-                                    subMessage.textContent = subStep.message;
-                                    subMessage.classList.remove('hidden');
-                                }
-                            });
-                        }
-                    });
-                }
-            };
-            
-            eventSource.onerror = function() {
-                eventSource.close();
-                generateBtn.disabled = false;
-            };
-        });
-
-        document.addEventListener('DOMContentLoaded', function() {
-            // Get elements
-            const structureRadios = document.querySelectorAll('input[name="structure"]');
-            const automaticOptions = document.getElementById('automaticOptions');
-            const includeReferences = document.getElementById('includeReferences');
-            const citationStyle = document.getElementById('citationStyle');
-            const citationStyleContainer = citationStyle.parentElement;
-
-            // Handle structure radio changes
-            structureRadios.forEach(radio => {
-                radio.addEventListener('change', function() {
-                    automaticOptions.style.display = 
-                        this.value === 'automatic' ? 'block' : 'none';
-                });
-            });
-
-            // Initialize automatic options visibility
-            automaticOptions.style.display = 
-                document.querySelector('input[name="structure"]:checked').value === 'automatic' 
-                    ? 'block' 
-                    : 'none';
-
-            // Handle references checkbox
-            includeReferences.addEventListener('change', function() {
-                citationStyleContainer.style.display = this.checked ? 'block' : 'none';
-                citationStyle.disabled = !this.checked;
-            });
-            // Initialize citation style visibility
-            citationStyleContainer.style.display = includeReferences.checked ? 'block' : 'none';
-
-            // Chapter count controls
-            const chapterCountTypeRadios = document.querySelectorAll('input[name="chapterCountType"]');
-            const chapterCountSelect = document.getElementById('chapterCount');
-            const customChapterCount = document.getElementById('customChapterCount');
-            const chapterCountContainer = chapterCountSelect.parentElement;
-
-            chapterCountTypeRadios.forEach(radio => {
-                radio.addEventListener('change', function() {
-                    const isManual = this.value === 'manual';
-                    chapterCountContainer.style.display = isManual ? 'flex' : 'none';
-                    chapterCountSelect.disabled = !isManual;
-                    customChapterCount.classList.toggle('hidden', true);
-                    chapterCountSelect.classList.toggle('hidden', false);
-                });
-            });
-            // Initialize chapter count visibility
-            chapterCountContainer.style.display = 
-                document.querySelector('input[name="chapterCountType"]:checked').value === 'manual' 
-                    ? 'flex' 
-                    : 'none';
-
-            chapterCountSelect.addEventListener('change', function() {
-                if (this.value === 'custom') {
-                    this.classList.toggle('hidden', true);
-                    customChapterCount.classList.toggle('hidden', false);
-                }
-            });
-
-            // Word count controls
-            const wordCountTypeRadios = document.querySelectorAll('input[name="wordCountType"]');
-            const wordCountSelect = document.getElementById('wordCount');
-            const customWordCount = document.getElementById('customWordCount');
-            const wordCountContainer = wordCountSelect.parentElement;
-
-            wordCountTypeRadios.forEach(radio => {
-                radio.addEventListener('change', function() {
-                    const isManual = this.value === 'manual';
-                    wordCountContainer.style.display = isManual ? 'flex' : 'none';
-                    wordCountSelect.disabled = !isManual;
-                    customWordCount.classList.toggle('hidden', true);
-                    wordCountSelect.classList.toggle('hidden', false);
-                });
-            });
-            // Initialize word count visibility
-            wordCountContainer.style.display = 
-                document.querySelector('input[name="wordCountType"]:checked').value === 'manual' 
-                    ? 'flex' 
-                    : 'none';
-
-            wordCountSelect.addEventListener('change', function() {
-                if (this.value === 'custom') {
-                    this.classList.toggle('hidden', true);
-                    customWordCount.classList.toggle('hidden', false);
-                }
-            });
-
-            // Add custom option to selects
-            chapterCountSelect.innerHTML += '<option value="custom">Custom...</option>';
-            wordCountSelect.innerHTML += '<option value="custom">Custom...</option>';
-        });
-    </script>
 </body>
 </html>