push again

smpl_generator.py

@@ -40,12 +40,32 @@ class SMPLGenerator:
             
             expected_smpl_dir = model_path_obj / "smpl"
             expected_models_dir = expected_smpl_dir / "models"
+            expected_models_dir.mkdir(parents=True, exist_ok=True)
             
-            if not expected_models_dir.exists() and models_source.exists():
-                print(f"Creating expected directory structure: {expected_models_dir}")
-                expected_models_dir.mkdir(parents=True, exist_ok=True)
+            name_mapping = {
+                "basicmodel_m_lbs": "SMPL_MALE.pkl",
+                "basicModel_f_lbs": "SMPL_FEMALE.pkl",
+                "basicModel_neutral_lbs": "SMPL_NEUTRAL.pkl"
+            }
+            
+            for model_file in model_files:
+                file_lower = model_file.name.lower()
+                target_name = None
+                
+                if "basicmodel_m" in file_lower or "male" in file_lower:
+                    target_name = "SMPL_MALE.pkl"
+                elif "basicmodel_f" in file_lower or "female" in file_lower:
+                    target_name = "SMPL_FEMALE.pkl"
+                elif "neutral" in file_lower:
+                    target_name = "SMPL_NEUTRAL.pkl"
                 
-                for model_file in model_files:
+                if target_name:
+                    target_file = expected_models_dir / target_name
+                    if not target_file.exists():
+                        import shutil
+                        shutil.copy2(model_file, target_file)
+                        print(f"Copied {model_file.name} -> {target_name}")
+                else:
                     target_file = expected_models_dir / model_file.name
                     if not target_file.exists():
                         import shutil