Update app.py

app.py

@@ -154,7 +154,37 @@ def preprocess_voynich_image(img_pil):
     
     return Image.fromarray(enhanced)
 
-# Create interface with preprocessing option
+def debug_line_detection(img_pil):
+    """Debug version that shows intermediate processing steps"""
+    if img_pil is None:
+        return None, None, None
+    
+    img = np.array(img_pil)
+    if len(img.shape) == 3:
+        gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
+    else:
+        gray = img
+    
+    # Try different thresholds
+    _, thresh1 = cv2.threshold(gray, 100, 255, cv2.THRESH_BINARY_INV)
+    thresh2 = cv2.adaptiveThreshold(gray, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, 
+                                   cv2.THRESH_BINARY_INV, 11, 2)
+    _, thresh3 = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
+    
+    # Combine all
+    thresh = cv2.bitwise_or(thresh1, cv2.bitwise_or(thresh2, thresh3))
+    
+    # Small dilation
+    kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (5, 2))
+    dilated = cv2.dilate(thresh, kernel, iterations=1)
+    
+    # Convert back to PIL for display
+    thresh_pil = Image.fromarray(thresh)
+    dilated_pil = Image.fromarray(dilated)
+    
+    return thresh_pil, dilated_pil, extract_first_line(img_pil)
+
+# Add debug interface
 with gr.Blocks(title="Voynich Manuscript Line Extractor") as demo:
     gr.Markdown("# Voynich Manuscript Line Extractor")
     gr.Markdown("Upload a scanned folio of the Voynich manuscript. The app will detect and crop the first visible line of text.")
@@ -164,10 +194,16 @@ with gr.Blocks(title="Voynich Manuscript Line Extractor") as demo:
             input_image = gr.Image(type="pil", label="Upload Voynich Folio")
             enhance_btn = gr.Button("Enhance Image First")
             extract_btn = gr.Button("Extract First Line")
+            debug_btn = gr.Button("Debug Line Detection")
             
         with gr.Column():
             enhanced_output = gr.Image(label="Enhanced Image")
             line_output = gr.Image(label="Extracted First Line")
+            
+    with gr.Row():
+        debug_thresh = gr.Image(label="Threshold Result")
+        debug_dilated = gr.Image(label="After Dilation")
+        debug_extracted = gr.Image(label="Final Extraction")
     
     enhance_btn.click(
         fn=preprocess_voynich_image,
@@ -180,6 +216,12 @@ with gr.Blocks(title="Voynich Manuscript Line Extractor") as demo:
         inputs=input_image,
         outputs=line_output
     )
+    
+    debug_btn.click(
+        fn=debug_line_detection,
+        inputs=input_image,
+        outputs=[debug_thresh, debug_dilated, debug_extracted]
+    )
 
 if __name__ == "__main__":
     demo.launch()