Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,6 @@ from google.oauth2 import service_account
|
|
| 13 |
|
| 14 |
# 由於您的環境可能沒有服務帳號金鑰,我們將直接使用 GEMINI_API_KEY
|
| 15 |
# 如果在 Hugging Face Space 中,這通常會從環境變數中自動加載
|
| 16 |
-
# 您仍然需要設置 GEMINI_API_KEY 作為您的 API 金鑰
|
| 17 |
|
| 18 |
# 初始化 AI Platform 服務客戶端
|
| 19 |
def initialize_ai_platform(api_key):
|
|
@@ -21,11 +20,7 @@ def initialize_ai_platform(api_key):
|
|
| 21 |
PROJECT_ID = "YOUR_GCP_PROJECT_ID" # 這裡需要您提供真實的GCP專案ID
|
| 22 |
LOCATION = "us-central1"
|
| 23 |
|
| 24 |
-
# 這裡的認證方式取決於您的環境。
|
| 25 |
-
# 這裡我們用一個假定的方式來適應 Gradio 環境,但實際上您可能需要在 GCP 內部運行。
|
| 26 |
-
# 如果您只是使用 API 金鑰,這部分會更簡單,但 'google-cloud-aiplatform' 函式庫
|
| 27 |
-
# 主要設計用於 GCP 環境,所以這可能會比較複雜。
|
| 28 |
-
|
| 29 |
# 創建一個不帶服務帳號憑證的客戶端,這會讓它嘗試從環境中自動尋找憑證
|
| 30 |
try:
|
| 31 |
aiplatform.init(project=PROJECT_ID, location=LOCATION)
|
|
@@ -34,7 +29,7 @@ def initialize_ai_platform(api_key):
|
|
| 34 |
print(f"Error initializing AI Platform: {e}")
|
| 35 |
return False
|
| 36 |
|
| 37 |
-
# 全局變數來儲存 API 金鑰
|
| 38 |
GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
|
| 39 |
|
| 40 |
if not GEMINI_API_KEY:
|
|
@@ -49,8 +44,8 @@ if not GEMINI_API_KEY:
|
|
| 49 |
raise ValueError("ERROR: GEMINI_API_KEY environment variable is not set. Please set it correctly.")
|
| 50 |
|
| 51 |
# 這裡我們將創建一個通用客戶端來處理 API 呼叫
|
| 52 |
-
# 由於 'google-cloud-aiplatform'
|
| 53 |
-
# 直接呼叫 API,以適應您在 Hugging Face Space 的需求。
|
| 54 |
class NanoBananaImageGenerator:
|
| 55 |
def __init__(self, api_key):
|
| 56 |
self.api_key = api_key
|
|
|
|
| 13 |
|
| 14 |
# 由於您的環境可能沒有服務帳號金鑰,我們將直接使用 GEMINI_API_KEY
|
| 15 |
# 如果在 Hugging Face Space 中,這通常會從環境變數中自動加載
|
|
|
|
| 16 |
|
| 17 |
# 初始化 AI Platform 服務客戶端
|
| 18 |
def initialize_ai_platform(api_key):
|
|
|
|
| 20 |
PROJECT_ID = "YOUR_GCP_PROJECT_ID" # 這裡需要您提供真實的GCP專案ID
|
| 21 |
LOCATION = "us-central1"
|
| 22 |
|
| 23 |
+
# 這裡的認證方式取決於您的環境。
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
# 創建一個不帶服務帳號憑證的客戶端,這會讓它嘗試從環境中自動尋找憑證
|
| 25 |
try:
|
| 26 |
aiplatform.init(project=PROJECT_ID, location=LOCATION)
|
|
|
|
| 29 |
print(f"Error initializing AI Platform: {e}")
|
| 30 |
return False
|
| 31 |
|
| 32 |
+
# 全局變數來儲存 API 金鑰
|
| 33 |
GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
|
| 34 |
|
| 35 |
if not GEMINI_API_KEY:
|
|
|
|
| 44 |
raise ValueError("ERROR: GEMINI_API_KEY environment variable is not set. Please set it correctly.")
|
| 45 |
|
| 46 |
# 這裡我們將創建一個通用客戶端來處理 API 呼叫
|
| 47 |
+
# 由於 'google-cloud-aiplatform' 函式庫主要設計用於 GCP 環境,我們將使用 `requests`
|
| 48 |
+
# 來模擬直接呼叫 API,以適應您在 Hugging Face Space 的需求。
|
| 49 |
class NanoBananaImageGenerator:
|
| 50 |
def __init__(self, api_key):
|
| 51 |
self.api_key = api_key
|