Create app.py

app.py

@@ -0,0 +1,96 @@
+import gradio as gr
+from google import genai
+import pandas as pd
+
+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."
+
+    try:
+        client = genai.Client(api_key=api_key)
+        models = list(client.models.list())
+        models.sort(key=lambda x: x.name)
+
+        # Mapping for Japanese translation
+        method_translation = {
+            "generateContent": "コンテンツ生成",
+            "countTokens": "トークン計算",
+            "createCachedContent": "キャッシュ作成",
+            "embedText": "テキスト埋め込み",
+            "batchGenerate": "バッチ生成",
+            "embedContent": "コンテンツ埋め込み"
+        }
+
+        data = []
+        for m in models:
+            if hasattr(m, "supported_actions") and m.supported_actions and "generateContent" in m.supported_actions:
+                translated_methods = [method_translation.get(method, method) for method in m.supported_actions]
+                methods = ", ".join(translated_methods)
+                
+                data.append({
+                    "Model Name": m.name,
+                    "Display Name": m.display_name,
+                    "Supported Methods (JP)": methods
+                })
+        
+        if not data:
+            return None, "No models found that support 'generateContent'."
+        
+        df = pd.DataFrame(data)
+        return df, "Success: Models fetched successfully."
+            
+    except Exception as e:
+        return None, f"An error occurred: {str(e)}"
+
+# Define Gradio Theme (Rich Aesthetics)
+theme = gr.themes.Soft(
+    primary_hue="indigo",
+    secondary_hue="blue",
+    neutral_hue="slate",
+).set(
+    button_primary_background_fill='*primary_600',
+    button_primary_background_fill_hover='*primary_700',
+)
+
+with gr.Blocks(theme=theme, title="Gemini Model Checker") as demo:
+    gr.Markdown(
+        """
+        # 💎 Gemini Model Checker
+        Hugging Face Spaces上のGradioで、利用可能なGeminiモデルを簡単に確認できます。
+        """
+    )
+    
+    with gr.Row():
+        api_key_input = gr.Textbox(
+            label="Gemini API Key",
+            placeholder="AIzaSy...",
+            type="password",
+            interactive=True,
+            scale=4
+        )
+        fetch_btn = gr.Button("Fetch Models", variant="primary", scale=1)
+    
+    output_status = gr.Textbox(label="Status", interactive=False)
+    output_table = gr.DataFrame(label="Available Models")
+
+    fetch_btn.click(
+        fn=fetch_gemini_models,
+        inputs=api_key_input,
+        outputs=[output_table, output_status]
+    )
+
+    gr.Markdown(
+        """
+        ---
+        ### 使い方
+        1. Google AI Studioから取得した **APIキー** を入力します。
+        2. **Fetch Models** ボタンをクリックします。
+        3. 利用可能なモデルの一覧が表示されます。
+        """
+    )
+
+if __name__ == "__main__":
+    demo.launch()