From 05f2fc33f93845eaae0e5837a2605ecf07030b6f Mon Sep 17 00:00:00 2001
From: Nastya <nastyab015@yandex.ru>
Date: Sun, 20 Jul 2025 15:25:48 +0300
Subject: [PATCH] =?UTF-8?q?=D0=BA=D0=BE=D1=80=D1=80=D0=B5=D0=BA=D1=82?=
 =?UTF-8?q?=D0=BD=D1=8B=D0=B9=20=D0=BF=D0=B5=D1=80=D0=B5=D0=B2=D0=BE=D0=B4?=
 =?UTF-8?q?=20=D0=BD=D0=B0=20=D0=90=D0=98=20=D0=BD=D0=B0=D1=81=D1=82=D1=80?=
 =?UTF-8?q?=D0=BE=D0=B9=D0=BA=D1=83=20=D0=BA=D0=B2=D0=B8=D0=B7=D0=B0=20?=
 =?UTF-8?q?=D0=BF=D0=BE=D1=81=D0=BB=D0=B5=20=D0=BE=D0=BF=D0=BB=D0=B0=D1=82?=
 =?UTF-8?q?=D1=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../PersonalizationAI/PersonalizationAI.tsx   |  2 +-
 src/utils/hooks/useAutoPay.ts                 | 32 +++++++++++++++++--
 2 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/src/pages/PersonalizationAI/PersonalizationAI.tsx b/src/pages/PersonalizationAI/PersonalizationAI.tsx
index 84ca429f..1476a6b3 100644
--- a/src/pages/PersonalizationAI/PersonalizationAI.tsx
+++ b/src/pages/PersonalizationAI/PersonalizationAI.tsx
@@ -242,7 +242,7 @@ export default function PersonalizationAI() {
       if (payError?.includes("insufficient funds") || payError?.includes("Payment Required")) {
         var link = document.createElement("a");
         link.href = generateHubWalletRequestURL({
-          wayback: "edit",
+          wayback: "personalization-ai",
           action: "buy",
           dif: "50000",
           userid: userId,
diff --git a/src/utils/hooks/useAutoPay.ts b/src/utils/hooks/useAutoPay.ts
index 7c1d9956..116fb66e 100644
--- a/src/utils/hooks/useAutoPay.ts
+++ b/src/utils/hooks/useAutoPay.ts
@@ -28,6 +28,14 @@ export const useAfterPay = () => {
   let URLadditionalinformation = searchParams.get("additionalinformation");//его токен
 
   useEffect(() => {
+    console.log("useAutoPay: Processing return from payment", {
+      URLaction,
+      URLuserId,
+      URLadditionalinformation,
+      userId,
+      wayback: searchParams.get("wayback")
+    });
+    
     setSearchParams({}, { replace: true });
 
     if (userId && URLuserId && userId === URLuserId) {
@@ -53,12 +61,32 @@ export const useAfterPay = () => {
 
 
       //Покупка ИИ тарифа из настройки квиза ИИ вкладки
-      if (location.pathname === "/personalization-ai") {
+      if (URLaction === "buy" && URLadditionalinformation) {
         const quizId = Number(URLadditionalinformation);
+        console.log("useAutoPay: Processing AI tariff purchase", {
+          quizId,
+          wayback: searchParams.get("wayback")
+        });
+        
         if (quizId) {
           setEditQuizId(Number(quizId)); //Выбираем квиз
           setCurrentStep(17); // Шаг для персонализации AI
-          navigate("/personalization-ai"); //Непосредственно переход
+          
+          // Проверяем wayback параметр для определения куда переходить
+          const wayback = searchParams.get("wayback");
+          if (wayback === "edit") {
+            console.log("useAutoPay: Navigating to /edit first, then /personalization-ai");
+            // Сначала переходим на /edit, затем на /personalization-ai
+            navigate("/edit");
+            // Используем setTimeout чтобы дать время для загрузки /edit
+            setTimeout(() => {
+              navigate("/personalization-ai");
+            }, 100);
+          } else {
+            console.log("useAutoPay: Direct navigation to /personalization-ai");
+            // Прямой переход на /personalization-ai
+            navigate("/personalization-ai");
+          }
         }
       }