Update app.py

app.py

@@ -7,6 +7,8 @@ def fetch_gemini_models(api_key):
     Fetches available Gemini models and returns them as a pandas DataFrame.
     """
 
+
+
     if not api_key:
         return None, "エラー: GEMINI_API_KEY が必要です。"
 
@@ -31,7 +33,9 @@ def fetch_gemini_models(api_key):
                 translated_methods = [method_translation.get(method, method) for method in m.supported_actions]
                 methods = ", ".join(translated_methods)
                 
+                # Checkbox column first
                 data.append({
+                    "選択": False,
                     "モデル名": m.name,
                     "表示名": m.display_name,
                     "対応メソッド": methods
@@ -47,25 +51,45 @@ def fetch_gemini_models(api_key):
     except Exception as e:
         return None, f"エラーが発生しました: {str(e)}"
 
-def check_model_access(api_key, model_name):
+def check_selected_models(api_key, df):
     """
-    Checks if the API key has access to the specified model by attempting a simple generation.
+    Checks access for all selected models in the dataframe.
     """
     if not api_key:
         return "エラー: GEMINI_API_KEY が必要です。"
-    if not model_name:
-        return "エラー: モデル名が必要です。"
+    if df is None or df.empty:
+        return "エラー: モデル一覧が空です。"
 
-    try:
-        client = genai.Client(api_key=api_key)
-        # Try a minimal generation request
-        response = client.models.generate_content(
-            model=model_name,
-            contents="test",
-        )
-        return f"成功: {model_name} へのアクセスが確認されました。応答を受信しました。"
-    except Exception as e:
-        return f"アクセス確認失敗: {str(e)}"
+    # Filter selected rows
+    # Gradio dataframe usually sends back a pandas dataframe or list of lists
+    if isinstance(df, dict): # Handle some Gradio versions wrapping it
+        df = pd.DataFrame(df['data'], columns=df['headers'])
+    
+    # Check if "選択" column exists
+    if "選択" not in df.columns:
+        return "エラー: モデル一覧の形式が不正です。"
+
+    selected_rows = df[df["選択"] == True]
+    
+    if selected_rows.empty:
+        return "警告: チェックするモデルが選択されていません。"
+
+    results = []
+    client = genai.Client(api_key=api_key)
+
+    for index, row in selected_rows.iterrows():
+        model_name = row["モデル名"]
+        try:
+            # Try a minimal generation request
+            response = client.models.generate_content(
+                model=model_name,
+                contents="test",
+            )
+            results.append(f"✅ {model_name}: アクセス確認成功")
+        except Exception as e:
+            results.append(f"❌ {model_name}: 失敗 - {str(e)}")
+    
+    return "\n".join(results)
 
 # Define Gradio Theme (Rich Aesthetics)
 theme = gr.themes.Soft(
@@ -96,7 +120,15 @@ with gr.Blocks(theme=theme, title="Gemini モデルチェッカー") as demo:
         fetch_btn = gr.Button("モデル一覧取得", variant="primary", scale=1)
     
     output_status = gr.Textbox(label="ステータス", interactive=False)
-    output_table = gr.DataFrame(label="利用可能なモデル")
+    # Make dataframe interactive to allow checkbox editing
+    output_table = gr.DataFrame(
+        label="利用可能なモデル (左端のチェックボックスで選択)",
+        interactive=True,
+        type="pandas",
+        col_count=(4, "fixed"), # 選択, モデル名, 表示名, 対応メソッド
+        headers=["選択", "モデル名", "表示名", "対応メソッド"],
+        datatype=["bool", "str", "str", "str"]
+    )
 
     fetch_btn.click(
         fn=fetch_gemini_models,
@@ -105,20 +137,14 @@ with gr.Blocks(theme=theme, title="Gemini モデルチェッカー") as demo:
     )
 
     gr.Markdown("## モデルアクセス確認")
-    with gr.Row():
-        model_name_input = gr.Textbox(
-            label="モデル名",
-            placeholder="gemini-1.5-pro-001",
-            interactive=True,
-            scale=4
-        )
-        check_btn = gr.Button("アクセス確認", variant="secondary", scale=1)
+    gr.Markdown("上の表でチェックを入れたモデルに対して、一括でアクセス確認を行います。")
     
-    access_status = gr.Textbox(label="アクセス確認結果", interactive=False)
+    check_btn = gr.Button("選択したモデルのアクセス確認", variant="secondary")
+    access_status = gr.Textbox(label="アクセス確認結果", interactive=False, lines=10)
 
     check_btn.click(
-        fn=check_model_access,
-        inputs=[api_key_input, model_name_input],
+        fn=check_selected_models,
+        inputs=[api_key_input, output_table],
         outputs=access_status
     )
 
@@ -128,7 +154,7 @@ with gr.Blocks(theme=theme, title="Gemini モデルチェッカー") as demo:
         ### 使い方
         1. Google AI Studioから取得��た **APIキー** を入力します。
         2. **モデル一覧取得** ボタンをクリックします。利用可能なモデルの一覧が表示されます。
-        3. 特定のモデルへのアクセスを確認するには、**モデルアクセス確認** セクションで **モデル名** を入力し、**アクセス確認** ボタンをクリックします。
+        3. 特定のモデルへのアクセスを確認するには、**モデルアクセス確認** セクションで **モデル名** を選択（または入力）し、**アクセス確認** ボタンをクリックします。
         """
     )