Add brain emoji model selector with fullscreen modal

- Create ModelSelectionModal component with list-style interface
- Add brain emoji (🧠) indicator below chat input, left-aligned
- Show current model name in lowercase or red X if not configured
- Fullscreen modal displays all available models with configuration status
- Green checkmark for configured models, red alert for unconfigured
- Integrate with existing API key management and model selection logic
- Modal handles API key setup workflow when selecting unconfigured models
- Clean separation between compact indicator and full selection interface
This commit is contained in:
Glen Maddern 2025-06-23 15:49:45 +10:00
commit 6a79264315
3 changed files with 194 additions and 0 deletions

View file

@ -140,6 +140,8 @@ const ChatApp: React.FC<ChatAppProps> = () => {
db={db as IDBPDatabase}
selectedModel={selectedModel}
onApiKeyUpdate={handleApiKeyUpdate}
onModelChange={handleModelChange}
apiKeyUpdateTrigger={apiKeyUpdateTrigger}
mcpTools={mcpTools}
/>
</div>

View file

@ -8,9 +8,11 @@ import { type IDBPDatabase } from 'idb'
import { type Tool } from 'use-mcp/react'
import ChatMessage from './ChatMessage'
import ChatInput from './ChatInput'
import ModelSelectionModal from './ModelSelectionModal'
import { useAutoscroll } from '../hooks/useAutoscroll'
import { useStreamResponse } from '../hooks/useStreamResponse'
import { setApiKey } from '../utils/apiKeys'
import { hasApiKey } from '../utils/apiKeys'
import ApiKeyModal from './ApiKeyModal'
interface ConversationThreadProps {
@ -21,6 +23,8 @@ interface ConversationThreadProps {
db: IDBPDatabase
selectedModel: Model
onApiKeyUpdate: () => void
onModelChange: (model: Model) => void
apiKeyUpdateTrigger: number
mcpTools: Tool[]
}
@ -32,6 +36,8 @@ const ConversationThread: React.FC<ConversationThreadProps> = ({
db,
selectedModel,
onApiKeyUpdate,
onModelChange,
apiKeyUpdateTrigger,
mcpTools,
}) => {
const [input, setInput] = useState<string>('')
@ -39,6 +45,7 @@ const ConversationThread: React.FC<ConversationThreadProps> = ({
isOpen: false,
model: null,
})
const [modelSelectionModal, setModelSelectionModal] = useState(false)
const { messagesEndRef, messagesContainerRef, scrollToBottom } = useAutoscroll()
@ -212,6 +219,23 @@ const ConversationThread: React.FC<ConversationThreadProps> = ({
controller={controller}
messagesCount={currentConversation.messages.length}
/>
{/* Model selector indicator */}
<div className="flex items-center gap-2 mt-2">
<button
onClick={() => setModelSelectionModal(true)}
className="flex items-center gap-2 text-zinc-600 hover:text-zinc-900 transition-colors"
>
<span className="text-lg">🧠</span>
<span className="text-sm text-zinc-500">
{hasApiKey(selectedModel.provider.id) ? (
selectedModel.name.toLowerCase()
) : (
<span className="text-red-500">✕</span>
)}
</span>
</button>
</div>
</div>
</div>
@ -221,6 +245,14 @@ const ConversationThread: React.FC<ConversationThreadProps> = ({
provider={apiKeyModal.model?.provider ?? { id: '', name: '', baseUrl: '', apiKeyHeader: '', documentationUrl: '' }}
onSave={handleApiKeySave}
/>
<ModelSelectionModal
isOpen={modelSelectionModal}
onClose={() => setModelSelectionModal(false)}
selectedModel={selectedModel}
onModelChange={onModelChange}
apiKeyUpdateTrigger={apiKeyUpdateTrigger}
/>
</div>
)
}

View file

@ -0,0 +1,160 @@
import React, { useState, useEffect } from 'react'
import { X, AlertCircle, CheckCircle } from 'lucide-react'
import { type Model, availableModels } from '../types/models'
import { hasApiKey, setApiKey, clearApiKey } from '../utils/apiKeys'
import ApiKeyModal from './ApiKeyModal'
interface ModelSelectionModalProps {
isOpen: boolean
onClose: () => void
selectedModel: Model
onModelChange: (model: Model) => void
apiKeyUpdateTrigger: number
}
const ModelSelectionModal: React.FC<ModelSelectionModalProps> = ({
isOpen,
onClose,
selectedModel,
onModelChange,
apiKeyUpdateTrigger,
}) => {
const [apiKeyModal, setApiKeyModal] = useState<{ isOpen: boolean; model: Model | null }>({
isOpen: false,
model: null,
})
const [apiKeyStatuses, setApiKeyStatuses] = useState<Record<string, boolean>>({})
useEffect(() => {
const statuses: Record<string, boolean> = {}
availableModels.forEach(model => {
statuses[model.provider.id] = hasApiKey(model.provider.id)
})
setApiKeyStatuses(statuses)
}, [apiKeyUpdateTrigger])
useEffect(() => {
if (isOpen) {
document.body.style.overflow = 'hidden'
} else {
document.body.style.overflow = 'unset'
}
return () => {
document.body.style.overflow = 'unset'
}
}, [isOpen])
const handleModelSelect = (model: Model) => {
const hasKey = hasApiKey(model.provider.id)
if (!hasKey) {
setApiKeyModal({ isOpen: true, model })
} else {
onModelChange(model)
onClose()
}
}
const handleApiKeySave = (apiKey: string) => {
if (apiKeyModal.model) {
setApiKey(apiKeyModal.model.provider.id, apiKey)
setApiKeyStatuses(prev => ({ ...prev, [apiKeyModal.model!.provider.id]: true }))
onModelChange(apiKeyModal.model)
setApiKeyModal({ isOpen: false, model: null })
onClose()
}
}
const handleClearApiKey = (providerId: string, e: React.MouseEvent) => {
e.stopPropagation()
clearApiKey(providerId)
setApiKeyStatuses(prev => ({ ...prev, [providerId]: false }))
}
const getStatusIcon = (providerId: string) => {
const hasKey = apiKeyStatuses[providerId]
if (hasKey) {
return <CheckCircle size={20} className="text-green-500" />
} else {
return <AlertCircle size={20} className="text-red-500" />
}
}
if (!isOpen) return null
return (
<>
<div className="fixed inset-0 bg-black bg-opacity-50 z-50 flex items-center justify-center p-4">
<div className="bg-white rounded-lg shadow-xl w-full max-w-2xl max-h-[80vh] overflow-hidden">
<div className="flex items-center justify-between p-6 border-b border-zinc-200">
<h2 className="text-xl font-semibold text-zinc-900">Select Model</h2>
<button
onClick={onClose}
className="text-zinc-400 hover:text-zinc-600 p-1"
>
<X size={24} />
</button>
</div>
<div className="p-6 overflow-y-auto">
<div className="space-y-3">
{availableModels.map((model) => {
const isSelected = model.id === selectedModel.id
const hasKey = apiKeyStatuses[model.provider.id]
return (
<div
key={model.id}
className={`border rounded-lg p-4 cursor-pointer transition-colors ${
isSelected
? 'border-blue-500 bg-blue-50'
: 'border-zinc-200 hover:border-zinc-300 hover:bg-zinc-50'
}`}
onClick={() => handleModelSelect(model)}
>
<div className="flex items-center justify-between">
<div className="flex items-center gap-3">
<span className="text-2xl">{model.provider.logo}</span>
<div>
<div className="font-medium text-zinc-900">{model.name}</div>
<div className="text-sm text-zinc-500">{model.provider.name}</div>
</div>
</div>
<div className="flex items-center gap-2">
{getStatusIcon(model.provider.id)}
<span className={`text-sm ${hasKey ? 'text-green-600' : 'text-red-600'}`}>
{hasKey ? 'Configured' : 'Not configured'}
</span>
{hasKey && (
<button
onClick={(e) => handleClearApiKey(model.provider.id, e)}
className="text-zinc-400 hover:text-red-500 p-1 ml-2"
title="Clear API key"
>
<X size={16} />
</button>
)}
</div>
</div>
</div>
)
})}
</div>
</div>
</div>
</div>
<ApiKeyModal
isOpen={apiKeyModal.isOpen}
onClose={() => setApiKeyModal({ isOpen: false, model: null })}
provider={apiKeyModal.model?.provider ?? { id: '', name: '', baseUrl: '', apiKeyHeader: '', documentationUrl: '' }}
onSave={handleApiKeySave}
/>
</>
)
}
export default ModelSelectionModal