Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -19,7 +19,7 @@ def detect_book_regions(image: np.ndarray, min_area=10000, eps_coef=0.02):
|
|
| 19 |
edges = cv2.Canny(blurred, 50, 150)
|
| 20 |
|
| 21 |
# Dilate + erode to close gaps
|
| 22 |
-
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (5,5))
|
| 23 |
closed = cv2.morphologyEx(edges, cv2.MORPH_CLOSE, kernel)
|
| 24 |
|
| 25 |
contours, _ = cv2.findContours(closed.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
|
|
@@ -56,23 +56,4 @@ def ocr_on_region(image: np.ndarray, box: tuple):
|
|
| 56 |
x, y, w, h = box
|
| 57 |
cropped = image[y:y+h, x:x+w]
|
| 58 |
gray_crop = cv2.cvtColor(cropped, cv2.COLOR_BGR2GRAY)
|
| 59 |
-
_, thresh_crop = cv2.threshold(gray_crop, 0, 255, cv2.
|
| 60 |
-
custom_config = r'--oem 3 --psm 6'
|
| 61 |
-
text = pytesseract.image_to_string(thresh_crop, config=custom_config)
|
| 62 |
-
return text.strip()
|
| 63 |
-
|
| 64 |
-
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 65 |
-
# 3. Query OpenLibrary API
|
| 66 |
-
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 67 |
-
def query_openlibrary(title_text: str, author_text: str = None):
|
| 68 |
-
"""
|
| 69 |
-
Search OpenLibrary by title (and optional author).
|
| 70 |
-
Return a dict with title, author_name, publisher, first_publish_year, or None.
|
| 71 |
-
"""
|
| 72 |
-
base_url = "https://openlibrary.org/search.json"
|
| 73 |
-
params = {"title": title_text}
|
| 74 |
-
if author_text:
|
| 75 |
-
params["author"] = author_text
|
| 76 |
-
|
| 77 |
-
try:
|
| 78 |
-
resp = requests.get(base_url, params=params, timeout=5)
|
|
|
|
| 19 |
edges = cv2.Canny(blurred, 50, 150)
|
| 20 |
|
| 21 |
# Dilate + erode to close gaps
|
| 22 |
+
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (5, 5))
|
| 23 |
closed = cv2.morphologyEx(edges, cv2.MORPH_CLOSE, kernel)
|
| 24 |
|
| 25 |
contours, _ = cv2.findContours(closed.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
|
|
|
|
| 56 |
x, y, w, h = box
|
| 57 |
cropped = image[y:y+h, x:x+w]
|
| 58 |
gray_crop = cv2.cvtColor(cropped, cv2.COLOR_BGR2GRAY)
|
| 59 |
+
_, thresh_crop = cv2.threshold(gray_crop, 0, 255, cv2._
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|