os.getenv("OPEN_AI_KEY")

app.py

@@ -1,13 +1,18 @@
 import gradio as gr
-from google.cloud import storage
 import gspread
 from oauth2client.service_account import ServiceAccountCredentials
 from openai import OpenAI
 import os
 import time
+import json
+from google.cloud import storage
+from google.oauth2 import service_account
+from googleapiclient.http import MediaIoBaseDownload
+from storage_service import GoogleCloudStorage
+
 
 # 初始化API和認證
-OPEN_AI_KEY = config["OPEN_AI_KEY"]
+OPEN_AI_KEY = os.getenv("OPEN_AI_KEY")
 OPEN_AI_CLIENT = OpenAI(api_key=OPEN_AI_KEY)
 
 GCS_KEY = os.getenv("GOOGLE_APPLICATION_CREDENTIALS_JSON")
@@ -20,6 +25,11 @@ scope = ['https://spreadsheets.google.com/feeds', 'https://www.googleapis.com/au
 creds = ServiceAccountCredentials.from_json_keyfile_name('your_google_sheets_credentials.json', scope)
 sheets_client = gspread.authorize(creds)
 
+GSHEET_KEY = os.getenv("GOOGLE_APPLICATION_CREDENTIALS_JSON")
+GSHEET_KEY_DICT = json.loads(GSHEET_KEY)
+sheets_client = gspread.service_account_from_dict(GSHEET_KEY_DICT)
+sheet_url = "your_google_sheets_url"
+
 # 函数定义
 def upload_image_to_gcs(image_file, bucket):
     # 为了避免文件名冲突，使用时间戳创建一个唯一的文件名

requirements.txt

@@ -1,5 +1,9 @@
 gradio 
-google-cloud-storage 
 gspread 
-oauth2client
-openai
+openai
+
+google-auth
+google-api-python-client
+google-auth-httplib2
+google-auth-oauthlib
+google-cloud-storage

storage_service.py

@@ -0,0 +1,44 @@
+import json
+from google.cloud import storage
+from google.oauth2 import service_account
+from googleapiclient.http import MediaIoBaseDownload
+
+
+class GoogleCloudStorage:
+    def __init__(self, service_account_key_string):
+        credentials_dict = json.loads(service_account_key_string)
+        credentials = service_account.Credentials.from_service_account_info(credentials_dict)
+        self.client = storage.Client(credentials=credentials, project=credentials_dict['project_id'])
+
+    def check_file_exists(self, bucket_name, file_name):
+        blob = self.client.bucket(bucket_name).blob(file_name)
+        return blob.exists()
+
+    def upload_file(self, bucket_name, destination_blob_name, file_path):
+        blob = self.client.bucket(bucket_name).blob(destination_blob_name)
+        blob.upload_from_filename(file_path)
+        print(f"File {file_path} uploaded to {destination_blob_name} in GCS.")
+
+    def upload_file_as_string(self, bucket_name, destination_blob_name, content):
+        blob = self.client.bucket(bucket_name).blob(destination_blob_name)
+        blob.upload_from_string(content)
+        print(f"String content uploaded to {destination_blob_name} in GCS.")
+        return None
+
+    def download_as_string(self, bucket_name, source_blob_name):
+        blob = self.client.bucket(bucket_name).blob(source_blob_name)
+        return blob.download_as_text()
+
+    def make_blob_public(self, bucket_name, blob_name):
+        blob = self.client.bucket(bucket_name).blob(blob_name)
+        blob.make_public()
+        print(f"Blob {blob_name} is now publicly accessible at {blob.public_url}")
+
+    def get_public_url(self, bucket_name, blob_name):
+        blob = self.client.bucket(bucket_name).blob(blob_name)
+        return blob.public_url
+
+    def upload_image_and_get_public_url(self, bucket_name, file_name, file_path):
+        self.upload_file(bucket_name, file_name, file_path)
+        self.make_blob_public(bucket_name, file_name)
+        return self.get_public_url(bucket_name, file_name)