Merge pull request #6 from Ghvstcode/fix-budget-daily-mode

fix(tauri,ui): switch budget to daily mode and increase default

Author: Ghvstcode

src-tauri/src/engine/budget.rs

@@ -56,7 +56,7 @@ pub struct BudgetConfig {
 impl Default for BudgetConfig {
     fn default() -> Self {
         Self {
-            weekly_token_budget: 700_000,
+            weekly_token_budget: 5_000_000,
             max_usage_percent: 80,
             reserve_percent: 10,
             billing_mode: "subscription".to_string(),
@@ -143,14 +143,19 @@ pub fn get_weekly_usage() -> Result<(i64, String), String> {
 }
 
 /// Calculate the full budget status given a budget config.
+///
+/// Uses **daily mode**: compares today's usage against a daily budget (weekly / 7)
+/// instead of comparing weekly totals against the weekly budget. This prevents heavy
+/// usage earlier in the week from falsely exhausting the budget for remaining days —
+/// the actual subscription resets on rolling windows, not as a rigid weekly lump sum.
 pub fn calculate_budget_status(config: &BudgetConfig) -> BudgetStatus {
     let (tokens_today, _) = get_today_usage().unwrap_or((0, "unavailable".to_string()));
     let (tokens_week, source) = get_weekly_usage().unwrap_or((0, "unavailable".to_string()));
 
-    let max_for_sustn =
-        config.weekly_token_budget * (config.max_usage_percent as i64) / 100;
-    let reserve = config.weekly_token_budget * (config.reserve_percent as i64) / 100;
-    let available = (max_for_sustn - tokens_week - reserve).max(0);
+    let daily_budget = config.weekly_token_budget / 7;
+    let max_for_sustn = daily_budget * (config.max_usage_percent as i64) / 100;
+    let reserve = daily_budget * (config.reserve_percent as i64) / 100;
+    let available = (max_for_sustn - tokens_today - reserve).max(0);
 
     BudgetStatus {
         weekly_token_budget: config.weekly_token_budget,

src-tauri/src/migrations.rs

@@ -292,5 +292,15 @@ pub fn migrations() -> Vec<Migration> {
             "#,
             kind: MigrationKind::Up,
         },
+        Migration {
+            version: 15,
+            description: "increase default weekly_token_budget from 700k to 5M",
+            sql: r#"
+                UPDATE budget_config
+                SET weekly_token_budget = 5000000
+                WHERE weekly_token_budget = 700000;
+            "#,
+            kind: MigrationKind::Up,
+        },
     ]
 }

src/core/db/agent-config.ts

@@ -165,7 +165,7 @@ export async function getBudgetConfig(): Promise<BudgetConfig> {
     }
 
     return {
-        weeklyTokenBudget: 700_000,
+        weeklyTokenBudget: 5_000_000,
         maxUsagePercent: 80,
         reservePercent: 10,
         billingMode: "subscription",

src/ui/components/settings/sections/BudgetSection.tsx

@@ -56,7 +56,7 @@ export function BudgetSection() {
                             />
                             <div className="flex items-center justify-between">
                                 <p className="text-sm font-medium text-foreground">
-                                    Use up to {ceiling}% of weekly budget
+                                    Use up to {ceiling}% of daily budget
                                 </p>
                                 <p className="text-[13px] text-muted-foreground">
                                     {reserved}% reserved for manual use

src/ui/components/sidebar/AiStatusCard.tsx

@@ -22,15 +22,13 @@ export function AiStatusCard() {
     const isEnabled = agentConfig?.enabled ?? true;
     const isExhausted = budget?.budgetExhausted ?? false;
 
-    // Budget bar calculation
-    const budgetCeiling = budget
-        ? (budget.weeklyTokenBudget * budget.maxUsagePercent) / 100
+    // Budget bar — uses daily budget (weekly / 7) to match backend calculation
+    const dailyBudget = budget
+        ? ((budget.weeklyTokenBudget / 7) * budget.maxUsagePercent) / 100
         : 0;
     const available = budget?.tokensAvailableForSustn ?? 0;
     const usedPercent =
-        budgetCeiling > 0
-            ? ((budgetCeiling - available) / budgetCeiling) * 100
-            : 0;
+        dailyBudget > 0 ? ((dailyBudget - available) / dailyBudget) * 100 : 0;
     const remainingPercent = 100 - usedPercent;
 
     // Bar color
@@ -46,7 +44,7 @@ export function AiStatusCard() {
     if (isExhausted) {
         icon = <AlertTriangle className="h-3 w-3 shrink-0 text-amber-500" />;
         line1 = "Budget limit";
-        line2 = "Resets Monday";
+        line2 = "Resets tomorrow";
     } else if (!isEnabled) {
         icon = (
             <svg