fix chat boxes and remove context indicator

apps/x/apps/renderer/src/App.tsx

@@ -2,11 +2,12 @@ import * as React from 'react'
 import { useCallback, useEffect, useState, useRef } from 'react'
 import { workspace } from '@x/shared';
 import { RunEvent } from '@x/shared/src/runs.js';
-import type { ChatStatus, LanguageModelUsage, ToolUIPart } from 'ai';
+import type { LanguageModelUsage, ToolUIPart } from 'ai';
 import './App.css'
 import z from 'zod';
 import { Button } from './components/ui/button';
-import { CheckIcon, LoaderIcon } from 'lucide-react';
+import { CheckIcon, LoaderIcon, ArrowUp } from 'lucide-react';
+import { cn } from '@/lib/utils';
 import { MarkdownEditor } from './components/markdown-editor';
 import { ChatInputBar } from './components/chat-button';
 import { ChatSidebar } from './components/chat-sidebar';
@@ -27,31 +28,15 @@ import {
   MessageResponse,
 } from '@/components/ai-elements/message';
 import {
-  PromptInput,
-  PromptInputBody,
-  PromptInputFooter,
   type PromptInputMessage,
   PromptInputProvider,
-  PromptInputSubmit,
   PromptInputTextarea,
-  PromptInputTools,
   usePromptInputController,
   type FileMention,
 } from '@/components/ai-elements/prompt-input';
 import { Reasoning, ReasoningContent, ReasoningTrigger } from '@/components/ai-elements/reasoning';
 import { Shimmer } from '@/components/ai-elements/shimmer';
 import { Tool, ToolContent, ToolHeader, ToolInput, ToolOutput } from '@/components/ai-elements/tool';
-import {
-  Context,
-  ContextCacheUsage,
-  ContextContent,
-  ContextContentBody,
-  ContextContentHeader,
-  ContextInputUsage,
-  ContextOutputUsage,
-  ContextReasoningUsage,
-  ContextTrigger,
-} from '@/components/ai-elements/context';
 import {
   SidebarInset,
   SidebarProvider,
@@ -259,61 +244,52 @@ const collectFilePaths = (nodes: TreeNode[]): string[] =>
 interface ChatInputInnerProps {
   onSubmit: (message: PromptInputMessage, mentions?: FileMention[]) => void
   isProcessing: boolean
-  contextUsage: LanguageModelUsage
-  maxTokens: number
-  usedTokens: number
 }
 
 function ChatInputInner({
   onSubmit,
   isProcessing,
-  contextUsage,
-  maxTokens,
-  usedTokens,
 }: ChatInputInnerProps) {
   const controller = usePromptInputController()
   const message = controller.textInput.value
   const canSubmit = Boolean(message.trim()) && !isProcessing
-  const submitStatus: ChatStatus = isProcessing ? 'streaming' : 'ready'
 
-  const handleSubmit = useCallback((msg: PromptInputMessage) => {
+  const handleSubmit = useCallback(() => {
-    onSubmit(msg, controller.mentions.mentions)
+    if (!canSubmit) return
+    onSubmit({ text: message.trim(), files: [] }, controller.mentions.mentions)
+    controller.textInput.clear()
     controller.mentions.clearMentions()
-  }, [onSubmit, controller.mentions])
+  }, [canSubmit, message, onSubmit, controller])
+
+  const handleKeyDown = useCallback((e: React.KeyboardEvent) => {
+    if (e.key === 'Enter' && !e.shiftKey) {
+      e.preventDefault()
+      handleSubmit()
+    }
+  }, [handleSubmit])
 
   return (
-    <PromptInput onSubmit={handleSubmit}>
+    <div className="flex items-center gap-2 bg-background border border-border rounded-3xl shadow-xl px-4 py-2.5">
-      <PromptInputBody>
+      <PromptInputTextarea
-        <PromptInputTextarea
+        placeholder="Type your message..."
-          placeholder="Type your message..."
+        disabled={isProcessing}
-          disabled={isProcessing}
+        onKeyDown={handleKeyDown}
-        />
+        className="min-h-[1.5rem] py-0 border-0 shadow-none focus-visible:ring-0 rounded-none"
-      </PromptInputBody>
+      />
-      <PromptInputFooter>
+      <Button
-        <PromptInputTools>
+        size="icon"
-          <Context
+        onClick={handleSubmit}
-            maxTokens={maxTokens}
+        disabled={!canSubmit}
-            usedTokens={usedTokens}
+        className={cn(
-            usage={contextUsage}
+          "h-7 w-7 rounded-full shrink-0 transition-all",
-          >
+          canSubmit
-            <ContextTrigger size="sm" />
+            ? "bg-primary text-primary-foreground hover:bg-primary/90"
-            <ContextContent>
+            : "bg-muted text-muted-foreground"
-              <ContextContentHeader />
+        )}
-              <ContextContentBody>
+      >
-                <ContextInputUsage />
+        <ArrowUp className="h-4 w-4" />
-                <ContextOutputUsage />
+      </Button>
-                <ContextReasoningUsage />
+    </div>
-                <ContextCacheUsage />
-              </ContextContentBody>
-            </ContextContent>
-          </Context>
-        </PromptInputTools>
-        <PromptInputSubmit
-          disabled={!canSubmit}
-          status={submitStatus}
-        />
-      </PromptInputFooter>
-    </PromptInput>
   )
 }
 
@@ -324,9 +300,6 @@ interface ChatInputWithMentionsProps {
   visibleFiles: string[]
   onSubmit: (message: PromptInputMessage, mentions?: FileMention[]) => void
   isProcessing: boolean
-  contextUsage: LanguageModelUsage
-  maxTokens: number
-  usedTokens: number
 }
 
 function ChatInputWithMentions({
@@ -335,18 +308,12 @@ function ChatInputWithMentions({
   visibleFiles,
   onSubmit,
   isProcessing,
-  contextUsage,
-  maxTokens,
-  usedTokens,
 }: ChatInputWithMentionsProps) {
   return (
     <PromptInputProvider knowledgeFiles={knowledgeFiles} recentFiles={recentFiles} visibleFiles={visibleFiles}>
       <ChatInputInner
         onSubmit={onSubmit}
         isProcessing={isProcessing}
-        contextUsage={contextUsage}
-        maxTokens={maxTokens}
-        usedTokens={usedTokens}
       />
     </PromptInputProvider>
   )
@@ -1259,7 +1226,7 @@ function App() {
                   <ConversationScrollButton className="bottom-24" />
                 </Conversation>
 
-                <div className="sticky bottom-0 z-10 bg-background pb-4 pt-6 shadow-lg">
+                <div className="sticky bottom-0 z-10 bg-background pb-12 pt-0 shadow-lg">
                   <div className="pointer-events-none absolute inset-x-0 -top-6 h-6 bg-linear-to-t from-background to-transparent" />
                   <div className="mx-auto w-full max-w-4xl px-4">
                     <ChatInputWithMentions
@@ -1268,9 +1235,6 @@ function App() {
                       visibleFiles={visibleKnowledgeFiles}
                       onSubmit={handlePromptSubmit}
                       isProcessing={isProcessing}
-                      contextUsage={contextUsage}
-                      maxTokens={maxTokens}
-                      usedTokens={usedTokens}
                     />
                   </div>
                 </div>

apps/x/apps/renderer/src/components/ai-elements/prompt-input.tsx

@@ -1123,7 +1123,7 @@ export const PromptInputTextarea = ({
       };
 
   return (
-    <div ref={containerRef} className="relative flex-1 min-w-0">
+    <div ref={containerRef} className="relative contents">
       {mentionHighlights.hasHighlights && (
         <div
           ref={highlightRef}
@@ -1146,10 +1146,7 @@ export const PromptInputTextarea = ({
       )}
       <InputGroupTextarea
         ref={textareaRef}
-        className={cn(
+        className={cn("field-sizing-content max-h-48 min-h-10", className)}
-          "field-sizing-content max-h-48 min-h-16 relative z-10",
-          className
-        )}
         name="message"
         onCompositionEnd={() => setIsComposing(false)}
         onCompositionStart={() => setIsComposing(true)}

apps/x/apps/renderer/src/components/chat-sidebar.tsx

@@ -478,7 +478,7 @@ export function ChatSidebar({
 
         {/* Input area - responsive to sidebar width, matches floating bar position exactly */}
         <div className="absolute bottom-6 left-14 right-6 z-10" ref={containerRef}>
-          <div className="flex items-center gap-2 bg-background border border-border rounded-2xl shadow-xl px-4 py-2.5">
+          <div className="flex items-center gap-2 bg-background border border-border rounded-3xl shadow-xl px-4 py-2.5">
             <div className="relative flex-1 min-w-0">
               {mentionHighlights.hasHighlights && (
                 <div
@@ -518,7 +518,7 @@ export function ChatSidebar({
               onClick={handleSubmit}
               disabled={!canSubmit}
               className={cn(
-                "h-7 w-7 rounded-full shrink-0 transition-all self-end",
+                "h-7 w-7 rounded-full shrink-0 transition-all",
                 canSubmit
                   ? "bg-primary text-primary-foreground hover:bg-primary/90"
                   : "bg-muted text-muted-foreground"