Spaces:
Running
Running
File size: 859 Bytes
449ce0a edd5b40 449ce0a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import os
from google.cloud import storage
GCP_PROJECT = "gen-lang-client-0350149082"
GCP_REGION = "us-central1"
DOCUMENT_API_ID = "b34a20d22dee16bb"
GCS_BUCKET_NAME = "vella-pdfs"
def upload_to_gcs(LOCAL_PDF_PATH: str) -> str:
# Path in GCS
GCS_DESTINATION_BLOB_NAME = "gemini_uploads/" + os.path.basename(LOCAL_PDF_PATH)
"""Uploads a file to a GCS bucket and returns its URI."""
storage_client = storage.Client(
project=GCP_PROJECT,
)
bucket = storage_client.bucket(GCS_BUCKET_NAME)
blob = bucket.blob(GCS_DESTINATION_BLOB_NAME)
print(
f"Uploading {LOCAL_PDF_PATH} to gs://{GCS_BUCKET_NAME}/{GCS_DESTINATION_BLOB_NAME}..."
)
blob.upload_from_filename(LOCAL_PDF_PATH)
gcs_uri = f"gs://{GCS_BUCKET_NAME}/{GCS_DESTINATION_BLOB_NAME}"
print(f"File uploaded to {gcs_uri}")
return gcs_uri
|