Update app.py

app.py

@@ -22,6 +22,17 @@ from io import BytesIO
 from pathlib import Path
 import os
 from utils.block_relation_builder_v1 import block_builder, variable_adder_main
+
+def log_execution_time(func):
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+        start_time = time.time()
+        result = func(*args, **kwargs)
+        end_time = time.time()
+        logger.info(f"⏱ {func.__name__} executed in {end_time - start_time:.2f} seconds")
+        return result
+    return wrapper
+
 global pdf_doc
 # ============================== #
 #     INITIALIZE CLIP EMBEDDER   #
@@ -1917,6 +1928,7 @@ def plan_logic_aligner_node(state: GameState):
 # Prepare manipulated sprite JSON structure
 manipulated_json = {}
 img_elements = []
+@log_execution_time
 # { changes: "pdf_stream" in place of "pdf_path"
 def extract_images_from_pdf(pdf_stream: io.BytesIO):
     ''' Extract images from PDF and generate structured sprite JSON '''
@@ -1999,6 +2011,7 @@ def extract_images_from_pdf(pdf_stream: io.BytesIO):
     except Exception as e:
         raise RuntimeError(f"❌ Error in extract_images_from_pdf: {str(e)}")
 
+@log_execution_time
 # def similarity_matching(input_json_path: str, project_folder: str) -> str:
 def similarity_matching(sprites_data: str, project_folder: str) -> str:
     logger.info("🔍 Running similarity matching…")
@@ -2374,6 +2387,7 @@ def similarity_matching(sprites_data: str, project_folder: str) -> str:
 #         buf.seek(0)
 #         buffers.append(buf)
 #     return buffers
+@log_execution_time
 def convert_pdf_stream_to_images(pdf_stream: io.BytesIO, dpi=300):
     # Ensure we are at the start of the stream
     pdf_stream.seek(0)
@@ -2436,7 +2450,8 @@ def upscale_image(image: Image.Image, scale: int = 2) -> Image.Image:
     except Exception as e:
         logger.error(f"❌ Error during image upscaling: {str(e)}")
         return image 
-        
+
+@log_execution_time
 def create_sb3_archive(project_folder, project_id):
     """
     Zips the project folder and renames it to an .sb3 file.
@@ -2476,6 +2491,7 @@ def create_sb3_archive(project_folder, project_id):
             os.remove(sb3_path)
         return sb3_path
 
+@log_execution_time
 #{ changes -> pdf_stream replacement of pdf_path
 # def save_pdf_to_generated_dir(pdf_path: str, project_id: str) -> str:
 def save_pdf_to_generated_dir(pdf_stream: io.BytesIO, project_id: str) -> str: