Improve debugging with better error handling and file size info

app.py

@@ -34,34 +34,52 @@ def ensure_weights():
         "pretrained_models/sfe_editor_light.pt",
         "pretrained_models/stylegan2-ffhq-config-f.pt",
     ]
-    if all(os.path.exists(p) for p in need):
+    
+    # Check if any of the needed files exist
+    files_exist = any(os.path.exists(p) for p in need)
+    if files_exist:
+        logger.info("Some weights already exist, skipping download")
         return
 
     repo_id = "LogicGoInfotechSpaces/Smile_Changer_pre_model"
     logger.info("Missing weights; downloading snapshot from %s", repo_id)
-    snapshot_download(
-        repo_id=repo_id,
-        local_dir="pretrained_models",
-        local_dir_use_symlinks=False,
-        allow_patterns=["**/*"],
-        token=os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN"),
-    )
+    
+    try:
+        snapshot_download(
+            repo_id=repo_id,
+            local_dir="pretrained_models",
+            local_dir_use_symlinks=False,
+            allow_patterns=["**/*"],
+            token=os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN"),
+        )
+        logger.info("Download completed successfully")
+    except Exception as e:
+        logger.error("Download failed: %s", e)
+        return
     
     # Add a small delay to ensure files are fully written
     import time
-    time.sleep(2)
+    time.sleep(3)
     
     # Debug: List all files in pretrained_models directory
     if os.path.exists("pretrained_models"):
         logger.info("Files in pretrained_models directory:")
-        for root, dirs, files in os.walk("pretrained_models"):
-            for file in files:
-                logger.info("  %s", os.path.join(root, file))
+        try:
+            for root, dirs, files in os.walk("pretrained_models"):
+                for file in files:
+                    full_path = os.path.join(root, file)
+                    logger.info("  %s (size: %d bytes)", full_path, os.path.getsize(full_path))
+        except Exception as e:
+            logger.error("Error listing files: %s", e)
+    else:
+        logger.error("pretrained_models directory does not exist!")
     
     # Verify critical files exist
     for file_path in need:
         if not os.path.exists(file_path):
             logger.warning("File %s still not found after download", file_path)
+        else:
+            logger.info("File %s found successfully", file_path)
 
 
 def get_runner() -> SimpleRunner: