Skip to content

Fix: add confidence threshold and prevent UI freezing on empty states #7

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
khushanuma-shabbir wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Fix: add confidence threshold and prevent UI freezing on empty states #7

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
67 changes: 40 additions & 27 deletions app.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@
last_letter = ""
letter_start_time = None
auto_add_delay = 5
CONFIDENCE_THRESHOLD = 0.70 # Minimum 70% confidence required


@app.route('/')
Expand Down Expand Up @@ -90,9 +91,13 @@ def predict():
imgCrop = img[y1:y2, x1:x2]

if imgCrop.size == 0:

last_letter = ""
letter_start_time = None
return jsonify({
"success": False
"success": True,
"letter": "Waiting...",
"confidence": 0.0,
"word": word
})

aspectRatio = h / w
Expand Down Expand Up @@ -137,43 +142,51 @@ def predict():
)

current_letter = labels[index]

confidence = prediction[index]

# Auto add letter logic
if current_letter == last_letter:

if (
letter_start_time and
time.time() - letter_start_time >= auto_add_delay
):

word += current_letter

letter_start_time = None
last_letter = ""
if confidence >= CONFIDENCE_THRESHOLD:
# Auto add letter logic
if current_letter == last_letter:
if (
letter_start_time and
time.time() - letter_start_time >= auto_add_delay
):
word += current_letter
letter_start_time = None
last_letter = ""
else:
last_letter = current_letter
letter_start_time = time.time()

return jsonify({
"success": True,
"letter": current_letter,
"confidence": float(confidence),
"word": word
})
else:

last_letter = current_letter
letter_start_time = time.time()

return jsonify({
"success": True,
"letter": current_letter,
"confidence": float(confidence),
"word": word
})
# Reset tracking if confidence is too low (e.g., hand transition)
last_letter = ""
letter_start_time = None
return jsonify({
"success": True,
"letter": "Unsure",
"confidence": float(confidence),
"word": word
})

else:

current_letter = ""
confidence = 0.0
last_letter = ""
letter_start_time = None

# Return success: True so the frontend updates to "Waiting..."
return jsonify({
"success": False
"success": True,
"letter": "Waiting...",
"confidence": 0.0,
"word": word
})

except Exception as e:
Expand Down
8 changes: 6 additions & 2 deletions templates/index.html
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -376,8 +376,12 @@

if(data.success){

currentLetter.textContent =
`${data.letter} (${Math.round(data.confidence * 100)}%)`;
if (data.letter === "Waiting..." || data.letter === "Unsure") {
currentLetter.textContent = data.letter;
} else {
currentLetter.textContent =
`${data.letter} (${Math.round(data.confidence * 100)}%)`;
}

document.getElementById(
'displayWord'
Expand Down