saifisvibinn commited on
Commit
dd2429a
·
1 Parent(s): cb9aca4

Add better error handling for model loading and directory creation

Browse files
Files changed (1) hide show
  1. app.py +13 -2
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 = Path(app.config['UPLOAD_FOLDER']) / filename
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
- load_model_once()
 
 
 
 
 
 
 
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