Upload 2 files

main_app.py

@@ -141,37 +141,35 @@ if 'server_started' not in st.session_state:
         logger.warning("⚠️ セッション管理サーバー起動失敗 - フォールバックモードで動作")
 
 
-SPACE_URL = "https://huggingface.co/spaces/sirochild/mari-chat-3"
-LOGIN_URL = f"{SPACE_URL}/oauth/huggingface/login"
-ME_URL = f"{SPACE_URL}/me"
+API_URL = "http://localhost:8000"
+LOGIN_URL = f"{API_URL}/oauth/huggingface/login"
+ME_URL = f"{API_URL}/me"
 def get_current_user_id():
     # --- セッションステートにユーザー情報を保存 ---
     if "user" not in st.session_state:
         st.session_state.user = None
 
-    # --- ログインしてなければボタンを表示 ---
-    if st.session_state.user is None:
-        st.markdown(f"[👉 Hugging Faceでログイン]({LOGIN_URL})")
-
-    else:
-        st.success(f"ログイン中: {st.session_state.user['preferred_username']}")
-        st.json(st.session_state.user)
-
-    # --- クエリパラメータに code があれば認証処理 ---
-    query_params = st.query_params
-    if "code" in query_params:
-        code = query_params["code"]
-        # /me にアクセスしてユーザー情報を取得
+    # セッション開始時に /me を自動でチェック
+    if "user_info" not in st.session_state:
         try:
-            resp = requests.get(ME_URL, cookies={"hf_oauth_code": code})
-            if resp.status_code == 200:
-                st.session_state.user = resp.json()
-                st.experimental_rerun()
+            res = requests.get(ME_URL, cookies=dict(st.session_state.get("cookies", {})))
+            data = res.json()
+            if "username" in data:
+                st.session_state["user_info"] = data
+                st.success(f"ログインしました: {data['username']}")
             else:
-                st.error("認証失敗しました")
+                st.warning("まだログインしていません。")
         except Exception as e:
             st.error(f"エラー: {e}")
 
+    # ログイン済みなら表示
+    if "user_info" in st.session_state:
+        st.write(st.session_state["user_info"])
+
+    # --- ログインしてなければボタンを表示 ---
+    if st.session_state.user is None:
+        st.markdown(f"[👉 Hugging Faceでログイン]({LOGIN_URL})")
+
 # --- 必要なモジュールのインポート ---
 
 # << 麻理チャット用モジュール >>

session_api_server.py

@@ -50,6 +50,18 @@ def me(request: Request):
         "email": oauth_info.user_info.email,
     }
 
+# ログイン後に /me へ自動遷移させる
+@app.get("/oauth/callback")
+async def callback(request: Request):
+    oauth_info = parse_huggingface_oauth(request)
+    if oauth_info is None:
+        return RedirectResponse(url="/?error=login_failed")
+
+    # セッション cookie を発行して Streamlit から /me で取れるようにする
+    response = RedirectResponse(url="http://localhost:8501/?logged_in=1")
+    response.set_cookie(key="session", value=oauth_info.json(), httponly=True)
+    return response
+
 
 class SessionManager:
     """セッション管理クラス"""