Major Project Finalization: Renamed Models, Implemented SQLite DB Auth, User Tracking, and adjusted model restraints per request.

ai_router.py

@@ -4,8 +4,6 @@ from models.gemini_vision import gemini_vision_answer
 from models.groq_vision import groq_vision_answer
 from models.blip_yolo_model import blip_yolo_answer
 from models.hf_boss_api import hf_boss_answer
-from models.florence_fusion_model import florence_answer
-
 try:
     from deep_translator import GoogleTranslator
 except:
@@ -31,8 +29,6 @@ def route_model(model_choice, image, question, lang="en"):
         cap, ans, exp = groq_vision_answer(image, question, lang)
     elif model_choice == "hf_boss":
         cap, ans, exp = hf_boss_answer(image, question, lang)
-    elif model_choice == "florence":
-        cap, ans, exp = florence_answer(image, question, lang)
     else:
         cap, ans, exp = "Unknown", "Invalid", "Invalid"
 

database.py

@@ -0,0 +1,137 @@
+import sqlite3
+import os
+import time
+
+DB_PATH = "vqa.db"
+
+def init_db():
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    
+    # Users table
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS users (
+            username TEXT PRIMARY KEY,
+            password TEXT
+        )
+    ''')
+    
+    # Logs table
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS logs (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            timestamp TEXT,
+            username TEXT,
+            model TEXT,
+            question TEXT
+        )
+    ''')
+
+    # Gemini Usage table
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS gemini_usage (
+            username TEXT,
+            date TEXT,
+            call_count INTEGER,
+            PRIMARY KEY (username, date)
+        )
+    ''')
+
+    # Seed Admin
+    cursor.execute("INSERT OR IGNORE INTO users (username, password) VALUES (?, ?)", ("vignesh", "vignesh"))
+    
+    conn.commit()
+    conn.close()
+
+def login_user(username, password):
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    cursor.execute("SELECT password FROM users WHERE username=?", (username,))
+    row = cursor.fetchone()
+    conn.close()
+    
+    if row and row[0] == password:
+        return True
+    return False
+
+def signup_user(username, password):
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    try:
+        cursor.execute("INSERT INTO users (username, password) VALUES (?, ?)", (username, password))
+        conn.commit()
+        success = True
+    except sqlite3.IntegrityError:
+        success = False
+    conn.close()
+    return success
+
+def log_request_db(username, model, question):
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    from datetime import datetime
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    cursor.execute("INSERT INTO logs (timestamp, username, model, question) VALUES (?, ?, ?, ?)",
+                   (timestamp, username, model, question))
+    conn.commit()
+    conn.close()
+
+def check_gemini_limit(username):
+    if username == "vignesh":
+        return True # Admin bypass
+
+    from datetime import datetime
+    today_str = datetime.now().strftime("%Y-%m-%d")
+    
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    
+    cursor.execute("SELECT call_count FROM gemini_usage WHERE username=? AND date=?", (username, today_str))
+    row = cursor.fetchone()
+    
+    if row is None:
+        count = 0
+    else:
+        count = row[0]
+        
+    conn.close()
+    
+    if count >= 10:
+        return False
+    return True
+
+def increment_gemini_limit(username):
+    if username == "vignesh":
+        return
+
+    from datetime import datetime
+    today_str = datetime.now().strftime("%Y-%m-%d")
+    
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    
+    cursor.execute("SELECT call_count FROM gemini_usage WHERE username=? AND date=?", (username, today_str))
+    row = cursor.fetchone()
+    
+    if row is None:
+        cursor.execute("INSERT INTO gemini_usage (username, date, call_count) VALUES (?, ?, ?)", (username, today_str, 1))
+    else:
+        cursor.execute("UPDATE gemini_usage SET call_count = call_count + 1 WHERE username=? AND date=?", (username, today_str))
+        
+    conn.commit()
+    conn.close()
+
+def get_user_logs(username):
+    conn = sqlite3.connect(DB_PATH)
+    conn.row_factory = sqlite3.Row
+    cursor = conn.cursor()
+    
+    if username == "vignesh":
+        cursor.execute("SELECT * FROM logs ORDER BY id DESC")
+    else:
+        cursor.execute("SELECT * FROM logs WHERE username=? ORDER BY id DESC", (username,))
+        
+    rows = cursor.fetchall()
+    conn.close()
+    
+    return [dict(row) for row in rows]

main.py

@@ -7,14 +7,18 @@ from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
 
 from ai_router import route_model
-from utils.logger import log_request
 from utils.security import check_rate_limit, validate_image_size
 from config import DEVICE, ENABLE_EXTERNAL_AI
+from database import init_db, login_user, signup_user, log_request_db, check_gemini_limit, increment_gemini_limit, get_user_logs
+from fastapi import HTTPException
 
 
 # ================= CREATE APP FIRST =================
 app = FastAPI(title="GenAI VQA System")
 
+@app.on_event("startup")
+async def startup_event():
+    init_db()
 
 # ================= CORS =================
 app.add_middleware(
@@ -38,34 +42,32 @@ async def home(request: Request):
 
 
 # ================= API =================
+@app.post("/api/login")
+async def handle_login(request: Request):
+    data = await request.json()
+    username = data.get("username")
+    password = data.get("password")
+    
+    if login_user(username, password):
+        return {"status": "success", "user": username}
+    return JSONResponse(status_code=401, content={"status": "error", "message": "Invalid password or user does not exist."})
+
+@app.post("/api/signup")
+async def handle_signup(request: Request):
+    data = await request.json()
+    username = data.get("username")
+    password = data.get("password")
+    
+    if not username or not password:
+        return JSONResponse(status_code=400, content={"status": "error", "message": "Username and password required."})
+        
+    if signup_user(username, password):
+        return {"status": "success", "user": username}
+    return JSONResponse(status_code=400, content={"status": "error", "message": "Username already taken."})
+
 @app.get("/api/logs")
 async def get_logs_endpoint(user: str):
-    import os
-    logs = []
-    if os.path.exists("logs/requests.log"):
-        with open("logs/requests.log", "r") as f:
-            for line in f:
-                if " - User: " in line:
-                    try:
-                        parts = line.split(" - INFO - User: ")
-                        timestamp = parts[0]
-                        rest = parts[1].split(" | Model: ")
-                        log_user = rest[0].strip()
-                        model_q = rest[1].split(" | Q: ")
-                        model = model_q[0].strip()
-                        question = model_q[1].strip()
-                        
-                        # Admin sees all, User sees only their own
-                        if user == "vignesh" or log_user == user:
-                            logs.append({
-                                "timestamp": timestamp, 
-                                "user": log_user, 
-                                "model": model, 
-                                "question": question
-                            })
-                    except Exception as e:
-                        pass
-    logs.reverse()
+    logs = get_user_logs(user)
     return {"logs": logs}
 
 
@@ -77,9 +79,19 @@ async def ask_question(
     user: str = Form("guest"),
     lang: str = Form("en")
 ):
-
     check_rate_limit(user)
 
+    if model_choice == "gemini":
+        if not check_gemini_limit(user):
+            return JSONResponse({
+                "device": "Error",
+                "model_used": "gemini",
+                "caption": "Quota Exceeded",
+                "answer": "You have reached your limit of 10 Gemini requests per day.",
+                "explanation": "To prevent abusive cost spikes, users are limited to 10 Gemini requests per day. The Admin has unlimited access.",
+                "external_enabled": True
+            })
+
     image_bytes = await file.read()
     validate_image_size(len(image_bytes))
 
@@ -102,9 +114,10 @@ async def ask_question(
         lang
     )
 
- 
-
-    log_request(user, model_choice, question)
+    log_request_db(user, model_choice, question)
+    
+    if model_choice == "gemini":
+        increment_gemini_limit(user)
 
     return JSONResponse({
         "device": DEVICE,

models/groq_vision.py

@@ -37,15 +37,15 @@ Question:
 CRITICAL RULES FOR YOUR RESPONSE:
 1. You MUST respond strictly in the language code: {lang}. All text in your output must be translated to '{lang}'.
 2. You MUST use EXACTLY the format below with these English labels.
-3. The Caption MUST be exactly 1 line only.
-4. The Final Answer MUST be exactly 1 line only.
-5. The Explanation MUST be strictly 2 to 3 lines maximum. Be very concise and do not exceed this limit!
+3. You MUST provide a detailed descriptive Caption.
+4. You MUST provide a highly accurate Final Answer.
+5. You MUST provide an expansive, detailed Explanation justifying your answer.
 
 Respond strictly following this exact structure without any deviations or markdown blocks.
 
-Caption: <one-line caption>
-Final Answer: <one-line answer>
-Explanation: <2-3 lines max explanation>"""
+Caption: <detailed caption>
+Final Answer: <detailed answer>
+Explanation: <expansive explanation>"""
 
         payload = {
             "model": model_id,

models/hf_boss_api.py

@@ -9,7 +9,8 @@ def hf_boss_answer(image, question, lang="en"):
     if not token:
         return "Setup Required", "HF Token is missing.", "Please set HF_TOKEN."
 
-    model_id = "google/gemma-3-12b-it:featherless"
+    # Using Qwen 2.5 VL as requested
+    model_id = "Qwen/Qwen2.5-VL-7B-Instruct:hyperbolic"
     
     try:
         api_url = "https://router.huggingface.co/v1/chat/completions"
@@ -68,12 +69,12 @@ Explanation: [2-3 sentences]"""
         
         # Fallback system
         if response.status_code != 200:
-            model_id = "zai-org/GLM-4.5V:novita"
+            model_id = "google/gemma-3-12b-it:featherless"
             payload["model"] = model_id
             response = requests.post(api_url, headers=headers, json=payload, timeout=90)
 
         if response.status_code != 200:
-            model_id = "Qwen/Qwen2.5-VL-7B-Instruct:hyperbolic"
+            model_id = "zai-org/GLM-4.5V:novita"
             payload["model"] = model_id
             response = requests.post(api_url, headers=headers, json=payload, timeout=90)
 

static/app.js

@@ -14,8 +14,14 @@ const els = {
     
     // Login
     loginOverlay: document.getElementById('login-overlay'),
-    loginInput: document.getElementById('login-input'),
-    loginBtn: document.getElementById('login-btn'),
+    loginUsername: document.getElementById('login-username'),
+    loginPassword: document.getElementById('login-password'),
+    authBtn: document.getElementById('auth-btn'),
+    googleAuthBtn: document.getElementById('google-auth-btn'),
+    authSwitchLink: document.getElementById('auth-switch-link'),
+    authTitle: document.getElementById('auth-title'),
+    authSubtitle: document.getElementById('auth-subtitle'),
+    authError: document.getElementById('auth-error'),
     mainApp: document.getElementById('main-app'),
     displayUsername: document.getElementById('display-username'),
     logoutBtn: document.getElementById('logout-btn'),
@@ -67,8 +73,10 @@ function init() {
 // --- Event Listeners ---
 function setupEventListeners() {
     // Login System
-    els.loginBtn.addEventListener('click', handleLogin);
-    els.loginInput.addEventListener('keypress', (e) => { if (e.key === 'Enter') handleLogin(); });
+    els.authBtn.addEventListener('click', handleAuth);
+    els.loginPassword.addEventListener('keypress', (e) => { if (e.key === 'Enter') handleAuth(); });
+    els.authSwitchLink.addEventListener('click', toggleAuthMode);
+    els.googleAuthBtn.addEventListener('click', handleGoogleAuth);
     els.logoutBtn.addEventListener('click', handleLogout);
 
     // Tabs
@@ -163,11 +171,77 @@ function setupEventListeners() {
 }
 
 // --- Login System ---
-function handleLogin() {
-    const val = els.loginInput.value.trim().toLowerCase();
-    if (!val) return alert("Please enter a username.");
+let isLoginMode = true;
+
+function toggleAuthMode(e) {
+    e.preventDefault();
+    isLoginMode = !isLoginMode;
+    els.authError.style.display = 'none';
+    if (isLoginMode) {
+        els.authTitle.innerText = "GenAI VQA Portal";
+        els.authSubtitle.innerText = "Sign in to access the system";
+        els.authBtn.innerHTML = 'Login <i class="fa-solid fa-arrow-right"></i>';
+        document.getElementById('auth-switch-text').innerText = "Don't have an account?";
+        els.authSwitchLink.innerText = "Sign Up";
+    } else {
+        els.authTitle.innerText = "Create Account";
+        els.authSubtitle.innerText = "Register to save your logs";
+        els.authBtn.innerHTML = 'Sign Up <i class="fa-solid fa-user-plus"></i>';
+        document.getElementById('auth-switch-text').innerText = "Already have an account?";
+        els.authSwitchLink.innerText = "Login";
+    }
+}
+
+async function handleAuth() {
+    const username = els.loginUsername.value.trim().toLowerCase();
+    const password = els.loginPassword.value.trim();
+    
+    if (!username || !password) {
+        els.authError.innerText = "Username and password required.";
+        els.authError.style.display = 'block';
+        return;
+    }
     
-    appState.username = val;
+    els.authBtn.disabled = true;
+    els.authBtn.innerText = "Processing...";
+    
+    const endpoint = isLoginMode ? '/api/login' : '/api/signup';
+    
+    try {
+        const res = await fetch(endpoint, {
+            method: 'POST',
+            headers: {'Content-Type': 'application/json'},
+            body: JSON.stringify({username, password})
+        });
+        const data = await res.json();
+        
+        if (data.status === 'success') {
+            onLoginSuccess(data.user);
+        } else {
+            els.authError.innerText = data.message;
+            els.authError.style.display = 'block';
+        }
+    } catch(err) {
+        els.authError.innerText = "Server connection error.";
+        els.authError.style.display = 'block';
+    } finally {
+        els.authBtn.disabled = false;
+        els.authBtn.innerHTML = isLoginMode ? 'Login <i class="fa-solid fa-arrow-right"></i>' : 'Sign Up <i class="fa-solid fa-user-plus"></i>';
+    }
+}
+
+function handleGoogleAuth() {
+    // Mock Google Auth logic
+    const mockGoogleId = "google_user_" + Math.floor(Math.random() * 1000);
+    const mockEmail = prompt("Google Auth Simulation:\n\nEnter your Google email address:");
+    if (mockEmail) {
+        const username = mockEmail.split('@')[0];
+        onLoginSuccess(username);
+    }
+}
+
+function onLoginSuccess(user) {
+    appState.username = user;
     els.loginOverlay.classList.add('hidden');
     els.mainApp.classList.remove('hidden');
     els.displayUsername.innerText = appState.username;
@@ -176,13 +250,9 @@ function handleLogin() {
         els.adminTabBtn.style.display = 'flex';
     } else {
         els.adminTabBtn.style.display = 'none';
-        if (document.querySelector('.menu-btn.active').dataset.tab === 'admin') {
-            switchTab('dashboard');
-        }
+        if (document.querySelector('.menu-btn.active').dataset.tab === 'admin') switchTab('dashboard');
     }
     fetchLogs();
-    
-    // Small welcome speech
     speak(`Welcome to GenAI Vision System, ${appState.username}`);
 }
 
@@ -190,7 +260,8 @@ function handleLogout() {
     appState.username = '';
     els.mainApp.classList.add('hidden');
     els.loginOverlay.classList.remove('hidden');
-    els.loginInput.value = '';
+    els.loginUsername.value = '';
+    els.loginPassword.value = '';
     switchTab('dashboard');
 }
 
@@ -412,12 +483,12 @@ function initChart() {
 
 function getChartData() {
     return {
-        labels: ['Local BLIP', 'Local YOLO', 'Gemini AI', 'Groq Vision', 'HF API', 'Florence Fusion'],
+        labels: ['Local BLIP', 'Local YOLO', 'Gemini AI', 'Groq Vision', 'HF API'],
         datasets: [{
             label: 'Queries',
-            data: [appState.stats.local, appState.stats.yolo, appState.stats.gemini, appState.stats.groq, appState.stats.hf_boss, appState.stats.florence],
-            backgroundColor: ['rgba(16, 185, 129, 0.6)', 'rgba(52, 211, 153, 0.6)', 'rgba(99, 102, 241, 0.6)', 'rgba(245, 158, 11, 0.6)', 'rgba(139, 92, 246, 0.6)', 'rgba(236, 72, 153, 0.6)'],
-            borderColor: ['rgba(16, 185, 129, 1)', 'rgba(52, 211, 153, 1)', 'rgba(99, 102, 241, 1)', 'rgba(245, 158, 11, 1)', 'rgba(139, 92, 246, 1)', 'rgba(236, 72, 153, 1)'],
+            data: [appState.stats.local, appState.stats.yolo, appState.stats.gemini, appState.stats.groq, appState.stats.hf_boss],
+            backgroundColor: ['rgba(16, 185, 129, 0.6)', 'rgba(52, 211, 153, 0.6)', 'rgba(99, 102, 241, 0.6)', 'rgba(245, 158, 11, 0.6)', 'rgba(139, 92, 246, 0.6)'],
+            borderColor: ['rgba(16, 185, 129, 1)', 'rgba(52, 211, 153, 1)', 'rgba(99, 102, 241, 1)', 'rgba(245, 158, 11, 1)', 'rgba(139, 92, 246, 1)'],
             borderWidth: 1, borderRadius: 6
         }]
     };

templates/index.html

@@ -12,12 +12,22 @@
 
     <!-- Login Overlay -->
     <div id="login-overlay" class="login-overlay active">
-        <div class="login-box glass-panel fade-in">
+        <div class="login-box glass-panel fade-in" id="auth-box">
             <i class="fa-solid fa-brain" style="font-size: 50px; color: var(--accent); margin-bottom: 24px;"></i>
-            <h2>GenAI VQA Portal</h2>
-            <p style="margin-bottom: 24px; color: var(--text-secondary);">Enter your username to access the system</p>
-            <input type="text" id="login-input" placeholder="Username (e.g. vignesh)" autocomplete="off">
-            <button id="login-btn" class="primary-btn" style="width: 100%;">Enter System <i class="fa-solid fa-arrow-right"></i></button>
+            <h2 id="auth-title">GenAI VQA Portal</h2>
+            <p id="auth-subtitle" style="margin-bottom: 24px; color: var(--text-secondary);">Sign in to access the system</p>
+            
+            <input type="text" id="login-username" placeholder="Username (e.g. vignesh)" autocomplete="off" style="margin-bottom: 12px; width:100%; padding: 12px; border-radius: 8px; border: 1px solid rgba(255,255,255,0.1); background: rgba(0,0,0,0.5); color: white;">
+            <input type="password" id="login-password" placeholder="Password" autocomplete="off" style="margin-bottom: 16px; width:100%; padding: 12px; border-radius: 8px; border: 1px solid rgba(255,255,255,0.1); background: rgba(0,0,0,0.5); color: white;">
+            
+            <button id="auth-btn" class="primary-btn" style="width: 100%; margin-bottom: 12px;">Login <i class="fa-solid fa-arrow-right"></i></button>
+            <button id="google-auth-btn" class="secondary-btn" style="width: 100%; margin-bottom: 16px; background: white; color: #333;"><i class="fa-brands fa-google" style="color: #4285F4;"></i> Sign in with Google</button>
+            
+            <div style="text-align: center; font-size: 14px;">
+                <span id="auth-switch-text" style="color: var(--text-secondary);">Don't have an account?</span>
+                <a href="#" id="auth-switch-link" style="color: var(--accent); font-weight: bold; text-decoration: none;">Sign Up</a>
+            </div>
+            <p id="auth-error" style="color: #ef4444; margin-top: 10px; font-size: 13px; display: none;"></p>
         </div>
     </div>
 
@@ -95,12 +105,11 @@
                                 <div class="control-group" style="flex: 1;">
                                     <label><i class="fa-solid fa-microchip"></i> Select Model</label>
                                     <select id="model-selector">
-                                        <option value="local">BLIP + FLAN (Local)</option>
-                                        <option value="local_yolo">BLIP + FLAN + YOLO (Local Advanced)</option>
-                                        <option value="gemini" id="gemini-option">Gemini 3.1 Flash Vision (High Quota)</option>
-                                        <option value="groq" id="groq-option">Groq Native Vision: Llama 4 Scout</option>
-                                        <option value="hf_boss" id="hf-option">Hugging Face API: Llama 11B Vision (Unlimited Boss)</option>
-                                        <option value="florence" id="florence-option">Florence-2 Fusion (Local Ultimate)</option>
+                                        <option value="local">VQA with LSTM and CNN</option>
+                                        <option value="local_yolo">Hybrid YOLOv8 + Contextual VQA</option>
+                                        <option value="hf_boss" id="hf-option">Web-Scale Autonomous Vision Reasoner (Qwen-VL)</option>
+                                        <option value="groq" id="groq-option">Mixture-of-Experts (MoE) Accelerated Vision</option>
+                                        <option value="gemini" id="gemini-option">Google Gemini (Baseline Benchmark)</option>
                                     </select>
                                 </div>
                                 <div class="control-group" style="flex: 1;">