prthm11 commited on
Commit
e80610d
·
verified ·
1 Parent(s): 7f63ea1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -2601,18 +2601,24 @@ def process_pdf():
2601
  logger.info(f"Saved uploaded PDF to: {pdf_stream}")
2602
 
2603
  # pdf= save_pdf_to_generated_dir(saved_pdf_path, project_id)
 
2604
  pdf= save_pdf_to_generated_dir(pdf_stream, project_id)
2605
  # logger.info(f"Created project folder: {project_folder}")
2606
  # logger.info(f"Saved uploaded PDF to: {saved_pdf_path}")
2607
  logger.info(f"Saved uploaded PDF to: {pdf_file}: {pdf}")
2608
- print("--------------------------------pdf_file_path---------------------",pdf_file,pdf_stream)
 
 
2609
  # }
2610
 
2611
  # {
2612
  # Extract & process
2613
  # output_path, result = extract_images_from_pdf(saved_pdf_path)
 
2614
  output_path = extract_images_from_pdf(pdf_stream)
2615
  print(" --------------------------------------- zip_path_str ---------------------------------------", output_path)
 
 
2616
  # }
2617
 
2618
  # Check extracted_sprites.json for "scratch block" in any 'name'
@@ -2624,9 +2630,11 @@ def process_pdf():
2624
 
2625
  # with open(extracted_sprites_json, 'r') as f:
2626
  # sprite_data = json.load(f)
2627
-
2628
  project_output = similarity_matching(output_path, project_folder)
2629
  logger.info("Received request to process PDF.")
 
 
2630
 
2631
  with open(project_output, 'r') as f:
2632
  project_skeleton = json.load(f)
 
2601
  logger.info(f"Saved uploaded PDF to: {pdf_stream}")
2602
 
2603
  # pdf= save_pdf_to_generated_dir(saved_pdf_path, project_id)
2604
+ start_time = time.time()
2605
  pdf= save_pdf_to_generated_dir(pdf_stream, project_id)
2606
  # logger.info(f"Created project folder: {project_folder}")
2607
  # logger.info(f"Saved uploaded PDF to: {saved_pdf_path}")
2608
  logger.info(f"Saved uploaded PDF to: {pdf_file}: {pdf}")
2609
+ print("--------------------------------pdf_file_path---------------------",pdf_file,pdf_stream)
2610
+ total_time = time.time() - start_time
2611
+ print(f"-----------------------------Execution Time save_pdf_to_generated_dir() : {total_time}-----------------------------\n")
2612
  # }
2613
 
2614
  # {
2615
  # Extract & process
2616
  # output_path, result = extract_images_from_pdf(saved_pdf_path)
2617
+ start_time = time.time()
2618
  output_path = extract_images_from_pdf(pdf_stream)
2619
  print(" --------------------------------------- zip_path_str ---------------------------------------", output_path)
2620
+ total_time = time.time() - start_time
2621
+ print(f"-----------------------------Execution Time extract_images_from_pdf() : {total_time}-----------------------------\n")
2622
  # }
2623
 
2624
  # Check extracted_sprites.json for "scratch block" in any 'name'
 
2630
 
2631
  # with open(extracted_sprites_json, 'r') as f:
2632
  # sprite_data = json.load(f)
2633
+ start_time = time.time()
2634
  project_output = similarity_matching(output_path, project_folder)
2635
  logger.info("Received request to process PDF.")
2636
+ total_time = time.time() - start_time
2637
+ print(f"-----------------------------Execution Time similarity_matching() : {total_time}-----------------------------\n")
2638
 
2639
  with open(project_output, 'r') as f:
2640
  project_skeleton = json.load(f)