chore: handle rebased, magled code of app in text to image

Author: msluszniak

apps/computer-vision/app/text_to_image/index.tsx

@@ -23,15 +23,14 @@ import { GeneratingContext } from '../../context';
 import ColorPalette from '../../colors';
 import ProgressBar from '../../components/ProgressBar';
 import { Ionicons } from '@expo/vector-icons';
+import { StatsBar } from '../../components/StatsBar';
 
 type TextToImageModelSources = TextToImageProps['model'];
 
 const MODELS: ModelOption<TextToImageModelSources>[] = [
   { label: 'BK-SDM 256', value: BK_SDM_TINY_VPRED_256 },
   { label: 'BK-SDM 512', value: BK_SDM_TINY_VPRED_512 },
 ];
-import { BottomBarWithTextInput } from '../../components/BottomBarWithTextInput';
-import { StatsBar } from '../../components/StatsBar';
 
 export default function TextToImageScreen() {
   const [inferenceStepIdx, setInferenceStepIdx] = useState<number>(0);
@@ -42,8 +41,6 @@ export default function TextToImageScreen() {
     BK_SDM_TINY_VPRED_256
   );
   const [generationTime, setGenerationTime] = useState<number | null>(null);
-  const [showTextInput, setShowTextInput] = useState(false);
-  const [keyboardVisible, setKeyboardVisible] = useState(false);
 
   const imageSize = 224;
   const model = useTextToImage({
@@ -62,13 +59,11 @@ export default function TextToImageScreen() {
     try {
       const start = Date.now();
       const output = await model.generate(input, imageSize, steps);
-      if (output.length) setImage(output);
-      else {
-        setImageTitle(prevImageTitle);
-        return;
+
+      if (output.length) {
+        setImage(output);
+        setGenerationTime(Date.now() - start);
       }
-      setGenerationTime(Date.now() - start);
-      setImage(output);
     } catch (e) {
       console.error(e);
     } finally {
@@ -117,6 +112,7 @@ export default function TextToImageScreen() {
           onSelect={(m) => {
             setSelectedModel(m);
             setImage(null);
+            setGenerationTime(null);
           }}
         />
 
@@ -136,6 +132,9 @@ export default function TextToImageScreen() {
           </TouchableOpacity>
         </View>
 
+        {/* Added StatsBar here, just above the input row */}
+        <StatsBar inferenceTime={generationTime} />
+
         <View style={styles.inputRow}>
           <TextInput
             style={styles.textInput}
@@ -146,20 +145,6 @@ export default function TextToImageScreen() {
             onSubmitEditing={runForward}
             returnKeyType="send"
           />
-        </View>
-        <StatsBar inferenceTime={generationTime} />
-        <View style={styles.bottomContainer}>
-          <BottomBarWithTextInput
-            runModel={runForward}
-            numSteps={steps}
-            setSteps={setSteps}
-            stopModel={model.interrupt}
-            isGenerating={model.isGenerating}
-            isReady={model.isReady}
-            showTextInput={showTextInput}
-            setShowTextInput={setShowTextInput}
-            keyboardVisible={keyboardVisible}
-          />
           {model.isGenerating ? (
             <TouchableOpacity
               style={styles.sendButton}