diff --git a/src/components/Chat/ChatContent/ChatContent.tsx b/src/components/Chat/ChatContent/ChatContent.tsx
index 53ce7b7..aafd070 100644
--- a/src/components/Chat/ChatContent/ChatContent.tsx
+++ b/src/components/Chat/ChatContent/ChatContent.tsx
@@ -91,7 +91,9 @@ const ChatContent = () => {
             </div>
           )}
           <div className='mt-4'>
-            <DownloadChat saveRef={saveRef} />
+            {useStore.getState().generating || (
+              <DownloadChat saveRef={saveRef} />
+            )}
           </div>
           <div className='w-full h-36'></div>
         </div>