Create app.py

app.py

@@ -0,0 +1,75 @@
+import gradio as gr
+import requests
+
+def check_key_gemini_availability(key):
+    """
+    Checks the availability of a Gemini API key.
+
+    Args:
+        key: The API key to check.
+
+    Returns:
+        A tuple containing a boolean indicating whether the key is valid and an error message (or None if the key is valid).
+    """
+    try:
+        url_getListModel = f"https://generativelanguage.googleapis.com/v1beta/models?key={key}"
+        rq = requests.get(url_getListModel)
+        rq.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
+        result = rq.json()
+        if 'models' in result:
+            return True, None
+        else:
+            return False, "Invalid key: 'models' field not found in response."
+    except requests.exceptions.HTTPError as e:
+        return False, f"Invalid key: HTTP error - {e.response.status_code}"
+    except Exception as e:
+        return False, f"Error: {e}"
+
+def check_keys(keys_text):
+    """
+    Checks a list of Gemini API keys and categorizes them into valid and invalid.
+
+    Args:
+        keys_text: A string containing API keys separated by lines.
+
+    Returns:
+        A tuple containing two strings: a list of valid keys and a list of invalid keys with error messages.
+    """
+    key_list = keys_text.strip().splitlines()
+    key_list = list(set(key_list))  # Remove duplicates
+
+    valid_keys = []
+    invalid_keys = []
+
+    for key in key_list:
+        is_valid, error_msg = check_key_gemini_availability(key)
+        if is_valid:
+            valid_keys.append(key)
+        else:
+            invalid_keys.append(f"{key} - {error_msg}")
+
+    return "\n".join(valid_keys), "\n".join(invalid_keys)
+
+def clear_inputs():
+    """Clears the input text area."""
+    return ""
+
+with gr.Blocks() as demo:
+    gr.Markdown('''
+    # Gemini API Key Status Checker - batch mode
+    ''')
+
+    with gr.Row():
+        with gr.Column():
+            keys = gr.TextArea(label="API Keys (by lines)")
+            with gr.Row():
+                clear_button = gr.Button("Clear")
+                submit_button = gr.Button("Submit", variant="primary")
+        with gr.Column():
+            infos = gr.TextArea(label="Alive Keys")
+            errs = gr.TextArea(label="Invalid Keys-status code")
+
+    clear_button.click(fn=clear_inputs, outputs=[keys])
+    submit_button.click(fn=check_keys, inputs=[keys], outputs=[infos, errs], api_name="check_keys")
+
+demo.launch()