Update app.py

app.py

@@ -6,8 +6,9 @@ from io import BytesIO
 from PIL import Image
 from urllib.parse import urlparse, parse_qs
 import google.generativeai as genai
+import time
 
-# 全局變數來儲存 API 金鑰
+# 全域變數來儲存 API 金鑰
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 
 if not GEMINI_API_KEY:
@@ -119,7 +120,6 @@ def generate_image(white_background_url, reference_image_url, prompt):
         output_dir = "generated_images"
         os.makedirs(output_dir, exist_ok=True)
         # 使用時間戳或唯一ID來確保檔名唯一
-        import time
         output_path = os.path.join(output_dir, f"generated_{int(time.time())}.png")
         with open(output_path, "wb") as f:
             f.write(generated_images_binary[0])
@@ -163,10 +163,8 @@ def process_sheet_data(sheet_url):
             error_msg = f"Error: Google Sheet has only {df.shape[1]} columns, but 3 are expected (White Background URL, Reference Image URL, Prompt)."
             raise gr.Error(error_msg)
 
-        # 這裡使用 to_dict('records') 來處理，以確保後續 Pandas 處理時的兼容性
-        data = df.to_dict('records')
         data_list = []
-        for i, row in enumerate(data):
+        for i, row in df.iterrows():
             if pd.notna(row.iloc[0]) and pd.notna(row.iloc[1]) and pd.notna(row.iloc[2]):
                 data_list.append([i, row.iloc[0], row.iloc[1], row.iloc[2]])
 
@@ -176,7 +174,6 @@ def process_sheet_data(sheet_url):
 
 def generate_image_for_row(row_index, dataframe_data):
     """根據 Gradio DataFrame 的行索引生成圖片。"""
-    # 將 Gradio 的 dataframe_data 轉換回 pandas DataFrame
     df = pd.DataFrame(dataframe_data, columns=["Index", "白背圖URL", "參考圖URL", "提示詞"])
     
     if not (0 <= row_index < len(df)):
@@ -222,7 +219,6 @@ if __name__ == "__main__":
             outputs=processed_df_state
         )
         
-        # 修正 click 觸發器，確保傳入正確的參數
         generate_selected_button.click(
             fn=generate_image_for_row,
             inputs=[row_index_input, output_dataframe],