Update app.py

app.py

@@ -6,8 +6,9 @@ def fetch_gemini_models(api_key):
     """
     Fetches available Gemini models and returns them as a pandas DataFrame.
     """
+
     if not api_key:
-        return None, "Error: GEMINI_API_KEY is required."
+        return None, "エラー: GEMINI_API_KEY が必要です。"
 
     try:
         client = genai.Client(api_key=api_key)
@@ -31,19 +32,40 @@ def fetch_gemini_models(api_key):
                 methods = ", ".join(translated_methods)
                 
                 data.append({
-                    "Model Name": m.name,
-                    "Display Name": m.display_name,
-                    "Supported Methods (JP)": methods
+                    "モデル名": m.name,
+                    "表示名": m.display_name,
+                    "対応メソッド": methods
                 })
         
         if not data:
-            return None, "No models found that support 'generateContent'."
+            return None, "'generateContent' をサポートするモデルが見つかりませんでした。"
         
         df = pd.DataFrame(data)
-        return df, "Success: Models fetched successfully."
+        return df, "成功: モデルの取得に成功しました。"
             
+
     except Exception as e:
-        return None, f"An error occurred: {str(e)}"
+        return None, f"エラーが発生しました: {str(e)}"
+
+def check_model_access(api_key, model_name):
+    """
+    Checks if the API key has access to the specified model by attempting a simple generation.
+    """
+    if not api_key:
+        return "エラー: GEMINI_API_KEY が必要です。"
+    if not model_name:
+        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)}"
 
 # Define Gradio Theme (Rich Aesthetics)
 theme = gr.themes.Soft(
@@ -55,26 +77,26 @@ theme = gr.themes.Soft(
     button_primary_background_fill_hover='*primary_700',
 )
 
-with gr.Blocks(theme=theme, title="Gemini Model Checker") as demo:
+with gr.Blocks(theme=theme, title="Gemini モデルチェッカー") as demo:
     gr.Markdown(
         """
-        # 💎 Gemini Model Checker
+        # 💎 Gemini モデルチェッカー
         Hugging Face Spaces上のGradioで、利用可能なGeminiモデルを簡単に確認できます。
         """
     )
     
     with gr.Row():
         api_key_input = gr.Textbox(
-            label="Gemini API Key",
+            label="Gemini APIキー",
             placeholder="AIzaSy...",
             type="password",
             interactive=True,
             scale=4
         )
-        fetch_btn = gr.Button("Fetch Models", variant="primary", scale=1)
+        fetch_btn = gr.Button("モデル一覧取得", variant="primary", scale=1)
     
-    output_status = gr.Textbox(label="Status", interactive=False)
-    output_table = gr.DataFrame(label="Available Models")
+    output_status = gr.Textbox(label="ステータス", interactive=False)
+    output_table = gr.DataFrame(label="利用可能なモデル")
 
     fetch_btn.click(
         fn=fetch_gemini_models,
@@ -82,13 +104,31 @@ with gr.Blocks(theme=theme, title="Gemini Model Checker") as demo:
         outputs=[output_table, output_status]
     )
 
+    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)
+    
+    access_status = gr.Textbox(label="アクセス確認結果", interactive=False)
+
+    check_btn.click(
+        fn=check_model_access,
+        inputs=[api_key_input, model_name_input],
+        outputs=access_status
+    )
+
     gr.Markdown(
         """
         ---
         ### 使い方
         1. Google AI Studioから取得した **APIキー** を入力します。
-        2. **Fetch Models** ボタンをクリックします。
-        3. 利用可能なモデルの一覧が表示されます。
+        2. **モデル一覧取得** ボタンをクリックします。利用可能なモデルの一覧が表示されます。
+        3. 特定のモデルへのアクセスを確認するには、**モデルアクセス確認** セクションで **モデル名** を入力し、**アクセス確認** ボタンをクリックします。
         """
     )