Spaces:
Running
Running
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 | |