Spaces:
Running
Running
Update storage_handler.py
Browse files- storage_handler.py +4 -4
storage_handler.py
CHANGED
|
@@ -76,8 +76,9 @@ _fs = HfFileSystem(token=HF_TOKEN)
|
|
| 76 |
# ------------------------------------------------------------------------
|
| 77 |
def ensure_bucket():
|
| 78 |
"""Create the private bucket if it doesn't exist."""
|
|
|
|
|
|
|
| 79 |
try:
|
| 80 |
-
# Try to create bucket
|
| 81 |
_api.create_bucket(
|
| 82 |
bucket_id=f"{OWNER}/{BUCKET_ID}",
|
| 83 |
private=True,
|
|
@@ -92,15 +93,14 @@ def ensure_bucket():
|
|
| 92 |
logger.warning(f"⚠️ Cannot create bucket - permission issue: {e}")
|
| 93 |
logger.info("The bucket may already exist or you don't have permissions")
|
| 94 |
else:
|
| 95 |
-
# Try alternative bucket name format
|
| 96 |
try:
|
| 97 |
_api.create_bucket(
|
| 98 |
bucket_id=BUCKET_ID,
|
| 99 |
private=True,
|
| 100 |
exist_ok=True
|
| 101 |
)
|
| 102 |
-
|
| 103 |
-
OWNER = ""
|
| 104 |
BUCKET_URI = f"hf://buckets/{BUCKET_ID}"
|
| 105 |
logger.info(f"✅ Bucket '{BUCKET_ID}' is ready (alternative format)")
|
| 106 |
except Exception as e2:
|
|
|
|
| 76 |
# ------------------------------------------------------------------------
|
| 77 |
def ensure_bucket():
|
| 78 |
"""Create the private bucket if it doesn't exist."""
|
| 79 |
+
global OWNER, BUCKET_URI, BUCKET_ID
|
| 80 |
+
|
| 81 |
try:
|
|
|
|
| 82 |
_api.create_bucket(
|
| 83 |
bucket_id=f"{OWNER}/{BUCKET_ID}",
|
| 84 |
private=True,
|
|
|
|
| 93 |
logger.warning(f"⚠️ Cannot create bucket - permission issue: {e}")
|
| 94 |
logger.info("The bucket may already exist or you don't have permissions")
|
| 95 |
else:
|
|
|
|
| 96 |
try:
|
| 97 |
_api.create_bucket(
|
| 98 |
bucket_id=BUCKET_ID,
|
| 99 |
private=True,
|
| 100 |
exist_ok=True
|
| 101 |
)
|
| 102 |
+
# now we can safely reassign globals
|
| 103 |
+
OWNER = ""
|
| 104 |
BUCKET_URI = f"hf://buckets/{BUCKET_ID}"
|
| 105 |
logger.info(f"✅ Bucket '{BUCKET_ID}' is ready (alternative format)")
|
| 106 |
except Exception as e2:
|