Create app.py

app.py

@@ -0,0 +1,52 @@
+import os
+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"
+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")
+    sys.exit(1)
+
+# Download files from Space 2's repository
+# Space 2 has already fetched the code from Space 1
+try:
+    print(f"Downloading code from Space 2: {SPACE2_URL}")
+    
+    local_dir = snapshot_download(
+        repo_id=SPACE2_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
+    )
+    
+    # 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")
+    
+    # List files for verification
+    files = list(Path("/app/code").rglob("*"))
+    print(f"Total files: {len([f for f in files if f.is_file()])}")
+    
+except Exception as e:
+    print(f"❌ Error downloading code: {e}")
+    sys.exit(1)