fix: prevent race condition in code generation state

- Set isGenerating flag immediately before triggering generation to avoid UI state inconsistencies - Added flag update for both new chat and reconnection scenarios

Author: AshishKumar4

src/routes/chat/hooks/use-chat.ts

@@ -294,6 +294,7 @@ export function useChat({
 // Request file generation for new chats only
 if (!disableGenerate && urlChatId === 'new') {
 logger.debug('🔄 Starting code generation for new chat');
+setIsGenerating(true);
 sendWebSocketMessage(ws, 'generate_all');
 }
 });

src/routes/chat/utils/handle-websocket-message.ts

@@ -254,6 +254,7 @@ export function createWebSocketMessageHandler(deps: HandleMessageDeps) {
 
  if (state.shouldBeGenerating && !isGenerating) {
  logger.debug('🔄 Reconnected with shouldBeGenerating=true, auto-resuming generation');
+ setIsGenerating(true); 
  updateStage('code', { status: 'active' });
  sendWebSocketMessage(websocket, 'generate_all');
  }