Create app.py

app.py

@@ -0,0 +1,237 @@
+import os
+import tempfile
+import gradio as gr
+import pandas as pd
+from urllib.parse import urlparse, parse_qs
+from PIL import Image
+from io import BytesIO
+import requests
+from google import genai
+
+# --- Google Sheets 相關函式 ---
+def read_google_sheet(sheet_url: str):
+    """
+    從 Google Sheet 的 URL 讀取資料。
+    """
+    if not sheet_url:
+        raise gr.Error("請提供 Google Sheet URL。")
+    try:
+        def build_csv_url(url: str) -> str:
+            parsed = urlparse(url)
+            doc_id = parsed.path.strip("/").split("/")[2] if len(parsed.path.strip("/").split("/")) >= 3 and parsed.path.strip("/").split("/")[1] == "d" else None
+            gid = parse_qs(parsed.query).get("gid", [None])[0] or parse_qs(parsed.fragment).get("gid", [None])[0] or "0"
+            if doc_id:
+                return f"https://docs.google.com/spreadsheets/d/{doc_id}/export?format=csv&gid={gid}"
+            if "/export" in parsed.path and "format=csv" in parsed.query:
+                return url
+            return url.replace("/edit#gid=0", "/export?format=csv&gid=0")
+
+        csv_url = build_csv_url(sheet_url)
+        # 增加 read_csv 的 robustness，防止某些 Google Sheet 導致的 ParserError
+        df = pd.read_csv(csv_url, engine='python', on_bad_lines='warn', encoding='utf-8')
+        return df
+    except Exception as e:
+        raise gr.Error(f"讀取 Google Sheet 時發生錯誤: {e}")
+
+def process_sheet_data(sheet_url):
+    """處理試算表資料，為 Gradio DataFrame 準備。"""
+    try:
+        df = read_google_sheet(sheet_url)
+        # 檢查是否至少有 3 列 (白背圖URL, 參考圖URL, 提示詞)，因為索引會在程式中生成
+        if df.shape[1] < 3:
+            error_msg = f"錯誤：Google Sheet 至少需要 3 列 (白背圖URL, 參考圖URL, 提示詞)。目前只有 {df.shape[1]} 列。"
+            raise gr.Error(error_msg)
+        
+        data_list = []
+        for i, row in df.iterrows():
+            # 確保有足夠的列且至少第一列(白背圖URL)不為空才加入
+            if len(row) >= 3 and pd.notna(row.iloc[0]):
+                data_list.append([i + 2, row.iloc[0], row.iloc[1], row.iloc[2]])
+        
+        log_message = f"成功讀取 {len(data_list)} 筆數據。"
+        return data_list, log_message
+    except Exception as e:
+        raise gr.Error(f"處理試算表資料時發生錯誤: {e}")
+
+def get_row_data(sheet_url, row_number):
+    """從 Google Sheet 讀取指定列的資料。"""
+    try:
+        df = read_google_sheet(sheet_url)
+        # Gradio顯示的行數從2開始，DataFrame索引從0開始，所以需要減2
+        row_index = int(row_number) - 2 
+        if row_index < 0 or row_index >= df.shape[0]:
+            raise gr.Error(f"指定的行數 {row_number} 不存在。")
+        
+        row_data = df.iloc[row_index]
+        # 確保取值時索引不越界，並處理 NaN
+        white_back_image_url = row_data.iloc[0] if len(row_data) > 0 and pd.notna(row_data.iloc[0]) else ""
+        ref_image_url = row_data.iloc[1] if len(row_data) > 1 and pd.notna(row_data.iloc[1]) else ""
+        prompt_text = row_data.iloc[2] if len(row_data) > 2 and pd.notna(row_data.iloc[2]) else ""
+        
+        return white_back_image_url, ref_image_url, prompt_text
+    except Exception as e:
+        raise gr.Error(f"讀取指定行時發生錯誤: {e}")
+
+# --- 下載圖片函式 ---
+def load_image_from_url(url: str):
+    """從 URL 下載圖片並以 PIL Image 格式回傳。"""
+    if not url or not isinstance(url, str) or not url.strip(): # 增加對空字串的檢查
+        return None
+    try:
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
+        }
+        response = requests.get(url, timeout=20, headers=headers)
+        response.raise_for_status()
+        image = Image.open(BytesIO(response.content)).convert("RGB")
+        return image
+    except requests.exceptions.HTTPError as e:
+        gr.Warning(f"下載圖片失敗：HTTP 錯誤 {e.response.status_code}")
+        return None
+    except Exception as e:
+        gr.Warning(f"下載圖片時發生意外錯誤：{e}")
+        return None
+
+# --- Gemini 核心函式 ---
+def generate_image(text, images, api_key, model="gemini-2.5-flash-image-preview"):
+    """使用 Gemini 模型生成圖片。"""
+    if not api_key or api_key.strip() == "":
+        raise gr.Error("請輸入有效的 Gemini API 金鑰。", duration=10)
+    try:
+        client = genai.Client(api_key=api_key.strip())
+        contents = images + [text]
+        response = client.models.generate_content(model=model, contents=contents)
+        text_response = ""
+        image_path = None
+        for part in response.candidates[0].content.parts:
+            if part.text is not None:
+                text_response += part.text + "\n"
+            elif part.inline_data is not None:
+                with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
+                    temp_path = tmp.name
+                    generated_image = Image.open(BytesIO(part.inline_data.data))
+                    generated_image.save(temp_path)
+                    image_path = temp_path
+        return image_path, text_response
+    except Exception as e:
+        raise gr.Error(f"Gemini API 呼叫失敗: {e}", duration=10)
+
+# --- Gradio 互動函式 ---
+def generate_image_from_row(sheet_url, row_number, gemini_api_key):
+    """
+    根據指定的行數，生成圖片，並返回白背圖、參考圖和生成的圖片。
+    """
+    if not sheet_url:
+        raise gr.Error("請先輸入 Google Sheet URL。", duration=5)
+    if not row_number or row_number <= 1:
+        raise gr.Error("請輸入有效的行數 (大於 1)。", duration=5)
+        
+    try:
+        white_back_url, ref_image_url, prompt_text = get_row_data(sheet_url, row_number)
+        
+        log_message = f"開始處理第 {row_number} 行...\n"
+        log_message += f"白背圖URL: {white_back_url if white_back_url else '無'}\n"
+        log_message += f"參考圖URL: {ref_image_url if ref_image_url else '無'}\n"
+        log_message += f"提示詞: {prompt_text if prompt_text else '無'}\n"
+
+        images_for_gemini = []
+        
+        # 下載圖片
+        wb_img = None
+        if white_back_url:
+            wb_img = load_image_from_url(white_back_url)
+            if wb_img:
+                images_for_gemini.append(wb_img)
+            else:
+                log_message += f"警告：無法下載白背圖 '{white_back_url}'。\n"
+
+        ref_img = None
+        if ref_image_url:
+            ref_img = load_image_from_url(ref_image_url)
+            if ref_img:
+                images_for_gemini.append(ref_img)
+            else:
+                log_message += f"警告：無法下載參考圖 '{ref_image_url}'。\n"
+
+        if not images_for_gemini:
+            # 如果沒有任何圖片，但有提示詞，嘗試純文字生成
+            if prompt_text:
+                log_message += "沒有可用的輸入圖片，嘗試僅使用提示詞生成圖片。\n"
+            else:
+                return None, None, None, "警告：無效的圖片 URL 且無提示詞，無法生成圖片。", log_message
+
+        log_message += "圖片已下載，開始呼叫 Gemini 模型...\n"
+        
+        image_path, text_response = generate_image(text=prompt_text, images=images_for_gemini, api_key=gemini_api_key)
+
+        if image_path:
+            log_message += "圖片生成成功！"
+            return wb_img, ref_img, image_path, log_message
+        else:
+            log_message += "圖片生成失敗。\n"
+            log_message += f"Gemini 文字回應: {text_response}"
+            return wb_img, ref_img, None, log_message
+            
+    except ValueError:
+        return None, None, None, "行數必須為數字。", "生成圖片時發生錯誤：行數必須為數字。"
+    except Exception as e:
+        return None, None, None, f"生成圖片時發生錯誤: {e}", f"生成圖片時發生錯誤: {e}"
+
+# --- Gradio 介面設定 ---
+with gr.Blocks() as demo:
+    gr.Markdown("## Google Sheets 圖片生成器")
+    
+    with gr.Row(elem_classes="main-content"):
+        with gr.Column(elem_classes="input-column"):
+            gemini_api_key = gr.Textbox(
+                lines=1,
+                placeholder="請在此輸入你的 Gemini API 金鑰",
+                label="Gemini API 金鑰",
+                elem_classes="api-key-input",
+                # 從環境變數讀取預設值 (如果存在)
+                value=os.environ.get("GEMINI_API_KEY", "") 
+            )
+            sheet_url_input = gr.Textbox(
+                label="Google Sheet URL", 
+                value="https://docs.google.com/spreadsheets/d/1G3olHxydDIbnyXdh5nnw5TG0akZFeMeYm-25JmCGDLg/edit?gid=0#gid=0"
+            )
+            process_button = gr.Button("處理試算表", elem_classes="generate-btn")
+            
+            with gr.Row():
+                row_index_input = gr.Number(label="要生成的行數", precision=0, value=2)
+                generate_selected_button = gr.Button("生成所選行的圖片", elem_classes="generate-btn")
+        
+        with gr.Column(elem_classes="output-column"):
+            output_dataframe = gr.DataFrame(
+                headers=["Index", "白背圖URL", "參考圖URL", "提示詞"],
+                col_count=(4, "fixed"),
+                interactive=False,
+                label="已處理的試算表數據"
+            )
+            
+            # 新增顯示白背圖和參考圖的欄位
+            with gr.Row():
+                original_white_back_output = gr.Image(label="原始白背圖", elem_classes="output-gallery", interactive=False)
+                original_ref_image_output = gr.Image(label="原始參考圖", elem_classes="output-gallery", interactive=False)
+            
+            generated_image_output = gr.Image(label="生成的圖片", elem_classes="output-gallery", interactive=False)
+            operation_log_output = gr.Textbox(
+                label="操作日誌", 
+                lines=10,
+                placeholder="文字回應和日誌會顯示在這裡。"
+            )
+
+    # 按鈕的事件綁定
+    process_button.click(
+        fn=process_sheet_data,
+        inputs=[sheet_url_input],
+        outputs=[output_dataframe, operation_log_output]
+    )
+
+    generate_selected_button.click(
+        fn=generate_image_from_row,
+        inputs=[sheet_url_input, row_index_input, gemini_api_key],
+        outputs=[original_white_back_output, original_ref_image_output, generated_image_output, operation_log_output]
+    )
+    
+    demo.queue().launch(mcp_server=True, share=True)