Update app.py

app.py

@@ -1957,18 +1957,39 @@ def similarity_matching(input_json_path: str, project_folder: str) -> str:
     for backdrop_idx, matched_idx in enumerate(most_similar_indices):
         matched_image_path = folder_image_paths[matched_idx]
         matched_folder     = os.path.dirname(matched_image_path)
-
+        matched_filename = os.path.basename(matched_image_path)
+        
         # CHANGED: use our new normalized backdrop_base_path
         if not matched_folder.startswith(backdrop_base_path):
             continue
 
         logger.info(f"Matched backdrop: {matched_image_path}")
+
+        # 1) Copy the matched backdrop image itself
+        try:
+            shutil.copy2(
+                matched_image_path,
+                os.path.join(project_folder, matched_filename)
+            )
+            logger.info(f"✅ Copied matched backdrop image {matched_filename} to {project_folder}")
+        except Exception as e:
+            logger.error(f"❌ Failed to copy matched backdrop {matched_image_path}: {e}")
+        
         # copy non‐matched files
         for fname in os.listdir(matched_folder):
-            if fname in (os.path.basename(matched_image_path), 'project.json'):
+            # if fname in (os.path.basename(matched_image_path), 'project.json'):
+            if fname in {matched_filename, 'project.json'}:
                 continue
-            shutil.copy2(os.path.join(matched_folder, fname),
-                         os.path.join(project_folder, fname))
+            # shutil.copy2(os.path.join(matched_folder, fname),
+            #              os.path.join(project_folder, fname))
+            src = os.path.join(matched_folder, fname)
+            dst = os.path.join(project_folder, fname)
+            if os.path.isfile(src):
+                try:
+                    shutil.copy2(src, dst)
+                    logger.info(f"Copied additional backdrop asset {fname} to project folder")
+                except Exception as e:
+                    logger.error(f"Failed to copy {src}: {e}")
 
         # append the stage‐target from its project.json
         pj = os.path.join(matched_folder, 'project.json')