Update app.py

app.py

@@ -1,54 +1,35 @@
-import requests
 import os
-from googleapiclient.discovery import build
-from google.oauth2.service_account import Credentials
-
-# Define the Google Drive API setup
-SCOPES = ['https://www.googleapis.com/auth/drive']
-SERVICE_ACCOUNT_FILE = 'service_account.json'
-
-# URL of the service account JSON file on Hugging Face
-HUGGINGFACE_JSON_URL = 'https://huggingface.co/username/repository-name/resolve/main/service_account.json'
-
-def download_service_account_file():
-    """Download the service account JSON file from Hugging Face."""
-    if not os.path.exists(SERVICE_ACCOUNT_FILE):  # Check if file already exists
-        print("Downloading service account file from Hugging Face...")
-        response = requests.get(HUGGINGFACE_JSON_URL)
-        if response.status_code == 200:
-            with open(SERVICE_ACCOUNT_FILE, 'wb') as f:
-                f.write(response.content)
-            print(f"Service account file downloaded: {SERVICE_ACCOUNT_FILE}")
-        else:
-            raise Exception(f"Failed to download service account file: {response.status_code}, {response.text}")
-    else:
-        print(f"Service account file already exists: {SERVICE_ACCOUNT_FILE}")
-
-# Download the service account file
-download_service_account_file()
-
-# Authenticate with the Google API
-def authenticate_google_drive():
-    """Authenticate and return the Google Drive service object."""
-    credentials = Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE, scopes=SCOPES)
-    drive_service = build('drive', 'v3', credentials=credentials)
-    print("Google Drive API authenticated successfully!")
-    return drive_service
-
-# Example usage of the authenticated service
-drive_service = authenticate_google_drive()
-
-# Example API call: List files in Google Drive
-def list_drive_files():
-    """List files in Google Drive."""
-    results = drive_service.files().list(pageSize=10).execute()
-    items = results.get('files', [])
-    if not items:
-        print("No files found.")
-    else:
-        print("Files:")
-        for item in items:
-            print(f"{item['name']} ({item['id']})")
-
-# Call the function to list files
-list_drive_files()
+from flask import Flask, jsonify
+from google.cloud import storage
+
+# Check if you're using Hugging Face Secrets or a local file
+service_account_secret = os.getenv("SERVICE_ACCOUNT_JSON")
+if service_account_secret:
+    # Save the service account from the secret to a file
+    with open("service-account.json", "w") as f:
+        f.write(service_account_secret)
+
+# Set the environment variable to point to the service account file
+os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "service-account.json"
+
+# Function to list Google Cloud Storage buckets
+def list_gcs_buckets():
+    try:
+        client = storage.Client()
+        buckets = list(client.list_buckets())
+        return [bucket.name for bucket in buckets]
+    except Exception as e:
+        return str(e)
+
+# Flask app to display bucket names
+app = Flask(__name__)
+
+@app.route("/")
+def home():
+    buckets = list_gcs_buckets()
+    if isinstance(buckets, str):  # If an error occurred
+        return jsonify({"error": buckets}), 500
+    return jsonify({"buckets": buckets})
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=8080)