feat: favicon + copy paste + open links in new tabs

app/main.py

@@ -37,7 +37,7 @@ app = FastAPI()
 templates = Jinja2Templates(directory="app/templates")
 
 
-from fastapi.responses import PlainTextResponse, HTMLResponse
+from fastapi.responses import PlainTextResponse, HTMLResponse, FileResponse
 
 # # Add a helper function to validate CVE-ID
 # def validate_cve_id(cve_id: str):
@@ -236,6 +236,17 @@ async def ghsa_to_cve_endpoint(ghsa_id: str):
         raise HTTPException(status_code=500, detail="Internal server error.")
 
 
+@app.get("/favicon.ico", include_in_schema=False)
+def favicon():
+    """
+    Endpoint to serve the favicon.
+
+    Returns:
+        FileResponse: The favicon image file.
+    """
+    return FileResponse("app/templates/favicon.png")
+
+
 mcp = FastApiMCP(
     app,  
     name="Get CVE Info",  # Name for your MCP server

app/templates/index.html

@@ -36,11 +36,38 @@
   <script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
 </head>
 <body>
+  <button id="copyTop" style="position: absolute; top: 10px; left: 10px;">Copy</button>
   <h2>🛡️ CVE Report</h2>
   <div id="spinner">⏳ Loading report...</div>
   <div id="output"></div>
+  <button id="copyBottom" style="position: absolute; bottom: 10px; left: 10px;">Copy</button>
 
   <script>
+    function copyToClipboard() {
+      const outputEl = document.getElementById("output");
+      const text = outputEl.innerHTML.replace(/<h1>(.*?)<\/h1>/g, '# $1')
+                                     .replace(/<h2>(.*?)<\/h2>/g, '## $1')
+                                     .replace(/<h3>(.*?)<\/h3>/g, '### $1')
+                                     .replace(/<h4>(.*?)<\/h4>/g, '#### $1')
+                                     .replace(/<h5>(.*?)<\/h5>/g, '##### $1')
+                                     .replace(/<h6>(.*?)<\/h6>/g, '###### $1')
+                                     .replace(/<strong>(.*?)<\/strong>/g, '**$1**')
+                                     .replace(/<em>(.*?)<\/em>/g, '*$1*')
+                                     .replace(/<ul>/g, '\n')
+                                     .replace(/<li>(.*?)<\/li>/g, '- $1')
+                                     .replace(/<\/ul>/g, '\n')
+                                     .replace(/<br\s*\/?>/g, '\n')
+                                     .replace(/<[^>]+>/g, ''); // Remove other HTML tags
+      navigator.clipboard.writeText(text).then(() => {
+        alert("Content copied to clipboard!");
+      }).catch(err => {
+        console.error("Failed to copy content: ", err);
+      });
+    }
+
+    document.getElementById("copyTop").addEventListener("click", copyToClipboard);
+    document.getElementById("copyBottom").addEventListener("click", copyToClipboard);
+
     document.addEventListener("DOMContentLoaded", async () => {
       const params = new URLSearchParams(window.location.search);
       const cveId = params.get("cve_id");
@@ -60,6 +87,14 @@
           const text = await res.text();
           spinnerEl.style.display = "none";
           outputEl.innerHTML = marked.parse(text);
+
+          // Ensure links open in a new tab
+          const links = outputEl.querySelectorAll("a");
+          links.forEach(link => {
+            link.setAttribute("target", "_blank");
+            link.setAttribute("rel", "noopener noreferrer");
+          });
+
           outputEl.style.display = "block";
         } catch {
           startStreaming();
@@ -74,6 +109,14 @@
         evtSource.onmessage = (e) => {
           buffer += e.data;
           outputEl.innerHTML = marked.parse(buffer);
+
+          // Ensure links open in a new tab
+          const links = outputEl.querySelectorAll("a");
+          links.forEach(link => {
+            link.setAttribute("target", "_blank");
+            link.setAttribute("rel", "noopener noreferrer");
+          });
+
           outputEl.style.display = "block";
           outputEl.scrollTop = outputEl.scrollHeight;
         };
@@ -83,6 +126,14 @@
           buffer += e.data;
           spinnerEl.style.display = "none";
           outputEl.innerHTML = marked.parse(buffer);
+
+          // Ensure links open in a new tab
+          const links = outputEl.querySelectorAll("a");
+          links.forEach(link => {
+            link.setAttribute("target", "_blank");
+            link.setAttribute("rel", "noopener noreferrer");
+          });
+
           outputEl.scrollTop = outputEl.scrollHeight;
         });