3 files changed, 28 insertions, 5 deletions

diff --git a/subprojects/frontend/src/TopBar.tsx b/subprojects/frontend/src/TopBar.tsx
index 4424e6b3..a060ff9e 100644
--- a/subprojects/frontend/src/TopBar.tsx
+++ b/subprojects/frontend/src/TopBar.tsx
@@ -1,10 +1,15 @@
 import AppBar from '@mui/material/AppBar';
 import Toolbar from '@mui/material/Toolbar';
 import Typography from '@mui/material/Typography';
+import { useTheme } from '@mui/material/styles';
+import useMediaQuery from '@mui/material/useMediaQuery';
 import { throttle } from 'lodash-es';
+import { observer } from 'mobx-react-lite';
 import React, { useEffect, useMemo, useState } from 'react';
 
+import { useRootStore } from './RootStore';
 import ToggleDarkModeButton from './ToggleDarkModeButton';
+import GenerateButton from './editor/GenerateButton';
 
 function useWindowControlsOverlayVisible(): boolean {
   const [windowControlsOverlayVisible, setWindowControlsOverlayVisible] =
@@ -39,8 +44,11 @@ function useWindowControlsOverlayVisible(): boolean {
   return windowControlsOverlayVisible;
 }
 
-export default function TopBar(): JSX.Element {
+export default observer(function TopBar(): JSX.Element {
+  const { editorStore } = useRootStore();
   const overlayVisible = useWindowControlsOverlayVisible();
+  const { breakpoints } = useTheme();
+  const showGenerateButton = useMediaQuery(breakpoints.down('sm'));
 
   return (
     <AppBar
@@ -74,8 +82,11 @@ export default function TopBar(): JSX.Element {
         <Typography variant="h6" component="h1" flexGrow={1}>
           Refinery
         </Typography>
+        {showGenerateButton && (
+          <GenerateButton editorStore={editorStore} hideWarnings />
+        )}
         <ToggleDarkModeButton />
       </Toolbar>
     </AppBar>
   );
-}
+});
diff --git a/subprojects/frontend/src/editor/EditorPane.tsx b/subprojects/frontend/src/editor/EditorPane.tsx
index 079ebcdc..e433e714 100644
--- a/subprojects/frontend/src/editor/EditorPane.tsx
+++ b/subprojects/frontend/src/editor/EditorPane.tsx
@@ -2,6 +2,8 @@ import Box from '@mui/material/Box';
 import Skeleton from '@mui/material/Skeleton';
 import Stack from '@mui/material/Stack';
 import Toolbar from '@mui/material/Toolbar';
+import { useTheme } from '@mui/material/styles';
+import useMediaQuery from '@mui/material/useMediaQuery';
 import { observer } from 'mobx-react-lite';
 import React, { useState } from 'react';
 
@@ -32,12 +34,14 @@ function EditorLoading(): JSX.Element {
 
 export default observer(function EditorPane(): JSX.Element {
   const { editorStore } = useRootStore();
+  const { breakpoints } = useTheme();
+  const showGenerateButton = useMediaQuery(breakpoints.up('sm'));
 
   return (
     <Stack direction="column" flexGrow={1} flexShrink={1} overflow="auto">
       <Toolbar variant="dense">
         <EditorButtons editorStore={editorStore} />
-        <GenerateButton editorStore={editorStore} />
+        {showGenerateButton && <GenerateButton editorStore={editorStore} />}
       </Toolbar>
       <Box flexGrow={1} flexShrink={1} overflow="auto">
         {editorStore === undefined ? (
diff --git a/subprojects/frontend/src/editor/GenerateButton.tsx b/subprojects/frontend/src/editor/GenerateButton.tsx
index 2ffb1a94..485989b5 100644
--- a/subprojects/frontend/src/editor/GenerateButton.tsx
+++ b/subprojects/frontend/src/editor/GenerateButton.tsx
@@ -7,10 +7,12 @@ import type EditorStore from './EditorStore';
 
 const GENERATE_LABEL = 'Generate';
 
-export default observer(function GenerateButton({
+const GenerateButton = observer(function GenerateButton({
   editorStore,
+  hideWarnings,
 }: {
   editorStore: EditorStore | undefined;
+  hideWarnings?: boolean | undefined;
 }): JSX.Element {
   if (editorStore === undefined) {
     return (
@@ -26,7 +28,7 @@ export default observer(function GenerateButton({
   if (errorCount > 0) {
     diagnostics.push(`${errorCount} error${errorCount === 1 ? '' : 's'}`);
   }
-  if (warningCount > 0) {
+  if (!(hideWarnings ?? false) && warningCount > 0) {
     diagnostics.push(`${warningCount} warning${warningCount === 1 ? '' : 's'}`);
   }
   const summary = diagnostics.join(' and ');
@@ -55,3 +57,9 @@ export default observer(function GenerateButton({
     </Button>
   );
 });
+
+GenerateButton.defaultProps = {
+  hideWarnings: false,
+};
+
+export default GenerateButton;