Add deep/quick search option

app.py

@@ -223,6 +223,7 @@ class BatchDocResponse(BaseModel):
 
 class KeywordRequest(BaseModel):
     keywords: str
+    search_mode: Literal["quick", "deep"]
     case_sensitive: Optional[bool] = False
     release: Optional[str] = None
     wg: Optional[str] = None
@@ -461,6 +462,7 @@ def search_spec(request: KeywordRequest):
         release = request.release
         working_group = request.wg
         spec_type = request.spec_type
+        search_mode = request.search_mode
 
         if spec.get('version', None) is None or (release is not None and spec["version"].split(".")[0] != str(release)):
             continue
@@ -471,31 +473,35 @@ def search_spec(request: KeywordRequest):
 
         contents = []
         version = finder_spec.search_document(spec['id'], spec['release']).split("/")[-1].replace(".zip", "").split("-")[-1]
-        doc = finder_spec.get_document(spec['id'], version)
-        docValid = not isinstance(doc, str)
+        if search_mode == "deep":
+            doc = finder_spec.get_document(spec['id'], version)
+            docValid = not isinstance(doc, str)
 
         if request.mode == "and":
             if all(kw in willLower(string, booleanLowered) for kw in kws):
                 put = True
-            if docValid:
-                for chapter in list(doc.keys())[1:]:
-                    if "references" not in chapter.lower() or "void" not in chapter.lower():
-                        if all(kw in willLower(doc[chapter], booleanLowered) for kw in kws):
-                            put = True
-                            contents.append(chapter)    
+            if search_mode == "deep":
+                if docValid:
+                    for chapter in list(doc.keys())[1:]:
+                        if "references" not in chapter.lower() or "void" not in chapter.lower() or "annexe" not in doc[chapter].lower():
+                            if all(kw in willLower(doc[chapter], booleanLowered) for kw in kws):
+                                put = True
+                                contents.append(chapter)    
         elif request.mode == "or":
             if any(kw in willLower(string, booleanLowered) for kw in kws):
                 put = True
-            if docValid:
-                for chapter in list(doc.keys())[1:]:
-                    if "references" not in chapter.lower() or "void" not in chapter.lower():
-                        if any(kw in willLower(doc[chapter], booleanLowered) for kw in kws):
-                            put = True
-                            contents.append(chapter)
+            if search_mode == "deep":
+                if docValid:
+                    for chapter in list(doc.keys())[1:]:
+                        if "references" not in chapter.lower() or "void" not in chapter.lower() or "annexe" not in doc[chapter].lower():
+                            if any(kw in willLower(doc[chapter], booleanLowered) for kw in kws):
+                                put = True
+                                contents.append(chapter)   
 
         if put:
             spec_content = spec
-            spec_content["contains"] = {chap: doc[chap] for chap in contents}
+            if search_mode == "deep":
+                spec_content["contains"] = {chap: doc[chap] for chap in contents}
             results.append(spec_content)
         else:
             unique_specs.add(spec['id'])

static/script.js

@@ -19,6 +19,7 @@ const modeFilter = document.querySelector("select[name=mode]")
 const specTypeFilter = document.querySelector("select[name=spec_type]")
 const workingGroupFilter = document.querySelector("select[name=working_group]")
 const caseSensitiveFilter = document.querySelector("input[name=case_sensitive]")
+const searchMode = document.querySelector("select[name=search_mode]")
 
 const searchBtn = document.getElementById('search-btn');
 const batchSearchBtn = document.getElementById('batch-search-btn');
@@ -82,7 +83,14 @@ document.getElementById('toggleFilters').onclick = function() {
 };
 
 keywordSearchBtn.addEventListener("click", async ()=>{
-    const keywords = keywordInput.value.trim();
+    let keywords = keywordInput.value.trim();
+    let release = releaseFilter.value;
+    let wg = workingGroupFilter.value;
+    let specType = specTypeFilter.value;
+    let search = searchMode.value;
+    let checked = caseSensitiveFilter.checked;
+    let mode = modeFilter.value;
+
     if (!keywords) {
         showError("Please enter at least one keyword");
         return;
@@ -94,12 +102,13 @@ keywordSearchBtn.addEventListener("click", async ()=>{
     try{
         let body = {
             keywords,
-            "case_sensitive": caseSensitiveFilter.checked,
-            "mode": modeFilter.value
+            "search_mode": search,
+            "case_sensitive": checked,
+            "mode": mode
         };
-        if (releaseFilter.value != ""){body.release = releaseFilter.value}
-        if (workingGroupFilter.value != ""){body["working_group"] = workingGroupFilter.value}
-        if (specTypeFilter.value != ""){body["spec_type"] = specTypeFilter.value}
+        if (release != ""){body.release = release}
+        if (wg != ""){body["working_group"] = wg}
+        if (specType != ""){body["spec_type"] = specType}
         const response = await fetch("/search-spec", {
             method: "POST",
             headers: {
@@ -110,7 +119,7 @@ keywordSearchBtn.addEventListener("click", async ()=>{
 
         const data = await response.json();
         if (response.ok){
-            displayKeywordResults(data);
+            displayKeywordResults(data, search);
         } else {
             showError('Error processing batch request');
         }
@@ -244,7 +253,7 @@ function displaySingleNotFound(docId, message) {
     resultsContainer.style.display = 'block';
 }
 
-function displayKeywordResults(data) {
+function displayKeywordResults(data, mode) {
     resultsList.innerHTML = '';
     
     data.results.forEach(spec => {
@@ -265,19 +274,24 @@ function displayKeywordResults(data) {
                 <p>URL: <a target="_blank" href="${spec.url}">${spec.url}</a></p>
                 <p>Scope: ${spec.scope}</p>
             </div>
-            <div class="result-actions">
-                <button class="get-section-btn btn" data-spec-id="${spec.id}">Get section</button>
-            </div>
         `;
+
+        if(mode == "deep"){
+            resultItem.innerHTML += `
+                <div class="result-actions">
+                    <button class="get-section-btn btn" data-spec-id="${spec.id}">Get section</button>
+                </div>
+            `
+        }
         
         // Ajouter le bouton au DOM
         resultsList.appendChild(resultItem);
         
         // Récupérer le bouton nouvellement créé
-        const button1 = resultItem.querySelector('.get-section-btn');
-        
-        // Stocker l'objet directement sur l'élément DOM
-        button1._sections = spec.contains;
+        if(mode == "deep"){
+            const button1 = resultItem.querySelector('.get-section-btn');
+            button1._sections = spec.contains;
+        }
     });
 
     document.querySelectorAll('.get-section-btn').forEach(button => {

templates/index.html

@@ -51,8 +51,11 @@
                 </div>
 
                 <div class="input-group keyword-input">
-                    <label for="keywords">Keywords</label>
                     <div class="input-field">
+                        <select name="search_mode" class="filter-select" id="search-mode">
+                            <option value="quick">Quick Search (only title & scope)</option>
+                            <option value="deep">Deep Search</option>
+                        </select>
                         <input type="text" id="keywords" placeholder="Enter your keywords separated by space">
                         <button id="keyword-search-btn" class="btn">Search</button>
                     </div>