From cc64e6a035ff6204319537d77bb67e64f3610a3b Mon Sep 17 00:00:00 2001
From: Adw8 <adwaitgharpure@gmail.com>
Date: Thu, 30 Jan 2025 12:30:36 +0530
Subject: [PATCH] Display message in UI before reloading

---
 src/components/AIServiceCard.tsx | 7 +++++--
 src/services/fluxService.ts      | 2 --
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/components/AIServiceCard.tsx b/src/components/AIServiceCard.tsx
index 2f73b79..52ddfef 100644
--- a/src/components/AIServiceCard.tsx
+++ b/src/components/AIServiceCard.tsx
@@ -60,7 +60,10 @@ const AIServiceCard: React.FC<AIServiceCardProps> = ({
           loading: false,
           error: result.error,
         })
-        return
+        // Reload the page to get latest prices
+        setTimeout(() => {
+          window.location.reload();
+        }, 3000);
       }
 
       if (result.imageUrl) {
@@ -121,7 +124,7 @@ const AIServiceCard: React.FC<AIServiceCardProps> = ({
 
         {generationState.error && (
           <div className="mt-4 bg-red-900/20 border border-red-500/20 text-red-400 px-4 py-3 rounded-xl text-center">
-            {generationState.error}
+            {generationState.error}, reloading...
           </div>
         )}
 
diff --git a/src/services/fluxService.ts b/src/services/fluxService.ts
index 77ec20c..a947fb0 100644
--- a/src/services/fluxService.ts
+++ b/src/services/fluxService.ts
@@ -65,8 +65,6 @@ export async function generateWithFlux(
     }
   } catch (error) {
     console.error('Flux generation error:', error)
-    // Reload the page to get latest prices
-    window.location.reload();
     return {
       error: error instanceof Error ? error.message : 'Generation failed'
     }