Upload 8 files

index.html

@@ -40,8 +40,15 @@
     </div>
     
     <button onclick="uploadFile()" style="margin-top:10px; width:100%;">Process Jungian Tree</button>
+    
     <div id="status" style="font-size: 0.8em; margin-top: 10px; color: var(--accent);">Status: Idle</div>
 
+    <!-- Download Button -->
+        <button id="downloadBtn" style="display:none; margin-top:10px; background:#2aa198; color:white; border:none; padding:10px; cursor:pointer; width:100%;">
+            📥 Download Knowledge Tree
+        </button>
+    </div>
+
     <div id="tree-container"></div> 
 
 </div>
@@ -56,13 +63,19 @@ async function uploadFile() {
     const isWhole = document.querySelector('input[name="mode"]:checked').value === 'whole';
     const status = document.getElementById('status');
     
+    // Explicitly grab the values from the inputs
+    const startVal = document.getElementById('startP').value;
+    const endVal = document.getElementById('endP').value;
+
     if (fileInput.files.length === 0) return alert("Please select a PDF");
 
     const formData = new FormData();
     formData.append('file', fileInput.files[0]);
     formData.append('whole', isWhole);
-    formData.append('start', document.getElementById('startP').value || 20);
-    formData.append('end', document.getElementById('endP').value || 30);
+    formData.append('start', startVal || 20); // Fallback to 20 if empty
+    formData.append('end', endVal || 30);     // Fallback to 30 if empty
+
+    console.log("Sending Range:", startVal, "to", endVal); // Check browser console
 
     status.innerText = "Status: Uploading...";
     await fetch('/upload', { method: 'POST', body: formData });
@@ -71,10 +84,18 @@ async function uploadFile() {
     listenToStream();
 }
 
+// --- VIEW function ---
+function view(title, text) {
+    const display = document.getElementById('content-display');
+    display.innerHTML = `<h2>${title}</h2><div style="white-space: pre-wrap;">${text}</div>`;
+}
+
+// --- listenToStream Function --
 function listenToStream() {
     const eventSource = new EventSource('/stream');
     const container = document.getElementById('tree-container');
     const status = document.getElementById('status');
+    const dlBtn = document.getElementById('downloadBtn');
     container.innerHTML = ""; 
 
     eventSource.onmessage = (event) => {
@@ -82,6 +103,8 @@ function listenToStream() {
         
         if (data.type === 'done') {
             status.innerText = "✅ EXTRACTION FINISHED!";
+            dlBtn.style.display = "block"; // Show the button
+            dlBtn.onclick = () => window.location.href = '/download-latest';
             eventSource.close();
             return;
         }
@@ -90,11 +113,13 @@ function listenToStream() {
         const node = document.createElement('div');
         node.className = data.type === 'summary' ? "tree-node summary-block" : "tree-node leaf-node";
         node.innerHTML = (data.type === 'summary' ? "⭐ " : "∟ ") + (data.name || data.filename);
+        // Pass the content to the view function when clicked
         node.onclick = () => view(data.name || data.filename, data.content);
         container.appendChild(node);
         status.innerText = `Status: Created ${data.type}...`;
     };
 }
+
 </script>
 
 

main.py

@@ -6,10 +6,12 @@ import os
 import asyncio
 import json
 from fastapi import FastAPI, UploadFile, File, Form, BackgroundTasks
-from fastapi.responses import HTMLResponse, StreamingResponse
+from fastapi.responses import HTMLResponse, StreamingResponse, FileResponse # Added FileResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi import Form # Add Form to your imports
 import shutil
+import glob
+
 
 # Import chunking logic from the existing combined script
 # Note: Ensure script functions are wrap-able or callable
@@ -82,6 +84,15 @@ async def handle_upload(
     ))
     return {"status": "Processing started"}
 
+@app.get("/download-latest")
+async def download_latest():
+    # Look for files matching our pattern
+    files = glob.glob("knowledge_tree_*.json")
+    if not files: 
+        return {"error": "No JSON files found yet. Finish an extraction first."}
+    # Sort by creation time to get the newest one
+    latest_file = max(files, key=os.path.getctime)
+    return FileResponse(path=latest_file, filename=os.path.basename(latest_file))
 
 
 if __name__ == "__main__":

phase0102_chunker_aggregator_2.py

@@ -99,7 +99,7 @@ async def call_groq_json(system_prompt, user_content):
     )
     return json.loads(completion.choices[0].message.content)
     """
-    
+
 #async def run_chunking_process(pdf_path, queue=None, whole=False, start_p=20, end_p=30):
 async def run_chunking_process(pdf_path, queue=None, whole=WHOLE, start_p=START_PAGE, end_p=END_PAGE):
     """
@@ -170,7 +170,7 @@ async def run_chunking_process(pdf_path, queue=None, whole=WHOLE, start_p=START_
             # PHASE II: AGGREGATION
             if len(temp_group) >= CHUNK_GROUP_SIZE:
                 from phase0102_chunker_aggregator_2 import generate_summary_block # Ensure helper is available
-                summary_res = generate_summary_block(temp_group)
+                summary_res = await generate_summary_block(temp_group)
                 
                 summary_node = {
                     "type": "summary",
@@ -214,7 +214,7 @@ async def run_chunking_process(pdf_path, queue=None, whole=WHOLE, start_p=START_
         await queue.put("DONE")
 
 # Helper for summary
-def generate_summary_block(chunks):
+async def generate_summary_block(chunks):
     combined = "\n\n".join([f"{c['filename']}: {c['content']}" for c in chunks])
-    prompt = "Synthesize these Jungian chunks into a dense Level-1 summary. JSON keys: 'summary_name', 'synthesis'."
-    return call_groq_json(prompt, combined)
+    prompt = "Synthesize these Jungian chunks into a single high-density Level-1 summary. JSON keys: 'summary_name', 'synthesis'."
+    return await call_groq_json(prompt, combined)