Update app.py

app.py

@@ -250,7 +250,9 @@ def display_detected_lines(original_path, output_path):
     txt_path = os.path.join('yolov3/runs/detect/mainlinedection/labels', os.path.basename(output_path).replace(".jpg", ".txt"))
 
     if os.path.exists(txt_path):
+        # Load both original and thresholded images
         original_image = Image.open(original_path)
+        thresholded_image = process_image(original_image)  # This is your function that returns a thresholded PIL image
         boxes = get_detected_boxes(txt_path, original_image.width, original_image.height)
         
         if not boxes:
@@ -269,7 +271,8 @@ def display_detected_lines(original_path, output_path):
                 x_max = int(x_center + (width / 2))
                 y_max = int(y_center + (height / 2))
                 
-                extracted_line = original_image.crop((x_min, y_min, x_max, y_max))
+                # Crop the thresholded image instead of the original
+                extracted_line = thresholded_image.crop((x_min, y_min, x_max, y_max))
                 
                 # Save the detected line image to the temporary directory
                 detected_line_path = os.path.join(temp_dir, f"detected_line_{index}.jpg")
@@ -279,6 +282,7 @@ def display_detected_lines(original_path, output_path):
             # Perform OCR on detected lines
             recognized_texts = perform_ocr_on_detected_lines(detected_line_paths)
 
+
             # print("Decoded OCR Results:", recognized_texts)
             # st.text(f"Detected Line: {recognized_texts}")