Update app.py

app.py

@@ -1,4 +1,4 @@
-
+'''
 from cv2 import threshold
 import keras_ocr
 import cv2
@@ -71,3 +71,73 @@ iface = gr.Interface(fn=RemoveText,
                     title="Remove Text for Image")
 iface.launch(debug=True)
 
+'''
+
+
+
+
+import cv2
+import numpy as np
+import keras_ocr
+
+def detect_and_draw_lines(img, rho=1, theta=1):
+    """Detect lines in the image and draw them."""
+    edges = cv2.Canny(img, 50, 150, apertureSize=3)
+    lines = cv2.HoughLines(edges, rho, theta, 100)
+    if lines is not None:
+        for line in lines:
+            for x1, y1, x2, y2 in line:
+                cv2.line(img, (x1, y1), (x2, y2), (0, 0, 255), 2)
+    return img
+
+def inpaint_lines(img):
+    """Inpaint the detected lines."""
+    # Create a mask where lines are white and the rest is black
+    mask = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+    mask[mask > 0] = 255
+    # Inpaint using the mask
+    inpainted_img = cv2.inpaint(img, mask, 3, cv2.INPAINT_TELEA)
+    return inpainted_img
+
+def inpaint_text(img_path, pipeline):
+    # Read the image
+    img = keras_ocr.tools.read(img_path)
+    # Detect and draw lines
+    img_with_lines = detect_and_draw_lines(img)
+    # Inpaint the lines
+    inpainted_img = inpaint_lines(img_with_lines)
+    # Recognize text
+    prediction_groups = pipeline.recognize([inpainted_img])
+    # Define the mask for inpainting text
+    mask = np.zeros(inpainted_img.shape[:2], dtype="uint8")
+    for box in prediction_groups[0]:
+        x0, y0, x1, y1, x2, y2, x3, y3 = box[1]
+        cv2.line(mask, (x0, y0), (x1, y1), 255, 10)
+        cv2.line(mask, (x2, y2), (x3, y3), 255, 10)
+    # Inpaint text
+    inpainted_img = cv2.inpaint(inpainted_img, mask, 3, cv2.INPAINT_TELEA)
+    return inpainted_img
+
+# keras-ocr will automatically download pretrained weights for the detector and recognizer.
+pipeline = keras_ocr.pipeline.Pipeline()
+def RemoveTextAndLines(image):
+    img_text_and_lines_removed = inpaint_text(image, pipeline)
+    return cv2.cvtColor(img_text_and_lines_removed, cv2.COLOR_BGR2RGB)
+
+# Example usage with Gradio interface (assuming you have Gradio installed)
+import gradio as gr
+
+iface = gr.Interface(
+    fn=RemoveTextAndLines,
+    inputs=gr.inputs.Image(label="Image to Remove Text and Lines From", type="numpy"),
+    outputs="image",
+    examples=[["1.jpg"]],
+    title="Remove Text and Lines for Image"
+)
+iface.launch(debug=True)
+
+
+
+
+
+