Spaces:
Sleeping
Sleeping
File size: 3,208 Bytes
74dd3f1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
import streamlit as st
import os
from utils.api_client import HuggingFaceClient
def check_authentication():
"""Handle user authentication with Hugging Face API"""
st.markdown(
"""
<div style="text-align: center; margin-bottom: 30px;">
<h1>🤗 Hugging Face Model Manager</h1>
<p>Manage your machine learning models and publish to the Hugging Face Model Hub</p>
</div>
""",
unsafe_allow_html=True,
)
st.markdown(
"""
<div style="background-color: #F9FAFB; padding: 20px; border-radius: 10px; border: 1px solid #E5E7EB;">
<h3 style="margin-top: 0;">Welcome to Hugging Face Model Manager</h3>
<p>This application allows you to:</p>
<ul>
<li>Create and manage model repositories</li>
<li>Upload and publish models to Hugging Face Hub</li>
<li>Update model metadata and documentation</li>
<li>Organize your models with tags and descriptions</li>
</ul>
</div>
""",
unsafe_allow_html=True,
)
with st.form("auth_form"):
st.subheader("Login with Hugging Face API Token")
# Info alert about creating a token
st.info(
"""
To use this application, you need a Hugging Face API token.
You can create one at: [https://huggingface.co/settings/tokens](https://huggingface.co/settings/tokens)
Make sure to grant **write** access if you want to upload models.
"""
)
# Token input
token = st.text_input("Enter your Hugging Face API token", type="password")
# Get token from environment or Secrets if available and not provided
if not token:
if os.environ.get("HF_TOKEN"):
token = os.environ.get("HF_TOKEN")
st.success("Using API token from Secrets.")
elif os.environ.get("HF_API_TOKEN"):
token = os.environ.get("HF_API_TOKEN")
st.success("Using API token from environment variables.")
submitted = st.form_submit_button("Login", use_container_width=True)
if submitted and token:
# Authenticate with Hugging Face
with st.spinner("Authenticating..."):
client = HuggingFaceClient()
success, user_info = client.authenticate(token)
if success:
st.session_state.authenticated = True
st.session_state.api_token = token
st.session_state.username = user_info.get("name", "User")
st.session_state.client = client
st.success(
f"Successfully authenticated as {st.session_state.username}"
)
st.rerun()
else:
st.error(f"Authentication failed: {user_info}")
elif submitted:
st.error("Please enter your Hugging Face API token")
def logout():
"""Log out the current user"""
for key in list(st.session_state.keys()):
del st.session_state[key]
st.session_state.authenticated = False
st.session_state.page = "home"
st.rerun() |