fix: stream

app/main.py

@@ -84,19 +84,27 @@ async def stream(cve_id: str, refresh: bool = Query(False)):
 
     async def yield_live():
         full_text = ""
+        logger.info(f"Starting OpenAI stream for {cve_id}")
         try:
             async for chunk in generate_openai_stream(cve_id):
+                logger.debug(f"Got chunk: {repr(chunk)}")
                 full_text += chunk
+                for line in chunk.splitlines():
+                    yield f"data: {line}\n"
+                yield "\n"
+            logger.info(f"Completed OpenAI stream for {cve_id}, preview: {repr(full_text[:200])}")
             await write_report(cve_id, full_text)
             yield f"event: end\ndata: {full_text}\n\n"
+        except Exception as e:
+            logger.exception(f"Error during OpenAI stream: {e}")
+            yield f"data: ❌ Error generating report: {str(e)}\n\n"
         finally:
-            os.remove(lock_path)
+            os.remove(_lock_path(cve_id))
 
     return StreamingResponse(yield_live(), media_type="text/event-stream")
 
 
 
-
 @app.get("/json")
 async def json_analysis(cve_id: str, refresh: bool = Query(False)):
     logger.info(f"Received JSON analysis request for cve_id={cve_id}, refresh={refresh}")

app/templates/index.html

@@ -46,34 +46,56 @@
       const cveId = params.get("cve_id");
       const outputEl = document.getElementById("output");
       const spinnerEl = document.getElementById("spinner");
-
+  
       if (!cveId) {
         spinnerEl.innerText = "❌ Missing CVE ID.";
         return;
       }
-
-      try {
-        const response = await fetch(`/raw?cve_id=${cveId}`);
-        if (!response.ok) {
-          spinnerEl.innerText = "❌ Report not found.";
-          return;
+  
+      async function fetchRawAndRender() {
+        try {
+          const res = await fetch(`/raw?cve_id=${cveId}`);
+          if (!res.ok) throw new Error("Report missing");
+          const text = await res.text();
+          spinnerEl.style.display = "none";
+          outputEl.innerHTML = marked.parse(text);
+          outputEl.style.display = "block";
+        } catch {
+          startStreaming();
         }
-
-        const markdown = await response.text();
-
-        // DEBUG: show what's fetched
-        console.log("Fetched markdown:", markdown.slice(0, 100));
-
-        // Parse and render markdown
-        outputEl.innerHTML = marked.parse(markdown);
-        spinnerEl.style.display = "none";
-        outputEl.style.display = "block";
-
-      } catch (err) {
-        console.error(err);
-        spinnerEl.innerText = "❌ Failed to load report.";
       }
+  
+      function startStreaming() {
+        spinnerEl.innerText = "⏳ Generating report...";
+        const evtSource = new EventSource(`/stream?cve_id=${cveId}&refresh=true`);
+        let buffer = "";
+  
+        evtSource.onmessage = (e) => {
+          buffer += e.data;
+          outputEl.innerHTML = marked.parse(buffer);
+          outputEl.style.display = "block";
+          outputEl.scrollTop = outputEl.scrollHeight;
+        };
+  
+        evtSource.addEventListener("end", (e) => {
+          evtSource.close();
+          buffer += e.data;
+          spinnerEl.style.display = "none";
+          outputEl.innerHTML = marked.parse(buffer);
+          outputEl.scrollTop = outputEl.scrollHeight;
+        });
+  
+        evtSource.onerror = (err) => {
+          console.error("Streaming error", err);
+          evtSource.close();
+          spinnerEl.innerText = "❌ Failed to generate report.";
+        };
+      }
+  
+      fetchRawAndRender();
     });
   </script>
+  
+  
 </body>
 </html>