chore: add suggestions from code review

Author: msluszniak

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

@@ -30,6 +30,7 @@ export default function ClassificationScreen() {
     if (typeof uri === 'string') {
       setImageUri(uri as string);
       setResults([]);
+      setInferenceTime(null);
     }
   };
 

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

@@ -48,6 +48,7 @@ export default function ObjectDetectionScreen() {
       setImageUri(image.uri as string);
       setImageDimensions({ width: width as number, height: height as number });
       setResults([]);
+      setInferenceTime(null);
     }
   };
 

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

@@ -60,6 +60,7 @@ export default function OCRScreen() {
     if (typeof uri === 'string') {
       setImageUri(uri as string);
       setResults([]);
+      setInferenceTime(null);
     }
   };
 

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

@@ -35,6 +35,7 @@ export default function VerticalOCRScree() {
     if (typeof uri === 'string') {
       setImageUri(uri as string);
       setResults([]);
+      setInferenceTime(null);
     }
   };
 

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

@@ -88,6 +88,7 @@ export default function SemanticSegmentationScreen() {
     setImageUri(image.uri);
     setImageSize({ width: image.width ?? 0, height: image.height ?? 0 });
     setSegImage(null);
+    setInferenceTime(null);
   };
 
   const runForward = async () => {

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

@@ -51,6 +51,7 @@ export default function StyleTransferScreen() {
     if (typeof uri === 'string') {
       setImageUri(uri);
       setStyledUri('');
+      setInferenceTime(null);
     }
   };
 

apps/llm/app/llm/index.tsx

@@ -46,7 +46,12 @@ function LLMScreen() {
   const { setGlobalGenerating } = useContext(GeneratingContext);
 
   const llm = useLLM({ model: selectedModel });
-  const { stats, onMessageSend } = useLLMStats(llm.response, llm.isGenerating);
+  const tokenCount = llm.isReady ? llm.getGeneratedTokenCount() : 0;
+  const { stats, onMessageSend } = useLLMStats(
+    llm.response,
+    llm.isGenerating,
+    tokenCount
+  );
 
   useEffect(() => {
     if (llm.error) {

apps/llm/app/llm_structured_output/index.tsx

@@ -88,7 +88,12 @@ function LLMScreen() {
   const { setGlobalGenerating } = useContext(GeneratingContext);
 
   const llm = useLLM({ model: selectedModel });
-  const { stats, onMessageSend } = useLLMStats(llm.response, llm.isGenerating);
+  const tokenCount = llm.isReady ? llm.getGeneratedTokenCount() : 0;
+  const { stats, onMessageSend } = useLLMStats(
+    llm.response,
+    llm.isGenerating,
+    tokenCount
+  );
 
   useEffect(() => {
     setGlobalGenerating(llm.isGenerating);

apps/llm/app/llm_tool_calling/index.tsx

@@ -60,7 +60,12 @@ function LLMToolCallingScreen() {
   const { setGlobalGenerating } = useContext(GeneratingContext);
 
   const llm = useLLM({ model: selectedModel });
-  const { stats, onMessageSend } = useLLMStats(llm.response, llm.isGenerating);
+  const tokenCount = llm.isReady ? llm.getGeneratedTokenCount() : 0;
+  const { stats, onMessageSend } = useLLMStats(
+    llm.response,
+    llm.isGenerating,
+    tokenCount
+  );
 
   useEffect(() => {
     setGlobalGenerating(llm.isGenerating);

apps/llm/app/multimodal_llm/index.tsx

@@ -46,7 +46,12 @@ function MultimodalLLMScreen() {
   const vlm = useLLM({
     model: LFM2_VL_1_6B_QUANTIZED,
   });
-  const { stats, onMessageSend } = useLLMStats(vlm.response, vlm.isGenerating);
+  const tokenCount = vlm.isReady ? vlm.getGeneratedTokenCount() : 0;
+  const { stats, onMessageSend } = useLLMStats(
+    vlm.response,
+    vlm.isGenerating,
+    tokenCount
+  );
 
   useEffect(() => {
     setGlobalGenerating(vlm.isGenerating);