Update app.py

app.py

@@ -103,27 +103,19 @@ def headers_with_location(doc, llm_headers):
     Always include all headers, even if location not found.
     """
     headersJson = []
-    print(f"DEBUG: Processing {len(llm_headers)} LLM headers in headers_with_location")
-    
-    for i, h in enumerate(llm_headers):
-        text = h.get("text", "")
-        llm_page = h.get("page", 0)
-        suggested_level = h.get("suggested_level")
-        confidence = h.get("confidence", 1.0)
-        
-        print(f"DEBUG: Header {i}: '{text}' on page {llm_page}")
-        
+
+    for h in llm_headers:
+        text = h["text"]
+        llm_page = h["page"]
+
         # Attempt to locate the header on the page
-        locations = getLocation_of_header(doc, text, llm_page)
-        
-        print(f"DEBUG: Found {len(locations)} locations for '{text}'")
-        
+        locations = getLocation_of_header(doc, text,llm_page)
+
         if locations:
             for loc in locations:
                 page = doc.load_page(loc["page"])
                 fontsize = None
-                
-                # Try to find fontsize
+
                 for block in page.get_text("dict")["blocks"]:
                     if block.get("type") != 0:
                         continue
@@ -134,36 +126,19 @@ def headers_with_location(doc, llm_headers):
                             break
                     if fontsize:
                         break
-                
                 entry = [
                     text,
                     fontsize,
                     loc["page"],
                     loc["y"],
-                    suggested_level,
-                    confidence
+                    h["suggested_level"],
+                    
                 ]
                 if entry not in headersJson:
                     headersJson.append(entry)
-                    print(f"DEBUG: Added header with location: page={loc['page']}, y={loc['y']}")
-        else:
-            # If header not found, still include it with placeholder values
-            print(f"DEBUG: Header '{text}' not found on page {llm_page}, using placeholders")
-            entry = [
-                text,
-                None,  # fontsize
-                llm_page,
-                None,  # y coordinate
-                suggested_level,
-                confidence
-            ]
-            headersJson.append(entry)
-    
-    print(f"DEBUG: headers_with_location returning {len(headersJson)} headers")
     return headersJson
 
 
-
 def build_hierarchy_from_llm(headers):
     nodes = []