Karl/KTVManager_UI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add local ocr support

Browse Source
This commit is contained in:
Karl Hudgell 2024-11-23 09:44:12 +00:00
parent 71f08f01d6
commit 7c8acfb418
3 changed files with 50 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
app.py
View File

@ -1,5 +1,5 @@
# app.py # app.py
from flask import Flask, render_template, request, redirect, url_for, session, flash, send_file from flask import Flask, render_template, request, redirect, url_for, session, send_file, jsonify
from flask_caching import Cache from flask_caching import Cache
import requests.auth import requests.auth
import os import os
@ -10,6 +10,10 @@ import requests
import base64 import base64
from flask import Flask from flask import Flask
from config import DevelopmentConfig from config import DevelopmentConfig
from paddleocr import PaddleOCR
from PIL import Image
import numpy as np
app = Flask(__name__) app = Flask(__name__)
app.config.from_object( app.config.from_object(
@ -17,6 +21,8 @@ app.config.from_object(
) )
cache = Cache(app, config={"CACHE_TYPE": "SimpleCache"}) cache = Cache(app, config={"CACHE_TYPE": "SimpleCache"})
ocr = PaddleOCR(use_angle_cls=True, lang='en') # Adjust language if needed
app.config['SESSION_COOKIE_SECURE'] = True # Only send cookie over HTTPS app.config['SESSION_COOKIE_SECURE'] = True # Only send cookie over HTTPS
app.config['SESSION_COOKIE_HTTPONLY'] = True # Prevent JavaScript access app.config['SESSION_COOKIE_HTTPONLY'] = True # Prevent JavaScript access
app.config['SESSION_COOKIE_SAMESITE'] = 'Lax' # Adjust for cross-site requests app.config['SESSION_COOKIE_SAMESITE'] = 'Lax' # Adjust for cross-site requests
@ -156,5 +162,23 @@ def delete_account():
return redirect(url_for("user_accounts")) return redirect(url_for("user_accounts"))
@app.route('/OCRupload', methods=['POST'])
def OCRupload():
if 'image' not in request.files:
return jsonify({"error": "No image file found"}), 400
# Get the uploaded file
file = request.files['image']
try:
image = Image.open(file.stream)
image_np = np.array(image)
result = ocr.ocr(image_np)
# Extract text
extracted_text = []
for line in result[0]:
extracted_text.append(line[1][0])
return render_template("add_account.html", username=extracted_text[2], password=extracted_text[3])
except Exception as e:
return jsonify({"error": str(e)}), 500
if __name__ == "__main__": if __name__ == "__main__":
app.run(debug=app.config["DEBUG"], host=app.config["HOST"], port=app.config["PORT"]) app.run(debug=app.config["DEBUG"], host=app.config["HOST"], port=app.config["PORT"])

BIN
requirements.txt
View File

Binary file not shown.

41
templates/add_account.html
View File

@ -6,16 +6,16 @@
<meta name="viewport" content="width=device-width, initial-scale=1.0"> <meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Add Account - KTVManager</title> <title>Add Account - KTVManager</title>
<link rel="stylesheet" href="https://maxcdn.bootstrapcdn.com/bootstrap/4.5.2/css/bootstrap.min.css"> <link rel="stylesheet" href="https://maxcdn.bootstrapcdn.com/bootstrap/4.5.2/css/bootstrap.min.css">
<link rel="stylesheet" href="{{ url_for('static', filename='styles.css') }}" /> <link rel="stylesheet" href="{{ url_for('static', filename='styles.css') }}" />
<style> <style>
/* Hide the spinner by default */ /* Hide the spinner by default */
#loadingSpinner { #loadingSpinner,
#ocrLoadingSpinner {
display: none; display: none;
} }
</style> </style>
</head> </head>
<body> <body>
<!-- Navbar --> <!-- Navbar -->
<nav class="navbar navbar-expand-lg navbar-dark bg-dark"> <nav class="navbar navbar-expand-lg navbar-dark bg-dark">
<a class="navbar-brand" href="/">KTVManager</a> <a class="navbar-brand" href="/">KTVManager</a>
@ -52,18 +52,17 @@
</div> </div>
<!-- Main Content --> <!-- Main Content -->
<main div class="container mt-5"> <main class="container mt-5">
<h1>Add Account</h1> <h1>Add Account</h1>
<br> <div>
<form action="/accounts/add" method="POST" onsubmit="showLoading()"> <form action="/accounts/add" method="POST" onsubmit="showLoading()">
<div class="form-group"> <div class="form-group">
<label for="username">Username</label> <label for="username">Username</label>
<input type="text" class="form-control" id="username" name="username" required> <input type="text" class="form-control" id="username" name="username" value="{{ username }}" required>
</div> </div>
<div class="form-group"> <div class="form-group">
<label for="password">Password</label> <label for="password">Password</label>
<input type="text" class="form-control" id="password" name="password" required> <input type="text" class="form-control" id="password" name="password" value="{{ password }}" required>
</div> </div>
<div class="form-group"> <div class="form-group">
<label for="stream">Stream Name</label> <label for="stream">Stream Name</label>
@ -74,24 +73,34 @@
<span id="buttonText">Add Account</span> <span id="buttonText">Add Account</span>
</button> </button>
</form> </form>
</div> <hr>
</main> <h4>Load Details Via OCR</h2>
<form action="/OCRupload" method="POST" enctype="multipart/form-data" onsubmit="showLoadingOCR()">
<div class="form-group">
<label for="image">Select Image</label>
<input type="file" class="form-control-file" id="image" name="image" accept="image/*" required>
</div>
<button type="submit" class="btn btn-success" id="ocrButton">
<span class="spinner-border spinner-border-sm" id="ocrLoadingSpinner" role="status" aria-hidden="true"></span>
<span id="ocrButtonText">Load Details</span>
</button>
</form>
</main>
<footer class="bg-dark text-white text-center py-3 mt-5"> <footer class="bg-dark text-white text-center py-3 mt-5">
<p></p> <p></p>
</footer> </footer>
<script src="https://code.jquery.com/jquery-3.5.1.slim.min.js"></script>
<script src="https://cdn.jsdelivr.net/npm/@popperjs/core@2.0.7/dist/umd/popper.min.js"></script>
<script src="https://stackpath.bootstrapcdn.com/bootstrap/4.5.2/js/bootstrap.min.js"></script>
<script> <script>
function showLoading() { function showLoading() {
// Disable the button to prevent multiple submissions
document.getElementById("submitButton").disabled = true; document.getElementById("submitButton").disabled = true;
// Show spinner and change button text
document.getElementById("loadingSpinner").style.display = "inline-block"; document.getElementById("loadingSpinner").style.display = "inline-block";
document.getElementById("buttonText").textContent = "Working..."; document.getElementById("buttonText").textContent = "Working...";
} }
function showLoadingOCR() {
document.getElementById("ocrButton").disabled = true;
document.getElementById("ocrLoadingSpinner").style.display = "inline-block";
document.getElementById("ocrButtonText").textContent = "Processing...";
}
</script> </script>
</body> </body>
</html> </html>