feat: implement subscription packages, message limit quotas, and 3-day discount timer

- Database: Add subscription_tier, messages_remaining, monthly_messages_used, and last_billing_date to User model
- Migration: Generate and apply Alembic migration with server_defaults for existing rows
- Backend: Enforce bearer token authorization and message limits on /analyst/chat endpoint
- Backend: Map Razorpay webhook payment capture to user subscription tiers and credit allocations
- Frontend: Implement live ticking 3-day countdown timer for Pro pack discount (₹10k vs ₹15k)
- Frontend: Show real-time remaining quota badge in Chatbot header and lock chat input when limit is reached

alembic/versions/bbdbf37359be_add_subscription_and_message_limit_.py

@@ -0,0 +1,38 @@
+"""add_subscription_and_message_limit_columns
+
+Revision ID: bbdbf37359be
+Revises: 15639881888f
+Create Date: 2026-07-04 19:07:25.250443
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'bbdbf37359be'
+down_revision: Union[str, Sequence[str], None] = '15639881888f'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('users', sa.Column('subscription_tier', sa.String(length=50), server_default='free', nullable=False))
+    op.add_column('users', sa.Column('messages_remaining', sa.Integer(), server_default='3', nullable=False))
+    op.add_column('users', sa.Column('monthly_messages_used', sa.Integer(), server_default='0', nullable=False))
+    op.add_column('users', sa.Column('last_billing_date', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('users', 'last_billing_date')
+    op.drop_column('users', 'monthly_messages_used')
+    op.drop_column('users', 'messages_remaining')
+    op.drop_column('users', 'subscription_tier')
+    # ### end Alembic commands ###

backend/app/api/endpoints.py

@@ -521,11 +521,29 @@ class ChatRequest(BaseModel):
     activeIndicators: Dict[str, bool]
 
 @router.post("/analyst/chat")
-async def chat_with_analyst(payload: ChatRequest):
+async def chat_with_analyst(
+    payload: ChatRequest,
+    current_user: models.User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db)
+):
     """
     Takes the user's message, drawing markers, active indicators, and chat history,
     and returns a cooperative analysis message from Gemini.
+    Enforces subscription message limits.
     """
+    # 1. Refresh user billing cycles / credits
+    user = await crud.refresh_user_credits(db, current_user)
+    is_admin = (user.email == "karanshelar8775@gmail.com")
+
+    # 2. Check limits
+    if not is_admin:
+        if user.subscription_tier in ("free", "analyst", "trader"):
+            if user.messages_remaining <= 0:
+                raise HTTPException(status_code=403, detail="Quota exhausted. Please upgrade your plan.")
+        elif user.subscription_tier == "pro":
+            if user.monthly_messages_used >= 100:
+                raise HTTPException(status_code=403, detail="Monthly message quota of 100 exhausted.")
+
     ticker = payload.ticker
     message = payload.message
     history = payload.history
@@ -595,4 +613,21 @@ async def chat_with_analyst(payload: ChatRequest):
 
     from backend.app.services.gemini import generate_text
     response_text = await generate_text(full_prompt)
-    return {"response": response_text}
+
+    # 3. Deduct/Increment message counts
+    if not is_admin:
+        if user.subscription_tier in ("free", "analyst", "trader"):
+            user.messages_remaining = max(0, user.messages_remaining - 1)
+        elif user.subscription_tier == "pro":
+            user.monthly_messages_used += 1
+        
+        db.add(user)
+        await db.commit()
+        await db.refresh(user)
+
+    return {
+        "response": response_text,
+        "subscription_tier": user.subscription_tier,
+        "messages_remaining": user.messages_remaining,
+        "monthly_messages_used": user.monthly_messages_used
+    }

backend/app/database/crud.py

@@ -39,15 +39,28 @@ async def create_user(db: AsyncSession, user_in: schemas.UserBase, google_id: st
 
 async def refresh_user_credits(db: AsyncSession, user: models.User) -> models.User:
     """
-    Checks if 7 days have passed since the last credit refresh.
-    If so, resets credits back to 5.
+    Checks and applies user refreshes:
+    - For Free tier: resets credits to 5 every 7 days.
+    - For Pro tier: resets monthly_messages_used to 0 every 30 days.
     """
     now = datetime.datetime.now(datetime.timezone.utc)
-    time_elapsed = now - user.last_credit_refresh
+    updated = False
     
-    if time_elapsed >= datetime.timedelta(days=7):
-        user.credits = 5
-        user.last_credit_refresh = now
+    if user.subscription_tier == "free":
+        time_elapsed = now - user.last_credit_refresh
+        if time_elapsed >= datetime.timedelta(days=7):
+            user.credits = 5
+            user.last_credit_refresh = now
+            updated = True
+            
+    if user.subscription_tier == "pro":
+        billing_elapsed = now - user.last_billing_date
+        if billing_elapsed >= datetime.timedelta(days=30):
+            user.monthly_messages_used = 0
+            user.last_billing_date = now
+            updated = True
+            
+    if updated:
         db.add(user)
         await db.commit()
         await db.refresh(user)
@@ -287,10 +300,24 @@ async def capture_payment_transaction(
     tx.status = "captured"
     db.add(tx)
     
-    # 3. Credit the user
+    # 3. Credit the user and update subscription tier
     user = await get_user(db, tx.user_id)
     if user:
         user.credits += tx.credits_credited
+        
+        # Determine plan from transaction amount in Rupees
+        amt_rupees = tx.amount // 100
+        if amt_rupees == 500:
+            user.subscription_tier = "analyst"
+            user.messages_remaining += 10
+        elif amt_rupees == 1500:
+            user.subscription_tier = "trader"
+            user.messages_remaining += 25
+        elif amt_rupees in (10000, 15000):
+            user.subscription_tier = "pro"
+            user.monthly_messages_used = 0
+            user.last_billing_date = datetime.datetime.now(datetime.timezone.utc)
+            
         db.add(user)
         
     await db.commit()

backend/app/database/models.py

@@ -27,6 +27,16 @@ class User(Base):
         nullable=False
     )
 
+    # Subscription parameters
+    subscription_tier: Mapped[str] = mapped_column(String(50), default="free", nullable=False)
+    messages_remaining: Mapped[int] = mapped_column(Integer, default=3, nullable=False)
+    monthly_messages_used: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
+    last_billing_date: Mapped[datetime.datetime] = mapped_column(
+        DateTime(timezone=True), 
+        server_default=func.now(), 
+        nullable=False
+    )
+
     # Relationships
     watchlists: Mapped[List["Watchlist"]] = relationship(back_populates="user", cascade="all, delete-orphan")
     alerts: Mapped[List["Alert"]] = relationship(back_populates="user", cascade="all, delete-orphan")

backend/app/graphql/schema.py

@@ -29,6 +29,10 @@ class UserType:
     credits: int
     last_credit_refresh: datetime.datetime
     created_at: datetime.datetime
+    subscription_tier: str
+    messages_remaining: int
+    monthly_messages_used: int
+    last_billing_date: datetime.datetime
 
 @strawberry.type
 class WatchlistType:
@@ -438,10 +442,8 @@ class Mutation:
             credits_credited = 10
         elif amount == 1500:
             credits_credited = 50
-        elif amount == 2500:
+        elif amount in (10000, 15000):
             credits_credited = 100
-        elif amount == 10500:
-            credits_credited = 999999  # Code for Unlimited / Lifetime
         else:
             credits_credited = amount // 50
         

backend/app/schemas/schemas.py

@@ -25,6 +25,10 @@ class UserResponse(UserBase):
     credits: int
     last_credit_refresh: datetime.datetime
     created_at: datetime.datetime
+    subscription_tier: str
+    messages_remaining: int
+    monthly_messages_used: int
+    last_billing_date: datetime.datetime
     
     model_config= ConfigDict(from_attributes= True)
 

frontend/src/App.tsx

@@ -129,6 +129,10 @@ export default function App() {
               pictureUrl
               credits
               createdAt
+              subscriptionTier
+              messagesRemaining
+              monthlyMessagesUsed
+              lastBillingDate
             }
           }
         `);
@@ -567,10 +571,21 @@ export default function App() {
                 query {
                   me {
                     credits
+                    subscriptionTier
+                    messagesRemaining
+                    monthlyMessagesUsed
+                    lastBillingDate
                   }
                 }
               `);
-              setUser((prev: any) => prev ? { ...prev, credits: updatedProfile.me.credits } : null);
+              setUser((prev: any) => prev ? { 
+                ...prev, 
+                credits: updatedProfile.me.credits,
+                subscriptionTier: updatedProfile.me.subscriptionTier,
+                messagesRemaining: updatedProfile.me.messagesRemaining,
+                monthlyMessagesUsed: updatedProfile.me.monthlyMessagesUsed,
+                lastBillingDate: updatedProfile.me.lastBillingDate
+              } : null);
             } catch (e) {
               console.error('Refresh credits error:', e);
             }

frontend/src/components/ChartChatbot.tsx

@@ -30,15 +30,38 @@ export default function ChartChatbot({ ticker, markers, activeIndicators, user,
   const [input, setInput] = useState('');
   const [loading, setLoading] = useState(false);
   
-  // Track chatbot trial message count
-  const [trialUsed, setTrialUsed] = useState<boolean>(() => {
-    return localStorage.getItem('quantiq_chatbot_trial_used') === 'true';
-  });
+  // Local quota tracking synchronized with parent user prop
+  const [localRemaining, setLocalRemaining] = useState<number>(() => user?.messagesRemaining ?? 0);
+  const [localUsed, setLocalUsed] = useState<number>(() => user?.monthlyMessagesUsed ?? 0);
+  const [localTier, setLocalTier] = useState<string>(() => user?.subscriptionTier || 'free');
+
+  useEffect(() => {
+    if (user) {
+      setLocalRemaining(user.messagesRemaining ?? 0);
+      setLocalUsed(user.monthlyMessagesUsed ?? 0);
+      setLocalTier(user.subscriptionTier || 'free');
+    }
+  }, [user]);
 
   const messagesEndRef = useRef<HTMLDivElement>(null);
   const textareaRef = useRef<HTMLTextAreaElement>(null);
   const API_URL = import.meta.env.VITE_API_URL || 'http://localhost:8000';
 
+  const isAdmin = user?.email === 'karanshelar8775@gmail.com';
+  
+  const isLimitReached = !isAdmin && (
+    (localTier !== 'pro' && localRemaining <= 0) ||
+    (localTier === 'pro' && localUsed >= 100)
+  );
+
+  const getQuotaDisplay = () => {
+    if (isAdmin) return 'Unlimited (Admin)';
+    if (localTier === 'pro') {
+      return `${100 - localUsed} left`;
+    }
+    return `${localRemaining} left`;
+  };
+
   // Helper to format text markdown bold/italics, headers, bullets, and clean up stray symbols
   const formatMessageContent = (text: string) => {
     if (!text) return '';
@@ -48,59 +71,56 @@ export default function ChartChatbot({ ticker, markers, activeIndicators, user,
       
       // 1. Convert headers (e.g., #### Header or ### Header) to clean bold blocks
       const headerMatch = cleanedLine.match(/^(#{1,6})\s*(.*)$/);
-      let isHeader = false;
       if (headerMatch) {
-        cleanedLine = headerMatch[2];
-        isHeader = true;
-      }
-      
-      // 2. Convert bullet list points into clean lists
-      const bulletMatch = cleanedLine.match(/^(\*|-)\s+(.*)$/);
-      let isBullet = false;
-      if (bulletMatch) {
-        cleanedLine = bulletMatch[2];
-        isBullet = true;
-      }
-      
-      // 3. Parse inline bold (**text**) and italic (*text*) segments
-      const parts = cleanedLine.split(/(\*\*.*?\*\*|\*.*?\*)/g);
-      const parsedInline = parts.map((part, i) => {
-        if (part.startsWith('**') && part.endsWith('**')) {
-          return <strong key={i} style={{ color: '#fff', fontWeight: 700 }}>{part.slice(2, -2)}</strong>;
-        } else if (part.startsWith('*') && part.endsWith('*')) {
-          return <span key={i} style={{ color: 'var(--neon-cyan)', fontWeight: 500 }}>{part.slice(1, -1)}</span>;
-        }
-        // Remove any leftover stray asterisks in plain text segments
-        return part.replace(/\*/g, '');
-      });
-
-      // Render line structure
-      if (isHeader) {
         return (
-          <h4 key={lineIdx} style={{ margin: '14px 0 6px 0', fontSize: '13px', fontWeight: 700, color: '#fff', lineHeight: 1.4 }}>
-            {parsedInline}
+          <h4 key={lineIdx} style={{ margin: '12px 0 6px', fontSize: '13px', fontWeight: 800, color: '#fff', borderBottom: '1px solid rgba(255,255,255,0.05)', paddingBottom: '4px' }}>
+            {headerMatch[2]}
           </h4>
         );
       }
+
+      // 2. Bold text matching **text**
+      const boldRegex = /\*\*(.*?)\*\*/g;
+      const parts = [];
+      let lastIndex = 0;
+      let match;
+      
+      while ((match = boldRegex.exec(cleanedLine)) !== null) {
+        if (match.index > lastIndex) {
+          parts.push(cleanedLine.substring(lastIndex, match.index));
+        }
+        parts.push(<strong key={match.index} style={{ color: 'var(--neon-cyan)', fontWeight: 800 }}>{match[1]}</strong>);
+        lastIndex = boldRegex.lastIndex;
+      }
       
+      if (lastIndex < cleanedLine.length) {
+        parts.push(cleanedLine.substring(lastIndex));
+      }
+
+      const isBullet = cleanedLine.trim().startsWith('-') || cleanedLine.trim().startsWith('*');
+      const content = parts.length > 0 ? parts : cleanedLine;
+
       if (isBullet) {
+        // Strip bullet point indicator
+        const bulletText = typeof content === 'string' 
+          ? content.replace(/^[\s-*]+/, '') 
+          : content;
         return (
-          <div key={lineIdx} style={{ display: 'flex', gap: '6px', alignItems: 'flex-start', margin: '4px 0 4px 8px' }}>
-            <span style={{ color: 'var(--neon-cyan)', fontSize: '12px', lineHeight: '18px' }}>•</span>
-            <div style={{ flex: 1 }}>{parsedInline}</div>
+          <div key={lineIdx} style={{ display: 'flex', gap: '6px', margin: '4px 0 4px 8px', fontSize: '12px', lineHeight: 1.5 }}>
+            <span style={{ color: 'var(--neon-cyan)' }}>•</span>
+            <span style={{ flex: 1 }}>{bulletText}</span>
           </div>
         );
       }
-      
+
       return (
-        <p key={lineIdx} style={{ margin: cleanedLine.trim() === '' ? '0' : '0 0 8px 0', minHeight: cleanedLine.trim() === '' ? '8px' : 'auto' }}>
-          {parsedInline}
+        <p key={lineIdx} style={{ margin: cleanedLine.trim() === '' ? '8px 0' : '4px 0', minHeight: cleanedLine.trim() === '' ? '12px' : 'auto' }}>
+          {content}
         </p>
       );
     });
   };
 
-  // Auto adjust textarea height on input change
   const adjustHeight = () => {
     if (textareaRef.current) {
       textareaRef.current.style.height = 'auto';
@@ -112,10 +132,6 @@ export default function ChartChatbot({ ticker, markers, activeIndicators, user,
     adjustHeight();
   }, [input]);
 
-  // Check if user is a paid tier (Trader/Pro packs have > 10 credits) or is the test email
-  const isPaidUser = user && (user.credits > 10 || user.email === 'karanshelar8775@gmail.com');
-  const isTrialLocked = !isPaidUser && trialUsed;
-
   // Load chat history on ticker change
   useEffect(() => {
     const saved = localStorage.getItem(`quantiq_chat_history_${ticker}`);
@@ -143,7 +159,7 @@ export default function ChartChatbot({ ticker, markers, activeIndicators, user,
 
   const handleSendMessage = async (e?: React.FormEvent | React.KeyboardEvent) => {
     if (e) e.preventDefault();
-    if (!input.trim() || loading || isTrialLocked) return;
+    if (!input.trim() || loading || isLimitReached) return;
 
     const userMessage = input.trim();
     setInput('');
@@ -152,7 +168,6 @@ export default function ChartChatbot({ ticker, markers, activeIndicators, user,
     }
     setLoading(true);
 
-    // Add user message to log
     const updatedMessages = [...messages, { role: 'user' as const, content: userMessage }];
     setMessages(updatedMessages);
 
@@ -161,6 +176,7 @@ export default function ChartChatbot({ ticker, markers, activeIndicators, user,
         method: 'POST',
         headers: {
           'Content-Type': 'application/json',
+          'Authorization': `Bearer ${localStorage.getItem('token')}`
         },
         body: JSON.stringify({
           ticker,
@@ -175,13 +191,16 @@ export default function ChartChatbot({ ticker, markers, activeIndicators, user,
         const data = await response.json();
         setMessages(prev => [...prev, { role: 'assistant', content: data.response }]);
         
-        // If they are a free user, trigger the 1-time trial limit
-        if (!isPaidUser) {
-          localStorage.setItem('quantiq_chatbot_trial_used', 'true');
-          setTrialUsed(true);
+        // Update local limits state
+        if (data.subscription_tier !== undefined) {
+          setLocalTier(data.subscription_tier);
+          setLocalRemaining(data.messages_remaining ?? 0);
+          setLocalUsed(data.monthly_messages_used ?? 0);
         }
       } else {
-        setMessages(prev => [...prev, { role: 'assistant', content: "Sorry, I encountered an issue compiling the strategy feedback. Please try again." }]);
+        const errData = await response.json();
+        const errMessage = errData.detail || "Quota exhausted or query error. Check your subscription.";
+        setMessages(prev => [...prev, { role: 'assistant', content: `Sorry, I encountered an issue: ${errMessage}` }]);
       }
     } catch (err) {
       console.error('Failed to chat with AI analyst:', err);
@@ -260,23 +279,21 @@ export default function ChartChatbot({ ticker, markers, activeIndicators, user,
           <span style={{ fontSize: '10px', color: 'var(--text-secondary)' }}>Real-time AI Copilot & Strategy Engine</span>
         </div>
 
-        {/* Trial Badge */}
-        {!isPaidUser && (
-          <span 
-            style={{
-              marginLeft: 'auto',
-              fontSize: '9px',
-              fontWeight: 700,
-              background: trialUsed ? 'rgba(239, 68, 68, 0.15)' : 'rgba(16, 185, 129, 0.15)',
-              color: trialUsed ? '#ef4444' : '#10b981',
-              padding: '2px 6px',
-              borderRadius: '4px',
-              border: `1px solid ${trialUsed ? 'rgba(239, 68, 68, 0.25)' : 'rgba(16, 185, 129, 0.25)'}`
-            }}
-          >
-            {trialUsed ? '0 Trials Left' : '1 Trial Left'}
-          </span>
-        )}
+        {/* Quota display badge */}
+        <span 
+          style={{
+            marginLeft: 'auto',
+            fontSize: '9px',
+            fontWeight: 700,
+            background: isLimitReached ? 'rgba(239, 68, 68, 0.15)' : 'rgba(16, 185, 129, 0.15)',
+            color: isLimitReached ? '#ef4444' : '#10b981',
+            padding: '2px 6px',
+            borderRadius: '4px',
+            border: `1px solid ${isLimitReached ? 'rgba(239, 68, 68, 0.25)' : 'rgba(16, 185, 129, 0.25)'}`
+          }}
+        >
+          {getQuotaDisplay()}
+        </span>
       </div>
 
       {/* Message Feed */}
@@ -365,7 +382,7 @@ export default function ChartChatbot({ ticker, markers, activeIndicators, user,
           background: 'rgba(10, 11, 20, 0.5)'
         }}
       >
-        {isTrialLocked ? (
+        {isLimitReached ? (
           <div 
             className="animate-fade"
             style={{
@@ -377,10 +394,10 @@ export default function ChartChatbot({ ticker, markers, activeIndicators, user,
             }}
           >
             <div style={{ display: 'flex', alignItems: 'center', justifyContent: 'center', gap: '6px', color: '#ef4444', fontSize: '12px', fontWeight: 700 }}>
-              <Lock size={12} /> Trial Expired
+              <Lock size={12} /> Message Limit Reached
             </div>
             <p style={{ margin: 0, fontSize: '10px', color: 'var(--text-secondary)', lineHeight: 1.4 }}>
-              Unlock unlimited interactive chatbot sessions and custom indicator weights.
+              Unlock high-accuracy Wall Street strategies and indicator calculations.
             </p>
             <button 
               onClick={() => {

frontend/src/pages/UpgradePage.tsx

@@ -1,4 +1,5 @@
-import { Check, ArrowLeft } from 'lucide-react';
+import { useState, useEffect } from 'react';
+import { Check, ArrowLeft, Clock } from 'lucide-react';
 import Logo from '../components/Logo';
 
 interface UpgradePageProps {
@@ -8,6 +9,8 @@ interface UpgradePageProps {
 }
 
 export default function UpgradePage({ user, onBack, onSelectPackage }: UpgradePageProps) {
+  const [timeLeft, setTimeLeft] = useState<string>('');
+
   // Check if account age is under 3 days (3 * 24 * 60 * 60 * 1000 = 259200000 ms)
   const isNewUser = (() => {
     if (!user || !user.createdAt) return false;
@@ -21,6 +24,41 @@ export default function UpgradePage({ user, onBack, onSelectPackage }: UpgradePa
     }
   })();
 
+  // Live countdown timer for the special offer
+  useEffect(() => {
+    if (!user || !user.createdAt) return;
+
+    const calculateTimeLeft = () => {
+      const createdDate = new Date(user.createdAt).getTime();
+      const expiryDate = createdDate + 3 * 24 * 60 * 60 * 1000; // 3 days limit
+      const difference = expiryDate - new Date().getTime();
+
+      if (difference <= 0) {
+        setTimeLeft('');
+        return;
+      }
+
+      const days = Math.floor(difference / (24 * 60 * 60 * 1000));
+      const hours = Math.floor((difference % (24 * 60 * 60 * 1000)) / (60 * 60 * 1000));
+      const minutes = Math.floor((difference % (60 * 60 * 1000)) / (60 * 1000));
+      const seconds = Math.floor((difference % (60 * 1000)) / 1000);
+
+      const dStr = days > 0 ? `${days}d ` : '';
+      const hStr = hours.toString().padStart(2, '0');
+      const mStr = minutes.toString().padStart(2, '0');
+      const sStr = seconds.toString().padStart(2, '0');
+
+      setTimeLeft(`${dStr}${hStr}h : ${mStr}m : ${sStr}s`);
+    };
+
+    calculateTimeLeft();
+    const interval = setInterval(calculateTimeLeft, 1000);
+    return () => clearInterval(interval);
+  }, [user]);
+
+  // Determine current active plan
+  const currentTier = user?.subscriptionTier || 'free';
+
   return (
     <div className="upgrade-page-container" style={{ minHeight: '100vh', background: '#07090e', color: 'var(--text-primary)', padding: '40px 24px', position: 'relative', overflowY: 'auto' }}>
       
@@ -62,7 +100,7 @@ export default function UpgradePage({ user, onBack, onSelectPackage }: UpgradePa
             Plans that grow with you
           </h1>
           <p style={{ fontSize: '16px', color: 'var(--text-muted)', maxWidth: '600px', margin: '0 auto' }}>
-            Whether you want to test the waters or build algorithmic trading strategies, choose the plan that fits your ambition.
+            Get access to wall-street level insights. Choose the plan that fits your ambition.
           </p>
         </div>
 
@@ -82,7 +120,7 @@ export default function UpgradePage({ user, onBack, onSelectPackage }: UpgradePa
             <ul style={{ flex: 1, listStyle: 'none', padding: 0, margin: '0 0 32px', display: 'flex', flexDirection: 'column', gap: '14px', fontSize: '13px', color: 'var(--text-secondary)' }}>
               <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
                 <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
-                <span>5 weekly AI strategies</span>
+                <span><strong>3 AI Chat Messages</strong> (Lifetime limit)</span>
               </li>
               <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
                 <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
@@ -99,23 +137,24 @@ export default function UpgradePage({ user, onBack, onSelectPackage }: UpgradePa
             </ul>
 
             <button disabled style={{ width: '100%', padding: '12px', borderRadius: '10px', border: '1px solid rgba(255,255,255,0.05)', background: 'rgba(255,255,255,0.02)', color: 'var(--text-muted)', fontSize: '14px', fontWeight: 700, cursor: 'not-allowed', textAlign: 'center' }}>
-              Active Plan
+              {currentTier === 'free' ? 'Active Plan' : 'Free Tier'}
             </button>
           </div>
 
-          {/* Card 1: 10 Credits */}
+          {/* Card 1: 10 Messages */}
           <div className="pricing-full-card" onClick={() => onSelectPackage(500)} style={{ display: 'flex', flexDirection: 'column', padding: '32px 24px', background: 'rgba(255,255,255,0.02)', border: '1px solid var(--border-glass)', borderRadius: '16px', textAlign: 'left', transition: 'all 0.3s ease', cursor: 'pointer', position: 'relative' }}>
             <span style={{ fontSize: '12px', color: 'var(--neon-cyan)', fontWeight: 700, letterSpacing: '0.05em', textTransform: 'uppercase' }}>Analyst Pack</span>
-            <h3 style={{ fontSize: '22px', fontWeight: 800, color: 'var(--text-primary)', marginTop: '8px' }}>10 Credits</h3>
+            <h3 style={{ fontSize: '22px', fontWeight: 800, color: 'var(--text-primary)', marginTop: '8px' }}>10 Messages</h3>
             <p style={{ fontSize: '13px', color: 'var(--text-muted)', marginTop: '8px', minHeight: '38px' }}>Great for casual traders looking for reliable market strategies.</p>
             <div style={{ margin: '24px 0' }}>
               <span style={{ fontSize: '36px', fontWeight: 900, color: 'var(--text-primary)' }}>₹500</span>
+              <span style={{ fontSize: '14px', color: 'var(--text-muted)', marginLeft: '4px' }}>one-time</span>
             </div>
             
             <ul style={{ flex: 1, listStyle: 'none', padding: 0, margin: '0 0 32px', display: 'flex', flexDirection: 'column', gap: '14px', fontSize: '13px', color: 'var(--text-secondary)' }}>
               <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
                 <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
-                <span>10 Strategy Generations</span>
+                <span><strong>10 AI Chat Messages</strong> (One-time)</span>
               </li>
               <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
                 <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
@@ -132,24 +171,24 @@ export default function UpgradePage({ user, onBack, onSelectPackage }: UpgradePa
             </ul>
 
             <button className="insight-btn" style={{ width: '100%', padding: '12px', borderRadius: '10px', border: 'none', background: 'var(--neon-cyan)', color: 'var(--bg-black)', fontSize: '14px', fontWeight: 700, cursor: 'pointer', transition: 'all 0.2s ease' }}>
-              Purchase Credits
+              {currentTier === 'analyst' ? 'Extend Balance' : 'Purchase Pack'}
             </button>
           </div>
 
-          {/* Card 2: 50 Credits */}
-          <div className="pricing-full-card hot-deal-card" onClick={() => onSelectPackage(1500)} style={{ display: 'flex', flexDirection: 'column', padding: '32px 24px', background: 'rgba(161, 84, 255, 0.05)', border: '1px solid rgba(161, 84, 255, 0.3)', borderRadius: '16px', textAlign: 'left', transition: 'all 0.3s ease', cursor: 'pointer', position: 'relative', boxShadow: '0 12px 48px rgba(161, 84, 255, 0.15)' }}>
-            <div style={{ position: 'absolute', top: '-12px', right: '20px', background: 'var(--neon-violet)', color: 'var(--text-primary)', fontSize: '11px', fontWeight: 800, padding: '4px 10px', borderRadius: '20px', letterSpacing: '0.05em', textTransform: 'uppercase' }}>Most Popular</div>
+          {/* Card 2: 25 Messages */}
+          <div className="pricing-full-card" onClick={() => onSelectPackage(1500)} style={{ display: 'flex', flexDirection: 'column', padding: '32px 24px', background: 'rgba(255,255,255,0.02)', border: '1px solid var(--border-glass)', borderRadius: '16px', textAlign: 'left', transition: 'all 0.3s ease', cursor: 'pointer', position: 'relative' }}>
             <span style={{ fontSize: '12px', color: 'var(--neon-violet)', fontWeight: 700, letterSpacing: '0.05em', textTransform: 'uppercase' }}>Trader Pack</span>
-            <h3 style={{ fontSize: '22px', fontWeight: 800, color: 'var(--text-primary)', marginTop: '8px' }}>50 Credits</h3>
+            <h3 style={{ fontSize: '22px', fontWeight: 800, color: 'var(--text-primary)', marginTop: '8px' }}>25 Messages</h3>
             <p style={{ fontSize: '13px', color: 'var(--text-muted)', marginTop: '8px', minHeight: '38px' }}>Designed for active traders seeking deep market intelligence.</p>
             <div style={{ margin: '24px 0' }}>
               <span style={{ fontSize: '36px', fontWeight: 900, color: 'var(--text-primary)' }}>₹1,500</span>
+              <span style={{ fontSize: '14px', color: 'var(--text-muted)', marginLeft: '4px' }}>one-time</span>
             </div>
             
             <ul style={{ flex: 1, listStyle: 'none', padding: 0, margin: '0 0 32px', display: 'flex', flexDirection: 'column', gap: '14px', fontSize: '13px', color: 'var(--text-secondary)' }}>
               <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
                 <Check size={16} style={{ color: 'var(--neon-violet)', flexShrink: 0 }} />
-                <span>50 Strategy Generations</span>
+                <span><strong>25 AI Chat Messages</strong> (One-time)</span>
               </li>
               <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
                 <Check size={16} style={{ color: 'var(--neon-violet)', flexShrink: 0 }} />
@@ -166,80 +205,114 @@ export default function UpgradePage({ user, onBack, onSelectPackage }: UpgradePa
             </ul>
 
             <button className="insight-btn" style={{ width: '100%', padding: '12px', borderRadius: '10px', border: 'none', background: 'var(--neon-violet)', color: 'var(--text-primary)', fontSize: '14px', fontWeight: 700, cursor: 'pointer', transition: 'all 0.2s ease', boxShadow: '0 4px 12px rgba(161, 84, 255, 0.3)' }}>
-              Purchase Credits
+              {currentTier === 'trader' ? 'Extend Balance' : 'Purchase Pack'}
             </button>
           </div>
 
-          {/* Card 3: 100 Credits / Lifetime Offer */}
-          {isNewUser ? (
-            /* Special Lifetime Offer (Age <= 3 days) */
-            <div className="pricing-full-card lifetime-deal-card" onClick={() => onSelectPackage(10500)} style={{ display: 'flex', flexDirection: 'column', padding: '32px 24px', background: 'rgba(0, 242, 254, 0.05)', border: '1px solid rgba(0, 242, 254, 0.5)', borderRadius: '16px', textAlign: 'left', transition: 'all 0.3s ease', cursor: 'pointer', position: 'relative', boxShadow: '0 12px 48px rgba(0, 242, 254, 0.2)' }}>
-              <div style={{ position: 'absolute', top: '-12px', right: '20px', background: 'var(--neon-cyan)', color: 'var(--bg-black)', fontSize: '11px', fontWeight: 800, padding: '4px 10px', borderRadius: '20px', letterSpacing: '0.05em', textTransform: 'uppercase' }}>Limited Special Offer</div>
-              <span style={{ fontSize: '12px', color: 'var(--neon-cyan)', fontWeight: 700, letterSpacing: '0.05em', textTransform: 'uppercase' }}>Lifetime Plan</span>
-              <h3 style={{ fontSize: '22px', fontWeight: 800, color: 'var(--text-primary)', marginTop: '8px' }}>Unlimited AI</h3>
-              <p style={{ fontSize: '13px', color: 'var(--text-muted)', marginTop: '8px', minHeight: '38px' }}>Ultimate freedom. Unlimited AI chat generations forever without ever paying again.</p>
-              <div style={{ margin: '24px 0' }}>
-                <span style={{ fontSize: '36px', fontWeight: 900, color: 'var(--text-primary)' }}>₹10,500</span>
-                <span style={{ fontSize: '14px', color: 'var(--text-muted)', marginLeft: '4px' }}>once</span>
+          {/* Card 3: Pro Pack */}
+          <div 
+            className="pricing-full-card hot-deal-card" 
+            onClick={() => onSelectPackage(isNewUser && timeLeft ? 10000 : 15000)} 
+            style={{ 
+              display: 'flex', 
+              flexDirection: 'column', 
+              padding: '32px 24px', 
+              background: 'rgba(0, 242, 254, 0.03)', 
+              border: '1px solid rgba(0, 242, 254, 0.4)', 
+              borderRadius: '16px', 
+              textAlign: 'left', 
+              transition: 'all 0.3s ease', 
+              cursor: 'pointer', 
+              position: 'relative', 
+              boxShadow: '0 12px 48px rgba(0, 242, 254, 0.15)' 
+            }}
+          >
+            {isNewUser && timeLeft && (
+              <div 
+                style={{ 
+                  position: 'absolute', 
+                  top: '-14px', 
+                  right: '16px', 
+                  background: 'linear-gradient(90deg, #00f2fe, #a154ff)', 
+                  color: '#07090e', 
+                  fontSize: '11px', 
+                  fontWeight: 900, 
+                  padding: '4px 12px', 
+                  borderRadius: '20px', 
+                  display: 'flex', 
+                  alignItems: 'center', 
+                  gap: '6px', 
+                  boxShadow: '0 4px 12px rgba(0, 242, 254, 0.4)' 
+                }}
+              >
+                <Clock size={12} />
+                <span>LIMITED OFFER</span>
               </div>
-              
-              <ul style={{ flex: 1, listStyle: 'none', padding: 0, margin: '0 0 32px', display: 'flex', flexDirection: 'column', gap: '14px', fontSize: '13px', color: 'var(--text-secondary)' }}>
-                <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
-                  <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
-                  <strong style={{ color: 'var(--text-primary)' }}>Lifetime Unlimited AI chats</strong>
-                </li>
-                <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
-                  <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
-                  <span>Instant strategy reports</span>
-                </li>
-                <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
-                  <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
-                  <span>Unlimited watchlist & alerts</span>
-                </li>
-                <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
-                  <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
-                  <span>Exclusive developer features</span>
-                </li>
-              </ul>
-
-              <button className="insight-btn" style={{ width: '100%', padding: '12px', borderRadius: '10px', border: 'none', background: 'linear-gradient(90deg, #00f2fe, #a154ff)', color: 'var(--bg-black)', fontSize: '14px', fontWeight: 800, cursor: 'pointer', transition: 'all 0.2s ease', boxShadow: '0 4px 12px rgba(0, 242, 254, 0.4)' }}>
-                Claim Lifetime Deal
-              </button>
+            )}
+            
+            <span style={{ fontSize: '12px', color: 'var(--neon-cyan)', fontWeight: 700, letterSpacing: '0.05em', textTransform: 'uppercase' }}>Pro Pack</span>
+            <h3 style={{ fontSize: '22px', fontWeight: 800, color: 'var(--text-primary)', marginTop: '8px' }}>100 Messages / mo</h3>
+            <p style={{ fontSize: '13px', color: 'var(--text-muted)', marginTop: '8px', minHeight: '38px' }}>For professional quantitative traders requiring maximum insight volume.</p>
+            
+            <div style={{ margin: '24px 0', display: 'flex', alignItems: 'baseline', gap: '8px' }}>
+              {isNewUser && timeLeft ? (
+                <>
+                  <span style={{ fontSize: '36px', fontWeight: 900, color: 'var(--neon-cyan)' }}>₹10,000</span>
+                  <span style={{ fontSize: '16px', color: 'var(--text-muted)', textDecoration: 'line-through' }}>₹15,000</span>
+                </>
+              ) : (
+                <span style={{ fontSize: '36px', fontWeight: 900, color: 'var(--text-primary)' }}>₹15,000</span>
+              )}
+              <span style={{ fontSize: '14px', color: 'var(--text-muted)' }}>/ month</span>
             </div>
-          ) : (
-            /* Standard 100 Credits Card */
-            <div className="pricing-full-card" onClick={() => onSelectPackage(2500)} style={{ display: 'flex', flexDirection: 'column', padding: '32px 24px', background: 'rgba(255,255,255,0.02)', border: '1px solid var(--border-glass)', borderRadius: '16px', textAlign: 'left', transition: 'all 0.3s ease', cursor: 'pointer', position: 'relative' }}>
-              <span style={{ fontSize: '12px', color: 'var(--text-secondary)', fontWeight: 700, letterSpacing: '0.05em', textTransform: 'uppercase' }}>Pro Pack</span>
-              <h3 style={{ fontSize: '22px', fontWeight: 800, color: 'var(--text-primary)', marginTop: '8px' }}>100 Credits</h3>
-              <p style={{ fontSize: '13px', color: 'var(--text-muted)', marginTop: '8px', minHeight: '38px' }}>Best suited for quantitative researchers and serious day-traders.</p>
-              <div style={{ margin: '24px 0' }}>
-                <span style={{ fontSize: '36px', fontWeight: 900, color: 'var(--text-primary)' }}>₹2,500</span>
+
+            {/* Countdown Widget */}
+            {isNewUser && timeLeft && (
+              <div 
+                style={{ 
+                  background: 'rgba(0, 242, 254, 0.05)', 
+                  border: '1px solid rgba(0, 242, 254, 0.15)', 
+                  borderRadius: '8px', 
+                  padding: '8px 12px', 
+                  fontSize: '12px', 
+                  color: 'var(--neon-cyan)', 
+                  fontWeight: 700, 
+                  textAlign: 'center', 
+                  marginBottom: '20px', 
+                  display: 'flex', 
+                  alignItems: 'center', 
+                  justifyContent: 'center', 
+                  gap: '8px' 
+                }}
+              >
+                <span>Closes in:</span>
+                <span style={{ fontFamily: 'monospace', fontSize: '13px' }}>{timeLeft}</span>
               </div>
-              
-              <ul style={{ flex: 1, listStyle: 'none', padding: 0, margin: '0 0 32px', display: 'flex', flexDirection: 'column', gap: '14px', fontSize: '13px', color: 'var(--text-secondary)' }}>
-                <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
-                  <Check size={16} style={{ color: 'var(--text-muted)', flexShrink: 0 }} />
-                  <span>100 Strategy Generations</span>
-                </li>
-                <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
-                  <Check size={16} style={{ color: 'var(--text-muted)', flexShrink: 0 }} />
-                  <span>Dedicated priority agent queue</span>
-                </li>
-                <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
-                  <Check size={16} style={{ color: 'var(--text-muted)', flexShrink: 0 }} />
-                  <span>Unlimited price alerts</span>
-                </li>
-                <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
-                  <Check size={16} style={{ color: 'var(--text-muted)', flexShrink: 0 }} />
-                  <span>Custom indicator weights</span>
-                </li>
-              </ul>
+            )}
+            
+            <ul style={{ flex: 1, listStyle: 'none', padding: 0, margin: '0 0 32px', display: 'flex', flexDirection: 'column', gap: '14px', fontSize: '13px', color: 'var(--text-secondary)' }}>
+              <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
+                <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
+                <span><strong>100 AI Chat Messages</strong> (Resets monthly)</span>
+              </li>
+              <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
+                <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
+                <span>Dedicated priority agent queue</span>
+              </li>
+              <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
+                <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
+                <span>Unlimited price alerts</span>
+              </li>
+              <li style={{ display: 'flex', alignItems: 'center', gap: '10px' }}>
+                <Check size={16} style={{ color: 'var(--neon-cyan)', flexShrink: 0 }} />
+                <span>Saved Strategy History</span>
+              </li>
+            </ul>
 
-              <button className="insight-btn" style={{ width: '100%', padding: '12px', borderRadius: '10px', border: '1px solid rgba(255,255,255,0.1)', background: 'rgba(255,255,255,0.05)', color: 'var(--text-primary)', fontSize: '14px', fontWeight: 700, cursor: 'pointer', transition: 'all 0.2s ease' }}>
-                Purchase Credits
-              </button>
-            </div>
-          )}
+            <button className="insight-btn" style={{ width: '100%', padding: '12px', borderRadius: '10px', border: 'none', background: 'linear-gradient(90deg, #00f2fe, #a154ff)', color: 'var(--bg-black)', fontSize: '14px', fontWeight: 800, cursor: 'pointer', transition: 'all 0.2s ease', boxShadow: '0 4px 12px rgba(0, 242, 254, 0.3)' }}>
+              {currentTier === 'pro' ? 'Active Plan' : 'Go Pro Now'}
+            </button>
+          </div>
 
         </div>
 
@@ -247,9 +320,9 @@ export default function UpgradePage({ user, onBack, onSelectPackage }: UpgradePa
           <p style={{ fontSize: '13px', color: 'var(--text-muted)', margin: 0 }}>
             Test Mode active. Pay securely using Razorpay Test UPI / Cards.
           </p>
-          {isNewUser && (
+          {isNewUser && timeLeft && (
             <p style={{ fontSize: '13px', color: 'var(--neon-cyan)', margin: 0, fontWeight: 700 }} className="offer-countdown">
-              ⚡ Exclusive Lifetime Purchase offer ends in 3 days!
+              ⚡ Exclusive 3-Day Pro Discount Offer active!
             </p>
           )}
         </div>