Upload main_app.py

main_app.py

@@ -31,9 +31,24 @@ os.makedirs(USER_DATA_DIR, exist_ok=True)
 # --- 2. ユーザー ID 判定（毎回再認証・再発行） ---
 
 def get_current_user_id():
-    hf_token = st.session_state.get("hf_token", None)
-    # HFトークンが存在する場合は毎回whoamiで認証・ID取得
-    if hf_token:
+    # OAuth認証情報を環境変数から取得
+    OAUTH_CLIENT_ID = os.getenv("OAUTH_CLIENT_ID")
+    OAUTH_CLIENT_SECRET = os.getenv("OAUTH_CLIENT_SECRET")
+    OAUTH_SCOPES = os.getenv("OAUTH_SCOPES", "openid profile email")
+    OPENID_PROVIDER_URL = os.getenv("OPENID_PROVIDER_URL", "https://huggingface.co")
+
+    # Streamlitセッションにトークンがなければ認証フロー
+    if "hf_token" not in st.session_state:
+        st.markdown("## HuggingFace OAuthログイン")
+        auth_url = f"{OPENID_PROVIDER_URL}/oauth/authorize?client_id={OAUTH_CLIENT_ID}&response_type=token&scope={OAUTH_SCOPES}"
+        if st.button("HuggingFaceでログイン"):
+            components.html(f"<script>window.location.href='{auth_url}';</script>", height=0)
+        st.info("ログイン後、アクセストークンが自動取得されます。")
+        user_id = str(uuid.uuid4())
+        is_hf_user = False
+        return user_id, is_hf_user
+    else:
+        hf_token = st.session_state["hf_token"]
         try:
             api = HfApi()
             user_info = whoami(token=hf_token)
@@ -43,12 +58,7 @@ def get_current_user_id():
             st.warning(f"HF API エラー: {e}. 一時 UUID を使用します。")
             user_id = str(uuid.uuid4())
             is_hf_user = False
-    else:
-        # トークンなしは毎回新規UUID
-        user_id = str(uuid.uuid4())
-        is_hf_user = False
-    # セッション状態に依存せず、毎回最新の認証結果を返す
-    return user_id, is_hf_user
+        return user_id, is_hf_user
 
 
 user_id, is_hf_user = get_current_user_id()