Create app.py

app.py

@@ -0,0 +1,86 @@
+import gradio as gr
+from huggingface_hub import HfApi
+from huggingface_hub.utils import HfHubHTTPError, RepositoryNotFoundError
+
+# Initialize the Hugging Face Hub API client
+api = HfApi()
+
+def check_model_access(model_id: str, oauth_token: gr.OAuthToken | None):
+    """
+    Checks if the logged-in user has access to a given model on the Hub.
+
+    Args:
+        model_id: The ID of the model to check (e.g., "meta-llama/Llama-2-7b-chat-hf").
+        oauth_token: The user's OAuth token, automatically injected by Gradio.
+    
+    Returns:
+        A string with the result of the access check, formatted for Markdown.
+    """
+    # 1. Check if the user is logged in
+    if oauth_token is None:
+        return "### <span style='color: red;'>Authentication Error 🔴</span>\nPlease log in using the button above to check model access."
+
+    # 2. Check if a model ID was provided
+    if not model_id:
+        return "### <span style='color: orange;'>Input Missing 🟡</span>\nPlease enter a model ID to check."
+
+    try:
+        # 3. The core of the test: try to get model info using the user's token
+        # This single API call is enough to verify read access.
+        model_info = api.model_info(repo_id=model_id, token=oauth_token.token)
+        
+        # 4. If the call succeeds, format a success message
+        return f"""
+        ### <span style='color: green;'>Access Granted ✅</span>
+        You have access to **{model_id}**.
+        
+        - **Author:** {model_info.author}
+        - **Last Modified:** {model_info.last_modified}
+        - **Gated:** {'Yes' if model_info.gated else 'No'}
+        """
+    
+    except RepositoryNotFoundError:
+        # 5. Handle the case where the repository does not exist
+        return f"### <span style='color: red;'>Not Found 🔴</span>\nThe repository **{model_id}** does not exist."
+        
+    except HfHubHTTPError as e:
+        # 6. Handle HTTP errors, which typically indicate permission issues for gated models
+        if e.response.status_code == 401 or e.response.status_code == 403:
+            return f"""
+            ### <span style='color: red;'>Access Denied 🔴</span>
+            You do not have access to **{model_id}**.
+            
+            - Please ensure you have accepted the terms and conditions on the model's page.
+            - This might be a private model you don't have permission for.
+            - **Status Code:** {e.response.status_code}
+            """
+        else:
+            return f"### <span style='color: red;'>An Error Occurred 🔴</span>\n**Details:** {str(e)}"
+
+# --- Gradio Interface ---
+with gr.Blocks(css="h1 { text-align: center; }") as demo:
+    gr.Markdown("# Gated Model Access Tester")
+    gr.Markdown("Log in with your Hugging Face account and enter a model ID to see if you have access.")
+    
+    # The LoginButton is the key to enabling OAuth
+    gr.LoginButton()
+
+    with gr.Row():
+        model_id_input = gr.Textbox(
+            label="Model ID", 
+            placeholder="e.g., meta-llama/Llama-2-7b-chat-hf",
+            scale=3,
+        )
+        check_button = gr.Button("Check Access", variant="primary", scale=1)
+
+    result_display = gr.Markdown("### Result will be displayed here.")
+
+    # When the button is clicked, call the check_model_access function.
+    # Gradio automatically finds the oauth_token from the session and passes it.
+    check_button.click(
+        fn=check_model_access,
+        inputs=[model_id_input],
+        outputs=[result_display]
+    )
+
+demo.launch()