inti

app.py

@@ -0,0 +1,58 @@
+import gradio as gr
+from google.cloud import storage
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
+import openai
+import os
+import time
+
+# 初始化API和認證
+api_key = 'your_openai_api_key'
+openai_client = openai.Client(api_key=api_key)
+gcs_client = storage.Client.from_service_account_json('your_gcs_credentials.json')
+bucket_name = 'your_gcs_bucket_name'
+bucket = gcs_client.bucket(bucket_name)
+
+scope = ['https://spreadsheets.google.com/feeds', 'https://www.googleapis.com/auth/drive']
+creds = ServiceAccountCredentials.from_json_keyfile_name('your_google_sheets_credentials.json', scope)
+sheets_client = gspread.authorize(creds)
+
+# 函数定义
+def upload_image_to_gcs(image_file, bucket):
+    # 为了避免文件名冲突，使用时间戳创建一个唯一的文件名
+    unique_filename = f"{int(time.time())}_{image_file.name}"
+    blob = bucket.blob(unique_filename)
+    blob.upload_from_string(image_file.read(), content_type=image_file.type)
+    blob.make_public()
+    return blob.public_url
+
+def process_image(image_url):
+    # 这里调用OpenAI API或其他图像处理逻辑
+    text = f"处理结果: {image_url}"  # 示例文本，实际应用中应调用API获取结果
+    return text
+
+def process_and_upload(images, sheet_url):
+    sheet = sheets_client.open_by_url(sheet_url).sheet1
+
+    for image_file in images:
+        image_url = upload_image_to_gcs(image_file, bucket)
+        text = process_image(image_url)
+        sheet.append_row([image_url, text])
+
+    return "处理完成，结果已上传到Google Sheets"
+
+# Gradio界面
+with gr.Blocks() as demo:
+    gr.Markdown("## 批量上传图片并处理")
+    with gr.Row():
+        image_input = gr.File(label="选择图片", type="file", multiple=True)
+        sheet_input = gr.Textbox(label="Google Sheets URL")
+        submit_button = gr.Button("开始处理")
+
+    submit_button.click(
+        fn=process_and_upload,
+        inputs=[image_input, sheet_input],
+        outputs=gr.Markdown()
+    )
+
+demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+gradio 
+google-cloud-storage 
+gspread 
+oauth2client
+openai