Upload main_app.py

main_app.py

@@ -25,7 +25,7 @@ HF_ENDPOINT = "https://huggingface.co"
 CLIENT_ID = os.environ.get("OAUTH_CLIENT_ID")
 CLIENT_SECRET = os.environ.get("OAUTH_CLIENT_SECRET")
 SPACE_ID = os.environ.get("SPACE_ID")
-
+SPACE_HOST = os.environ.get("SPACE_HOST")
 
 st.set_page_config(
         page_title="麻理プロジェクト", page_icon="🤖",
@@ -38,14 +38,17 @@ USER_DATA_DIR = "/mnt/data/mari_users"
 os.makedirs(USER_DATA_DIR, exist_ok=True)
 
 
+# vvv 修正点: この関数を全面的に書き換え
 def get_redirect_uri():
-    """リダイレクトURIを動的に生成する"""
-    if SPACE_ID:
+    """
+    SpaceのURL形式（直接形式 or パス形式）に応じて正しいリダイレクトURIを生成する
+    """
+    # SPACE_HOSTが設定されている場合（直接形式URL）はそれを使うのが最も確実
+    if SPACE_HOST:
+        return f"https://{SPACE_HOST}"
+    # SPACE_HOSTがない場合（古い形式のSpaceなど）はSPACE_IDから組み立てる
+    elif SPACE_ID:
         return f"https://huggingface.co/spaces/{SPACE_ID}"
-    else:
-        # ローカル開発用のフォールバック（必要に応じて変更）
-        # st.query_paramsはローカルではうまく機能しない場合がある
-        return "http://localhost:8501"
 
 def get_hf_token(code: str, redirect_uri: str) -> dict | None:
     """認可コード(code)をアクセストークンに交換する"""