Update app.py

app.py

@@ -1965,13 +1965,19 @@ def similarity_matching(input_json_path: str, project_folder:str) -> str:
         with open(sprite_json_path, 'r') as f:
             sprite_data = json.load(f)
             # print(f"SPRITE DATA: \n{sprite_data}")
-        # Copy only non-matched files
+        # # Copy only non-matched files
+        # for fname in os.listdir(matched_folder):
+        #     fpath = os.path.join(matched_folder, fname)
+        #     if os.path.isfile(fpath) and fname not in {os.path.basename(matched_image_path), 'sprite.json'}:
+        #         shutil.copy2(fpath, os.path.join(project_folder, fname))
+        #         # logger.info(f"Copied Sprite asset: {fname}")
+        project_data.append(sprite_data)
         for fname in os.listdir(matched_folder):
             fpath = os.path.join(matched_folder, fname)
+            dest_path = os.path.join(project_folder, fname)
             if os.path.isfile(fpath) and fname not in {os.path.basename(matched_image_path), 'sprite.json'}:
-                shutil.copy2(fpath, os.path.join(project_folder, fname))
-                # logger.info(f"Copied Sprite asset: {fname}")
-        project_data.append(sprite_data)
+                shutil.copy2(fpath, dest_path)
+                logger.info(f"🟢 Copied Sprite Asset: {fpath} → {dest_path}")
         
     # ================================================================== #
     #   Loop through most similar images from Backdrops folder           #
@@ -1991,11 +1997,18 @@ def similarity_matching(input_json_path: str, project_folder:str) -> str:
             logger.info(f"Backdrop matched image: {matched_image_path}")
 
             # Copy only non-matched files
+            # for fname in os.listdir(matched_folder):
+            #     fpath = os.path.join(matched_folder, fname)
+            #     if os.path.isfile(fpath) and fname not in {os.path.basename(matched_image_path), 'project.json'}:
+            #         shutil.copy2(fpath, os.path.join(project_folder, fname))
+            #         # logger.info(f"Copied Backdrop asset: {fname}")
             for fname in os.listdir(matched_folder):
                 fpath = os.path.join(matched_folder, fname)
+                dest_path = os.path.join(project_folder, fname)
                 if os.path.isfile(fpath) and fname not in {os.path.basename(matched_image_path), 'project.json'}:
-                    shutil.copy2(fpath, os.path.join(project_folder, fname))
-                    # logger.info(f"Copied Backdrop asset: {fname}")
+                    shutil.copy2(fpath, dest_path)
+                    logger.info(f"🟡 Copied Backdrop Asset: {fpath} → {dest_path}")
+
 
             # Append backdrop's project.json
             backdrop_json_path = os.path.join(matched_folder, 'project.json')
@@ -2117,11 +2130,13 @@ def create_sb3_archive(project_folder: Path, project_id: str) -> Path | None:
     Returns:
         Path: The path to the created .sb3 file, or None if an error occurred.
     """
+    print(" --------------------------------------- create_sb3_archive INITIALIZE ---------------------------------------")
     output_base_name = GEN_PROJECT_DIR / project_id
     zip_path = None
     sb3_path = None
     try:
         # shutil.make_archive automatically adds .zip extension
+        print(" --------------------------------------- zip_path_str ---------------------------------------", output_base_name, project_folder)
         zip_path_str = shutil.make_archive(str(output_base_name), 'zip', root_dir=str(project_folder))
         zip_path = Path(zip_path_str) # Convert back to Path object
         logger.info(f"Project folder zipped to: {zip_path}")
@@ -2227,19 +2242,19 @@ def process_pdf():
         img_bytes = buf.getvalue()
         img_b64 = base64.b64encode(img_bytes).decode("utf-8")
         
-        initial_state_dict = {
-            "project_json": project_skeleton,
-            "description": "The pseudo code for the script",
-            "project_id": project_id,
-            "project_image": img_b64,
-            "action_plan": {},
-            "pseudo_code": {},
-            "temporary_node": {},
-        }
+        # initial_state_dict = {
+        #     "project_json": project_skeleton,
+        #     "description": "The pseudo code for the script",
+        #     "project_id": project_id,
+        #     "project_image": img_b64,
+        #     "action_plan": {},
+        #     "pseudo_code": {},
+        #     "temporary_node": {},
+        # }
         
-        final_state_dict = app_graph.invoke(initial_state_dict)
+        # final_state_dict = app_graph.invoke(initial_state_dict)
 
-        final_project_json = final_state_dict['project_json']
+        final_project_json = project_skeleton # final_state_dict['project_json']
         
         with open(project_output, "w") as f:
             json.dump(final_project_json, f, indent=2)