Spaces:
Running
Running
File size: 2,751 Bytes
7647e70 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import logging
import streamlit as st
import time
from constants import MAX_FILE_SIZE_MB
# Configure logging
logger = logging.getLogger("error_handler")
logger.setLevel(logging.INFO)
def handle_ocr_error(exception, progress_reporter=None):
"""
Handle OCR processing errors and provide user-friendly messages
Args:
exception: The exception that occurred
progress_reporter: ProgressReporter instance for UI updates
Returns:
str: User-friendly error message
"""
error_message = str(exception)
# Complete progress reporting if provided
if progress_reporter:
progress_reporter.complete(success=False)
# Check for specific error types and provide helpful user-facing messages
if "rate limit" in error_message.lower() or "429" in error_message or "requests rate limit exceeded" in error_message.lower():
friendly_message = "The AI service is currently experiencing high demand. Please try again in a few minutes."
logger.error(f"Rate limit error: {error_message}")
return friendly_message
elif "quota" in error_message.lower() or "credit" in error_message.lower() or "subscription" in error_message.lower():
friendly_message = "The API usage quota has been reached. Please check your API key and subscription limits."
logger.error(f"API quota error: {error_message}")
return friendly_message
elif "timeout" in error_message.lower() or "timed out" in error_message.lower():
friendly_message = "The request timed out. This may be due to a large document or high server load. Please try again or use a smaller document."
logger.error(f"Timeout error: {error_message}")
return friendly_message
elif "file size" in error_message.lower() or "too large" in error_message.lower():
friendly_message = f"The file is too large. Maximum file size is {MAX_FILE_SIZE_MB}MB."
logger.error(f"File size error: {error_message}")
return friendly_message
else:
# Generic error message for other errors
logger.error(f"OCR processing error: {error_message}", exc_info=True)
return f"An error occurred during processing: {error_message}"
def check_file_size(file_bytes):
"""
Check if file size is within limits
Args:
file_bytes: File content as bytes
Returns:
tuple: (is_valid, file_size_mb, error_message)
"""
file_size_mb = len(file_bytes) / (1024 * 1024)
if file_size_mb > MAX_FILE_SIZE_MB:
error_message = f"File size {file_size_mb:.2f} MB exceeds limit of {MAX_FILE_SIZE_MB} MB"
return False, file_size_mb, error_message
return True, file_size_mb, None
|