feat: enhance

Author: Tommy Tong

src/cmd-intensive-chat-ui.tsx

@@ -59,7 +59,14 @@ const updateHeartbeat = async () => {
   if (!options.outputDir) return;
 
   const heartbeatPath = path.join(options.outputDir, 'heartbeat.txt');
-  await fs.writeFile(heartbeatPath, Date.now().toString(), 'utf8');
+  try {
+      const dir = path.dirname(heartbeatPath);
+      await fs.mkdir(dir, { recursive: true }); // Ensure directory exists
+      await fs.writeFile(heartbeatPath, Date.now().toString(), 'utf8');
+  } catch (writeError) {
+      // Log the specific error but allow the poll cycle to continue
+      console.error(`Failed to write heartbeat file ${heartbeatPath}:`, writeError);
+  }
 };
 
 // Register process termination handlers

src/cmd-intensive-chat.ts

@@ -301,8 +301,16 @@ export async function isSessionActive(sessionId: string): Promise<boolean> {
     }
 
     return true;
-  } catch (e) {
-    // Error accessing heartbeat file, session is likely dead
+  } catch (e: any) { // Added ': any' to access e.code safely
+    // If error is ENOENT (file not found), assume session is still starting
+    if (e.code === 'ENOENT') {
+      // Optional: Could add a check here to see if the session is very new
+      // e.g., if (Date.now() - session.startTime < 2000) return true;
+      // For now, let's assume ENOENT means it's possibly still starting.
+      return true;
+    }
+    // For any other error, session is likely dead
+    console.error(`Error checking heartbeat for session ${sessionId}:`, e); // Log other errors
     session.isActive = false;
     return false;
   }

src/components/InteractiveInput.tsx

@@ -99,7 +99,7 @@ export const InteractiveInput: FC<InteractiveInputProps> = ({ question, question
 
       {predefinedOptions && predefinedOptions.length > 0 && (
         <Box flexDirection="column" marginBottom={1}>
-          <Text>Use ↑/↓ to select options, type any key for custom input, Enter to submit</Text>
+          <Text dimColor={true}>Use ↑/↓ to select options, Enter to submit</Text>
           {predefinedOptions.map((opt, i) => (
             <Text key={i} color={i === selectedIndex ? (mode === "option" ? 'greenBright' : 'green') : undefined}>
               {i === selectedIndex ? (mode === "option" ? '› ' : '  ') : '  '}{opt}