wip

lama_cleaner/app/src/components/Header/Header.tsx

@@ -1,17 +1,19 @@
 import { ArrowLeftIcon, UploadIcon } from '@heroicons/react/outline'
 import React, { useState } from 'react'
-import { useRecoilState } from 'recoil'
-import { fileState } from '../../store/Atoms'
+import { useRecoilState, useRecoilValue } from 'recoil'
+import { fileState, isSDState } from '../../store/Atoms'
 import Button from '../shared/Button'
 import Shortcuts from '../Shortcuts/Shortcuts'
 import useResolution from '../../hooks/useResolution'
 import { ThemeChanger } from './ThemeChanger'
 import SettingIcon from '../Settings/SettingIcon'
+import PromptInput from './PromptInput'
 
 const Header = () => {
   const [file, setFile] = useRecoilState(fileState)
   const resolution = useResolution()
   const [uploadElemId] = useState(`file-upload-${Math.random().toString()}`)
+  const isSD = useRecoilValue(isSDState)
 
   const renderHeader = () => {
     return (
@@ -37,6 +39,8 @@ const Header = () => {
           </label>
         </div>
 
+        {isSD && file ? <PromptInput /> : <></>}
+
         <div className="header-icons-wrapper">
           <ThemeChanger />
           {file && (