diff --git a/src/hooks/useTaskPolling.ts b/src/hooks/useTaskPolling.ts
index 91e9b694..4af45eb2 100644
--- a/src/hooks/useTaskPolling.ts
+++ b/src/hooks/useTaskPolling.ts
@@ -14,6 +14,7 @@ interface TaskPollingState {
   isComplete: boolean;
   taskId: string | null;
   error: string | null;
+  result: Record<string, any> | null;
 }
 
 interface TaskPollingControls {
@@ -32,6 +33,7 @@ const initialState: TaskPollingState = {
   isComplete: false,
   taskId: null,
   error: null,
+  result: null,
 };
 
 export function useTaskPolling(
@@ -56,7 +58,7 @@ export function useTaskPolling(
         if (cancelled) return;
 
         if (result.status === 'completed') {
-          setState(prev => ({ ...prev, isGenerating: false, isComplete: true, hasError: false, error: null }));
+          setState(prev => ({ ...prev, isGenerating: false, isComplete: true, hasError: false, error: null, result: result.result || null }));
         } else if (result.status === 'failed') {
           setState(prev => ({
             ...prev,