-
-
Notifications
You must be signed in to change notification settings - Fork 271
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Faster Captcha Retrieval Using Local Flask Server (#88)
* add passenger data * add flash server to handle captha for faster captha resolution * add .gitignore * rollback passenger details * update code to club everything together[one command and booking done] * add wait on as dev dependencies
- Loading branch information
1 parent
88be91a
commit 3361382
Showing
12 changed files
with
465 additions
and
192 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Binary file not shown.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file not shown.
Binary file not shown.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,72 @@ | ||
| import argparse | ||
| import numpy as np | ||
| from PIL import Image | ||
| import io | ||
| import base64 | ||
| import easyocr | ||
| from flask import Flask, request, jsonify | ||
|
|
||
| # Initialize EasyOCR Reader | ||
| reader = easyocr.Reader(["en"], model_storage_directory="./EasyOCR") | ||
|
|
||
| # Initialize Flask app | ||
| app = Flask(__name__) | ||
|
|
||
| def extract_text_from_image(base64_image): | ||
| try: | ||
| # Convert the base64 image to bytes | ||
| image_bytes = base64.b64decode(base64_image[22:]) | ||
| # Create a BytesIO object from the image bytes | ||
| image_buffer = io.BytesIO(image_bytes) | ||
| # Open the image using PIL (Python Imaging Library) | ||
| image = Image.open(image_buffer) | ||
| # Convert the image to grayscale (optional but can improve OCR accuracy) | ||
| image = image.convert("L") | ||
|
|
||
| # Convert PIL image to OpenCV format | ||
| open_cv_image = np.array(image) | ||
|
|
||
| result = reader.readtext(open_cv_image, detail=0) | ||
| if result: | ||
| return result[0].replace(" ", "") | ||
| else: | ||
| return "ABCDEF" | ||
| except Exception as e: | ||
| return f"Error processing image: {str(e)}" | ||
|
|
||
| @app.route("/extract-text", methods=["POST"]) | ||
| def extract_text(): | ||
| # Get the base64 image from the request | ||
| data = request.get_json() | ||
| base64_image = data.get("image", "") | ||
|
|
||
| if not base64_image: | ||
| return jsonify({"error": "No base64 image string provided"}), 400 | ||
|
|
||
| extracted_text = extract_text_from_image(base64_image) | ||
| return jsonify({"extracted_text": extracted_text}) | ||
|
|
||
| @app.route('/') | ||
| def health_check(): | ||
| return "Server is running", 200 | ||
|
|
||
| if __name__ == "__main__": | ||
| parser = argparse.ArgumentParser( | ||
| description="Run the OCR extraction server." | ||
| ) | ||
| parser.add_argument( | ||
| "--host", | ||
| type=str, | ||
| default="0.0.0.0", | ||
| help="Host address to run the server on (default: 0.0.0.0)", | ||
| ) | ||
| parser.add_argument( | ||
| "--port", | ||
| type=int, | ||
| default=5000, | ||
| help="Port to run the server on (default: 5000)", | ||
| ) | ||
| args = parser.parse_args() | ||
|
|
||
| # Run Flask server | ||
| app.run(host=args.host, port=args.port) |
Oops, something went wrong.