Cleanup

Author: IgorWounds

app/api/user-settings/route.ts

@@ -58,28 +58,31 @@ export async function PATCH(req: NextRequest) {
   const body = await req.json()
   const { model } = body
   if (!model) return NextResponse.json({ error: 'Model required' }, { status: 400 })
-  // Get current count and last reset
+  const today = new Date().toISOString().slice(0, 10)
+  // Atomic update: reset if needed, then decrement if premium
   const { data, error: userError } = await supabase
     .from('users')
     .select('daily_query_count, last_query_reset')
     .eq('id', user.id)
     .single()
   if (userError) return NextResponse.json({ error: userError.message }, { status: 500 })
-  const today = new Date().toISOString().slice(0, 10)
-  let daily_query_count = data?.daily_query_count ?? 50
-  const last_query_reset = data?.last_query_reset
-  // Reset if last_query_reset is not today
+  let { daily_query_count, last_query_reset } = data ?? {}
   if (last_query_reset !== today) {
     daily_query_count = 50
-    await supabase.from('users').update({ daily_query_count, last_query_reset: today }).eq('id', user.id)
+    last_query_reset = today
   }
-  // Only decrement if not free model
+  // Only decrement for premium models
   if (!/free/i.test(model)) {
     if (daily_query_count <= 0) {
-      return NextResponse.json({ error: 'Query limit reached', dailyQueryCount: 0 }, { status: 403 })
+      return NextResponse.json({ error: 'Query limit reached' }, { status: 403 })
     }
-    daily_query_count -= 1
-    await supabase.from('users').update({ daily_query_count }).eq('id', user.id)
+    daily_query_count--
   }
+  // Single update
+  const { error: updateError } = await supabase
+    .from('users')
+    .update({ daily_query_count, last_query_reset })
+    .eq('id', user.id)
+  if (updateError) return NextResponse.json({ error: updateError.message }, { status: 500 })
   return NextResponse.json({ dailyQueryCount: daily_query_count })
 } 

components/chat/chat-input.tsx

@@ -463,7 +463,7 @@ const ChatInput = React.memo(function ChatInput({ onOpenSearch, defaultModel }:
                 </Tooltip.Content>
               </Tooltip.Portal>
             </Tooltip.Root>
-            {[Search, Globe, ImageIcon, MoreHorizontal].map((Icon, i) => (
+            {[Search].map((Icon, i) => (
               <Tooltip.Root key={i} delayDuration={200}>
                 <Tooltip.Trigger asChild>
                   <Button
@@ -487,7 +487,7 @@ const ChatInput = React.memo(function ChatInput({ onOpenSearch, defaultModel }:
                       fontFamily: "var(--font-sans)",
                     }}
                   >
-                    {["Search", "Web search", "Create image", "More"][i]}
+                    {["Search"][i]}
                     <Tooltip.Arrow className="fill-[var(--popover)]" />
                   </Tooltip.Content>
                 </Tooltip.Portal>

components/chat/file-upload.tsx

@@ -6,7 +6,7 @@ import { Input } from '@/components/ui/input'
 const MAX_SIZE_MB = 10
 const ALLOWED_TYPES = [
   'image/png', 'image/jpeg', 'image/webp', 'application/pdf',
-  'text/plain', 'application/zip', 'application/json',
+  'text/plain', 'application/zip', 'application/json', 'text/markdown'
 ]
 
 export type FileUploadProps = {

components/chat/sidebar-footer.tsx

@@ -1,4 +1,4 @@
-import { useEffect } from 'react'
+import { useEffect, useState } from 'react'
 import useSWR from 'swr'
 import { usePremiumQueryCountStore } from '@/hooks/use-premium-query-count-store'
 
@@ -7,15 +7,32 @@ export default function SidebarFooter() {
   const { data: userSettings } = useSWR('/api/user-settings', fetcher, { revalidateOnFocus: false })
   const count = usePremiumQueryCountStore(s => s.count)
   const isUnlimited = usePremiumQueryCountStore(s => s.isUnlimited)
-  const set = usePremiumQueryCountStore(s => s.set)
+  const syncFromDb = usePremiumQueryCountStore(s => s.syncFromDb)
+  const [hydrated, setHydrated] = useState(false)
+
+  useEffect(() => {
+    setHydrated(true)
+  }, [])
+
   useEffect(() => {
-    if (userSettings) set(userSettings.dailyQueryCount ?? 50, !!userSettings.hasTogetherApiKey)
-  }, [userSettings, set])
+    if (userSettings && typeof userSettings.dailyQueryCount === 'number') {
+      syncFromDb(userSettings.dailyQueryCount, !!userSettings.hasTogetherApiKey)
+    }
+  }, [userSettings, syncFromDb])
+
+  if (!hydrated) return null
+
   return (
     <div className="flex items-center justify-center p-3 border-t border-[#2a2b32] text-xs text-[#b4bcd0] mt-auto w-full">
-      <span className="px-3 py-1 rounded-full bg-[#23272f] border border-[#353740] text-[#b4bcd0] text-sm font-medium" title="Premium queries available today">
-        Daily premium queries: {isUnlimited ? 'Unlimited' : `${count}/50`}
-      </span>
+      {typeof count === 'number' ? (
+        <span className="px-3 py-1 rounded-full bg-[#23272f] border border-[#353740] text-[#b4bcd0] text-sm font-medium" title="Premium queries available today">
+          Premium queries: {isUnlimited ? 'Unlimited' : `${count}/50`}
+        </span>
+      ) : (
+        <span className="px-3 py-1 rounded-full bg-[#23272f] border border-[#353740] text-[#b4bcd0] text-sm font-medium opacity-60 animate-pulse" title="Loading premium queries...">
+          Premium queries: ...
+        </span>
+      )}
     </div>
   )
 } 

hooks/use-premium-query-count-store.ts

@@ -1,17 +1,53 @@
 import { create } from 'zustand'
 
 interface PremiumQueryCountState {
-  count: number
+  count: number | null
   isUnlimited: boolean
   set: (count: number, isUnlimited: boolean) => void
   decrement: () => void
   reset: () => void
+  syncFromDb: (count: number, isUnlimited: boolean) => void
 }
 
-export const usePremiumQueryCountStore = create<PremiumQueryCountState>((set) => ({
-  count: 50,
-  isUnlimited: false,
-  set: (count, isUnlimited) => set({ count, isUnlimited }),
-  decrement: () => set((s) => s.isUnlimited ? s : { ...s, count: Math.max(0, s.count - 1) }),
-  reset: () => set({ count: 50, isUnlimited: false }),
-})) 
+function persist(count: number | null, isUnlimited: boolean) {
+  if (typeof window !== 'undefined') {
+    if (count !== null) localStorage.setItem('premiumQueryCount', String(count))
+    localStorage.setItem('isUnlimited', String(isUnlimited))
+  }
+}
+
+export const usePremiumQueryCountStore = create<PremiumQueryCountState>((set, get) => {
+  // Hydrate from localStorage
+  let count: number | null = 50
+  let isUnlimited = false
+  if (typeof window !== 'undefined') {
+    const stored = localStorage.getItem('premiumQueryCount')
+    const unlimited = localStorage.getItem('isUnlimited')
+    if (stored) count = Number(stored)
+    if (unlimited) isUnlimited = unlimited === 'true'
+  }
+  return {
+    count,
+    isUnlimited,
+    set: (count, isUnlimited) => {
+      persist(count, isUnlimited)
+      set({ count, isUnlimited })
+    },
+    decrement: () => {
+      const { count, isUnlimited } = get()
+      if (!isUnlimited && count !== null) {
+        const newCount = Math.max(0, count - 1)
+        persist(newCount, isUnlimited)
+        set({ count: newCount })
+      }
+    },
+    reset: () => {
+      persist(50, false)
+      set({ count: 50, isUnlimited: false })
+    },
+    syncFromDb: (count, isUnlimited) => {
+      persist(count, isUnlimited)
+      set({ count, isUnlimited })
+    },
+  }
+}) 

knowledge_base/tasks.md

@@ -110,14 +110,13 @@
 
 ## Phase 6: User Settings and Preferences
 
-- [ ] Create settings page
-- [ ] Allow user to store their together.ai api key and use it for requests
-- [ ] Force users to use their own API key for model requests that are not free (we know free ones as they have "free" in the name)
-- [ ] Add message history preferences
-- [ ] Implement settings persistence
-- [ ] Add user profile management [name, email, account deletion]
-
-## Phase 7: Performance and Polish
+- [x] Create settings page
+- [x] Allow user to store their together.ai api key and use it for requests
+- [x] Force users to use their own API key for model requests that are not free (we know free ones as they have "free" in the name)
+- [x] Implement settings persistence
+- [x] Add user profile management [name, email, account deletion]
+
+## Phase 7: Performance and Polish (NEXT)
 
 - [ ] Implement proper error handling
 - [ ] Add loading states and skeletons