rtkquery codegen

invoke-ai · Feb 5, 2023 · d9937e1 · d9937e1
1 parent 312d53a
commit d9937e1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/invokeai/frontend/src/app/App.tsx b/invokeai/frontend/src/app/App.tsx
@@ -80,6 +80,7 @@ const App = () => {
       setInvocationProgress(undefined);
       setSessionId(undefined);
       console.log('invocation_complete', data);
+      socket.emit('unsubscribe', { session: sessionId });
     });
     socket.on('invocation_started', (data) =>
       console.log('invocation_started', data)
@@ -100,7 +101,9 @@ const App = () => {
       <ImageUploader>
         <ProgressBar />
         <Flex gap={2} p={2} alignItems="center">
-          <Button onClick={handleCreateSession}>Create Session</Button>
+          <Button onClick={handleCreateSession} isDisabled={!!sessionId}>
+            Create Session
+          </Button>
           <Button
             onClick={handleInvokeSession}
             isDisabled={!sessionId || isInvocationPending}