Update app.py

app.py

@@ -13,10 +13,9 @@ def generate_code_challenge(code_verifier):
     hashed = hashlib.sha256(code_verifier.encode('utf-8')).digest()
     return base64.urlsafe_b64encode(hashed).decode('utf-8').rstrip('=')
 
-# Инициализация session_state для хранения code_verifier и code_challenge
+# Инициализация session_state для хранения code_verifier
 if 'code_verifier' not in st.session_state:
     st.session_state.code_verifier = generate_code_verifier()
-    st.session_state.code_challenge = generate_code_challenge(st.session_state.code_verifier)
 
 # Заголовок приложения
 st.title("VK ID Authorization with PKCE")
@@ -27,9 +26,12 @@ client_secret = "piaMXHEDj0afa4QObJCD"  # Ваш client_secret
 redirect_uri = "https://huggingface.co/spaces/PRAnalytics/prtooltest"  # Ваш redirect_uri
 scope = "wall"  # Запрошенные разрешения
 
+# Генерация code_challenge на основе сохраненного code_verifier
+code_challenge = generate_code_challenge(st.session_state.code_verifier)
+
 # Отображение code_verifier и code_challenge (для отладки)
 st.write("code_verifier:", st.session_state.code_verifier)
-st.write("code_challenge:", st.session_state.code_challenge)
+st.write("code_challenge:", code_challenge)
 
 # Шаг 1: Авторизация
 st.markdown("### 1) Авторизация")
@@ -38,8 +40,8 @@ auth_params = {
     "redirect_uri": redirect_uri,
     "response_type": "code",
     "scope": scope,
-    "state": "my_state",
-    "code_challenge": st.session_state.code_challenge,  # Используем сохраненный code_challenge
+    "state": st.session_state.code_verifier,  # Передаем code_verifier в параметре state
+    "code_challenge": code_challenge,
     "code_challenge_method": "S256",
     "response_mode": "query"
 }
@@ -55,9 +57,13 @@ st.markdown("""
 st.markdown("### 2) Обмен кода на токены")
 user_code = st.text_input("Введите полный code, полученный из URL", value="")
 user_device_id = st.text_input("Введите полный device_id, полученный из URL", value="")
+user_state = st.text_input("Введите state, полученный из URL", value="")  # Поле для ввода state
 
-if user_code:
+if user_code and user_state:
     if st.button("Обменять code"):
+        # Используем code_verifier из параметра state
+        code_verifier = user_state
+
         url = "https://id.vk.com/oauth2/auth"  # Endpoint для обмена токенов
         data = {
             "grant_type": "authorization_code",
@@ -65,7 +71,7 @@ if user_code:
             "client_secret": client_secret,
             "redirect_uri": redirect_uri,
             "code": user_code,
-            "code_verifier": st.session_state.code_verifier  # Используем сохраненный code_verifier
+            "code_verifier": code_verifier  # Используем code_verifier из state
         }
         if user_device_id:
             data["device_id"] = user_device_id