do not go to flexgen if post failed (#1375)

* do not go to flexgen if post failed

* what

Author: adhami3310

pcweb/pages/landing/views/hero.py

@@ -37,18 +37,22 @@ async def redirect_to_ai_builder(self, form_data: dict):
             random_uuid = uuid.uuid4()
 
             async with httpx.AsyncClient() as client:
-                await client.post(
+                response = await client.post(
                     RX_BUILD_BACKEND.rstrip("/") + "/prompt",
                     json={
-                        "prompt": prompt_map[prompt],
+                        "prompt": prompt_map.get(prompt, prompt),
                         "token": str(random_uuid),
                         "images": self.image_data_uris,
                     },
                 )
             async with self:
                 self.reset()
-            return rx.redirect(
-                REFLEX_BUILD_URL.strip("/") + f"/prompt?token={random_uuid!s}"
+            return (
+                rx.redirect("/")
+                if not response.is_success
+                else rx.redirect(
+                    REFLEX_BUILD_URL.strip("/") + f"/prompt?token={random_uuid!s}"
+                )
             )
 
     @rx.event