Update app.py

app.py

@@ -55,6 +55,81 @@ def main():
     `?code=...&device_id=...&state=...`
     """)
 
+    # Шаг 2: Автоматическое извлечение параметров из URL
+    query_params = st.experimental_get_query_params()
+    user_code = query_params.get("code", [None])[0]
+    user_device_id = query_params.get("device_id", [None])[0]
+    user_state = query_params.get("state", [None])[0]
+
+    # Если параметры есть, выполняем обмен кода на токены
+    if user_code and user_state:
+        st.write("Автоматически извлечены параметры из URL:")
+        st.write(f"code: {user_code}")
+        st.write(f"device_id: {user_device_id}")
+        st.write(f"state: {user_state}")
+
+        # Используем code_verifier из параметра state
+        code_verifier = user_state
+
+        url = "https://id.vk.com/oauth2/auth"  # Endpoint для обмена токенов
+        data = {
+            "grant_type": "authorization_code",
+            "client_id": client_id,
+            "client_secret": client_secret,
+            "redirect_uri": redirect_uri,
+            "code": user_code,
+            "code_verifier": code_verifier  # Используем code_verifier из state
+        }
+        if user_device_id:
+            data["device_id"] = user_device_id
+
+        st.write("Отправляем запрос на:", url)
+        st.write("Параметры запроса:", data)
+
+        # Отправка запроса на обмен кода на токены
+        resp = requests.post(url, data=data)
+        st.write("Status code:", resp.status_code)
+        st.write("Response text:", resp.text)
+
+        try:
+            tokens = resp.json()
+            st.write("Ответ (JSON):")
+            st.json(tokens)
+
+            # Сохраняем токены и device_id в session_state
+            st.session_state["VK_ACCESS_TOKEN"] = tokens["access_token"]
+            st.session_state["VK_REFRESH_TOKEN"] = tokens["refresh_token"]
+            st.session_state["VK_DEVICE_ID"] = user_device_id
+            st.success("Токены успешно получены и сохранены!")
+        except Exception as e:
+            st.error(f"Ошибка при декодировании ответа: {e}")
+            st.text(resp.text)
+    else:
+        st.warning("Параметры code, device_id и state не найдены в URL. Пройдите авторизацию.")
+
+# Если скрипт запущен напрямую, вызываем main()
+if __name__ == "__main__":
+    main()
+    # Шаг 1: Авторизация
+    st.markdown("### 1) Авторизация")
+    auth_params = {
+        "client_id": client_id,
+        "redirect_uri": redirect_uri,
+        "response_type": "code",
+        "scope": scope,
+        "state": st.session_state.code_verifier,  # Передаем code_verifier в параметре state
+        "code_challenge": code_challenge,
+        "code_challenge_method": "S256",
+        "response_mode": "query"
+    }
+    auth_url = "https://id.vk.com/authorize?" + urllib.parse.urlencode(auth_params, safe="-._~")
+    st.markdown(f"[Перейти к авторизации]({auth_url})")
+
+    st.markdown("""
+    После авторизации VK вернёт вас на redirect_uri с параметрами:  
+    `?code=...&device_id=...&state=...`
+    """)
+
     # Шаг 2: Обмен кода на токены
     st.markdown("### 2) Обмен кода на токены")
     user_code = st.text_input("Введите полный code, полученный из URL", value="")