Update fetch_code.py

fetch_code.py

@@ -3,50 +3,54 @@ import sys
 from pathlib import Path
 from huggingface_hub import snapshot_download
 
-# Get Space 2 URL from environment
-SPACE2_URL = os.getenv("SPACE2_URL")  # e.g., "username/space2-fetcher"
+# Get PRIVATE Space 1 URL from environment (not Space 2!)
+PRIVATE_SPACE_URL = os.getenv("PRIVATE_SPACE_URL")  # e.g., "username/private-space"
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-if not SPACE2_URL or not HF_TOKEN:
-    print("ERROR: SPACE2_URL and HF_TOKEN must be set as environment variables")
+if not PRIVATE_SPACE_URL or not HF_TOKEN:
+    print("ERROR: PRIVATE_SPACE_URL and HF_TOKEN must be set")
+    print("Add them as secrets in your Space settings")
     sys.exit(1)
 
-# Download files from Space 2's repository
-# Space 2 has already fetched the code from Space 1
+# Download files directly from the private Space 1
 try:
-    print(f"Downloading code from Space 2: {SPACE2_URL}")
+    print(f"Downloading code from private space: {PRIVATE_SPACE_URL}")
     
-    local_dir = snapshot_download(
-        repo_id=SPACE2_URL,
+    # Create /app/code directory
+    code_dir = Path("/app/code")
+    code_dir.mkdir(exist_ok=True, parents=True)
+    
+    # Download all files from Space 1
+    snapshot_download(
+        repo_id=PRIVATE_SPACE_URL,
         repo_type="space",
         token=HF_TOKEN,
-        local_dir="/app/code",
-        local_dir_use_symlinks=False,
-        allow_patterns=["fetched_code/**/*"]  # Only get the fetched_code directory
+        local_dir=str(code_dir),
+        local_dir_use_symlinks=False
     )
     
-    # Move files from fetched_code subdirectory to /app/code
-    fetched_dir = Path("/app/code/fetched_code")
-    if fetched_dir.exists():
-        import shutil
-        for item in fetched_dir.iterdir():
-            dest = Path("/app/code") / item.name
-            if dest.exists():
-                if dest.is_dir():
-                    shutil.rmtree(dest)
-                else:
-                    dest.unlink()
-            shutil.move(str(item), str(dest))
-        
-        # Remove the now-empty fetched_code directory
-        fetched_dir.rmdir()
-    
-    print("✅ Successfully downloaded and extracted code")
+    print("✅ Successfully downloaded code")
     
     # List files for verification
-    files = list(Path("/app/code").rglob("*"))
-    print(f"Total files: {len([f for f in files if f.is_file()])}")
+    files = [f for f in code_dir.rglob("*") if f.is_file()]
+    print(f"Total files downloaded: {len(files)}")
+    
+    # Show first 20 files
+    for i, f in enumerate(files[:20]):
+        print(f"  - {f.relative_to(code_dir)}")
+    
+    if len(files) > 20:
+        print(f"  ... and {len(files) - 20} more files")
+    
+    # Check for package.json
+    package_json = code_dir / "package.json"
+    if package_json.exists():
+        print("✅ Found package.json")
+    else:
+        print("⚠️  No package.json found - npm install will be skipped")
     
 except Exception as e:
     print(f"❌ Error downloading code: {e}")
+    import traceback
+    traceback.print_exc()
     sys.exit(1)