Spaces:
Sleeping
Sleeping
saifisvibinn commited on
Commit ·
dd2429a
1
Parent(s): cb9aca4
Add better error handling for model loading and directory creation
Browse files
app.py
CHANGED
|
@@ -334,8 +334,12 @@ def process_file_background(task_id: str, file_data: bytes, filename: str, extra
|
|
| 334 |
include_images = extraction_mode != 'markdown'
|
| 335 |
include_markdown = extraction_mode != 'images'
|
| 336 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 337 |
# Save uploaded file from bytes data
|
| 338 |
-
upload_path =
|
| 339 |
upload_path.write_bytes(file_data)
|
| 340 |
|
| 341 |
_update_task_progress(task_id, filename, 15, f'Saved {filename}, preparing output...')
|
|
@@ -355,7 +359,14 @@ def process_file_background(task_id: str, file_data: bytes, filename: str, extra
|
|
| 355 |
logger.info(f"Processing {filename} (images={include_images}, markdown={include_markdown})")
|
| 356 |
|
| 357 |
if include_images:
|
| 358 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 359 |
|
| 360 |
_update_task_progress(task_id, filename, 30, f'Extracting content from {filename}...')
|
| 361 |
|
|
|
|
| 334 |
include_images = extraction_mode != 'markdown'
|
| 335 |
include_markdown = extraction_mode != 'images'
|
| 336 |
|
| 337 |
+
# Ensure upload directory exists
|
| 338 |
+
upload_dir = Path(app.config['UPLOAD_FOLDER'])
|
| 339 |
+
upload_dir.mkdir(parents=True, exist_ok=True)
|
| 340 |
+
|
| 341 |
# Save uploaded file from bytes data
|
| 342 |
+
upload_path = upload_dir / filename
|
| 343 |
upload_path.write_bytes(file_data)
|
| 344 |
|
| 345 |
_update_task_progress(task_id, filename, 15, f'Saved {filename}, preparing output...')
|
|
|
|
| 359 |
logger.info(f"Processing {filename} (images={include_images}, markdown={include_markdown})")
|
| 360 |
|
| 361 |
if include_images:
|
| 362 |
+
try:
|
| 363 |
+
load_model_once()
|
| 364 |
+
logger.info(f"Model loaded successfully for {filename}")
|
| 365 |
+
except Exception as model_error:
|
| 366 |
+
logger.error(f"Failed to load model for {filename}: {model_error}")
|
| 367 |
+
import traceback
|
| 368 |
+
logger.error(traceback.format_exc())
|
| 369 |
+
raise Exception(f"Model loading failed: {str(model_error)}. The processing service may be unavailable.")
|
| 370 |
|
| 371 |
_update_task_progress(task_id, filename, 30, f'Extracting content from {filename}...')
|
| 372 |
|