Spaces:

pmmdot
/

askbookie

Sleeping

App Files Files Community

pmmdot commited on Dec 30, 2025

Commit

6f26f80

1 Parent(s): 1701946

Add HTML assets and update dependencies

Browse files

- Create new index.html asset file
- Update requirements.txt with new dependencies
- Refactor main.py implementation

Files changed (3) hide show

assets/index.html +333 -0
requirements.txt +1 -0
src/main.py +45 -36

assets/index.html ADDED Viewed

	@@ -0,0 +1,333 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AskBookie API</title>
+    <link href="https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600&display=swap" rel="stylesheet">
+    <style>
+        * {
+            margin: 0;
+            padding: 0;
+            box-sizing: border-box;
+        }
+        body {
+            font-family: 'Inter', sans-serif;
+            background: #000;
+            color: #fafafa;
+            min-height: 100vh;
+            padding: 2rem;
+        }
+        .container {
+            max-width: 1200px;
+            margin: 0 auto;
+        }
+        .header {
+            margin-bottom: 2rem;
+            padding-bottom: 1rem;
+            border-bottom: 1px solid #27272a;
+        }
+        .header h1 {
+            font-size: 1.5rem;
+            font-weight: 600;
+            margin-bottom: 0.25rem;
+        }
+        .header p {
+            font-size: 0.875rem;
+            color: #71717a;
+        }
+        .status {
+            display: inline-block;
+            width: 8px;
+            height: 8px;
+            background: #22c55e;
+            border-radius: 50%;
+            margin-right: 0.5rem;
+        }
+        .grid {
+            display: grid;
+            grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
+            gap: 1rem;
+            margin-bottom: 2rem;
+        }
+        .card {
+            background: #0a0a0a;
+            border: 1px solid #27272a;
+            border-radius: 8px;
+            padding: 1.25rem;
+        }
+        .card:hover {
+            border-color: #3f3f46;
+        }
+        .card-label {
+            font-size: 0.75rem;
+            color: #71717a;
+            text-transform: uppercase;
+            letter-spacing: 0.05em;
+            margin-bottom: 0.5rem;
+        }
+        .card-value {
+            font-size: 2rem;
+            font-weight: 600;
+            margin-bottom: 0.25rem;
+        }
+        .card-desc {
+            font-size: 0.875rem;
+            color: #a1a1aa;
+        }
+        .section {
+            background: #0a0a0a;
+            border: 1px solid #27272a;
+            border-radius: 8px;
+            padding: 1.5rem;
+            margin-bottom: 1rem;
+        }
+        .section-title {
+            font-size: 1rem;
+            font-weight: 600;
+            margin-bottom: 1rem;
+        }
+        .user-card {
+            background: #000;
+            border: 1px solid #27272a;
+            border-radius: 6px;
+            padding: 1rem;
+            margin-bottom: 0.75rem;
+        }
+        .user-card:hover {
+            border-color: #3f3f46;
+        }
+        .user-header {
+            display: flex;
+            align-items: center;
+            gap: 0.5rem;
+            margin-bottom: 1rem;
+            padding-bottom: 0.75rem;
+            border-bottom: 1px solid #27272a;
+        }
+        .user-name {
+            font-size: 0.875rem;
+            font-weight: 500;
+        }
+        .badge {
+            font-size: 0.625rem;
+            padding: 0.125rem 0.5rem;
+            background: #27272a;
+            border-radius: 4px;
+            text-transform: uppercase;
+            letter-spacing: 0.05em;
+            color: #a1a1aa;
+        }
+        .stats-grid {
+            display: grid;
+            grid-template-columns: repeat(auto-fit, minmax(100px, 1fr));
+            gap: 0.75rem;
+        }
+        .stat {
+            text-align: center;
+        }
+        .stat-value {
+            font-size: 1.25rem;
+            font-weight: 600;
+            margin-bottom: 0.125rem;
+        }
+        .stat-label {
+            font-size: 0.75rem;
+            color: #71717a;
+        }
+        .btn {
+            position: fixed;
+            bottom: 3rem;
+            right: 2rem;
+            background: #fafafa;
+            color: #000;
+            border: none;
+            padding: 0.75rem 1.5rem;
+            border-radius: 6px;
+            font-size: 0.875rem;
+            font-weight: 500;
+            cursor: pointer;
+            font-family: 'Inter', sans-serif;
+        }
+        .btn:hover {
+            background: #e5e5e5;
+        }
+        .btn:active {
+            background: #d4d4d4;
+        }
+        .footer {
+            text-align: center;
+            font-size: 0.75rem;
+            color: #52525b;
+            margin-top: 2rem;
+        }
+        @media (max-width: 768px) {
+            body {
+                padding: 1rem;
+            }
+            .grid {
+                grid-template-columns: 1fr;
+            }
+        }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <div class="header">
+            <h1>AskBookie API</h1>
+            <p><span class="status"></span>Live Dashboard</p>
+        </div>
+        <div class="grid">
+            <div class="card">
+                <div class="card-label">Uptime</div>
+                <div class="card-value" id="uptime">--</div>
+                <div class="card-desc">hours</div>
+            </div>
+            <div class="card">
+                <div class="card-label">API Calls</div>
+                <div class="card-value" id="total-calls">--</div>
+                <div class="card-desc">total</div>
+            </div>
+            <div class="card">
+                <div class="card-label">Questions</div>
+                <div class="card-value" id="total-questions">--</div>
+                <div class="card-desc">asked</div>
+            </div>
+            <div class="card">
+                <div class="card-label">Active Jobs</div>
+                <div class="card-value" id="active-jobs">--</div>
+                <div class="card-desc">processing</div>
+            </div>
+            <div class="card">
+                <div class="card-label">Memory</div>
+                <div class="card-value" id="memory">--</div>
+                <div class="card-desc">MB</div>
+            </div>
+            <div class="card">
+                <div class="card-label">CPU</div>
+                <div class="card-value" id="cpu">--</div>
+                <div class="card-desc">percent</div>
+            </div>
+        </div>
+        <div class="section">
+            <div class="section-title">Per-User Analytics</div>
+            <div id="user-stats"></div>
+        </div>
+    </div>
+    <button class="btn" onclick="loadMetrics()">
+        <span id="refresh-text">Refresh</span>
+    </button>
+    <div class="footer" id="last-update"></div>
+    <script>
+        async function loadMetrics() {
+            const refreshText = document.getElementById('refresh-text');
+            const originalText = refreshText.textContent;
+            refreshText.textContent = 'Loading...';
+            try {
+                const response = await fetch('/health');
+                const health = await response.json();
+                document.getElementById('uptime').textContent = (health.uptime_hours || 0).toFixed(2);
+                document.getElementById('active-jobs').textContent = health.active_jobs || 0;
+                document.getElementById('total-calls').textContent = health.total_api_calls || 0;
+                document.getElementById('total-questions').textContent = health.total_questions || 0;
+                document.getElementById('memory').textContent = (health.memory_mb || 0).toFixed(0);
+                document.getElementById('cpu').textContent = (health.cpu_percent || 0).toFixed(1);
+                renderUserStats(health.per_user || {});
+                document.getElementById('last-update').textContent =
+                    `Updated ${new Date().toLocaleTimeString()}`;
+            } catch (error) {
+                console.error('Failed to load metrics:', error);
+            } finally {
+                refreshText.textContent = originalText;
+            }
+        }
+        function renderUserStats(users) {
+            const container = document.getElementById('user-stats');
+            container.innerHTML = '';
+            for (const [userName, stats] of Object.entries(users)) {
+                const userCard = document.createElement('div');
+                userCard.className = 'user-card';
+                userCard.innerHTML = `
+                    <div class="user-header">
+                        <span class="user-name">${userName}</span>
+                        <span class="badge">${stats.role || 'user'}</span>
+                    </div>
+                    <div class="stats-grid">
+                        <div class="stat">
+                            <div class="stat-value">${stats.api_calls || 0}</div>
+                            <div class="stat-label">calls</div>
+                        </div>
+                        <div class="stat">
+                            <div class="stat-value">${stats.questions_asked || 0}</div>
+                            <div class="stat-label">questions</div>
+                        </div>
+                        <div class="stat">
+                            <div class="stat-value">${stats.uploads_attempted || 0}</div>
+                            <div class="stat-label">uploads</div>
+                        </div>
+                        <div class="stat">
+                            <div class="stat-value">${stats.success_rate || 100}%</div>
+                            <div class="stat-label">success</div>
+                        </div>
+                        <div class="stat">
+                            <div class="stat-value">${stats.average_latency_seconds || 0}s</div>
+                            <div class="stat-label">latency</div>
+                        </div>
+                    </div>
+                `;
+                container.appendChild(userCard);
+            }
+        }
+        setInterval(loadMetrics, 5000);
+        loadMetrics();
+    </script>
+</body>
+</html>

requirements.txt CHANGED Viewed

@@ -8,6 +8,7 @@ langchain-qdrant>=1.1.0
 psutil>=7.2.1
 pydantic>=2.12.5
 pypdf>=6.5.0
 python-multipart>=0.0.21
 qdrant-client>=1.16.2
 sentence-transformers>=5.2.0

 psutil>=7.2.1
 pydantic>=2.12.5
 pypdf>=6.5.0
+python-dotenv>=1.0.0
 python-multipart>=0.0.21
 qdrant-client>=1.16.2
 sentence-transformers>=5.2.0

src/main.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from fastapi import FastAPI, HTTPException, UploadFile, File, BackgroundTasks, Form, Depends, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
-from fastapi.responses import JSONResponse
 from pydantic import BaseModel, Field, field_validator
 from typing import Optional, List, Dict
 import os
@@ -19,6 +19,9 @@ import asyncio
 import hashlib
 from enum import Enum
 import logging
 from rag import RAGService, process_pdf
@@ -29,9 +32,9 @@ ALLOWED_ORIGINS = [
 API_KEYS = {
     os.getenv("ADMIN_API_KEY", "your-admin-secret-key"): {"role": "admin", "name": "admin"},
-    os.getenv("USER_API_KEY_1", "your-user-secret-key-1"): {"role": "user", "name": "user_1"},
-    os.getenv("USER_API_KEY_2", "your-user-secret-key-2"): {"role": "user", "name": "user_2"},
-    os.getenv("USER_API_KEY_3", "your-user-secret-key-3"): {"role": "user", "name": "user_3"},
 }
 MAX_FILE_SIZE = 10 * 1024 * 1024
@@ -355,16 +358,51 @@ async def process_with_retry(
             except Exception as e:
                 logger.error(f"Failed to delete temp file {file_path}: {e}")
-@app.get("/", tags=["Health"])
 async def health_check():
-    """Health check with system status"""
     return {
         "status": "healthy",
         "service": "AskBookie API",
         "version": "2.0.0",
         "rag_initialized": rag_service is not None,
         "active_jobs": job_manager.active_jobs,
-        "timestamp": datetime.now().isoformat()
     }
 @app.post("/embed", tags=["Documents"], status_code=202)
@@ -521,35 +559,6 @@ async def list_jobs(
         "jobs": dict(sorted_jobs[:limit])
     }
-@app.get("/metrics", tags=["Monitoring"])
-async def get_metrics(
-    user: dict = Depends(verify_api_key)
-):
-    process = psutil.Process(os.getpid())
-    memory_mb = process.memory_info().rss / 1024 / 1024
-    cpu_percent = process.cpu_percent(interval=0.1)
-    if user["role"] == "admin":
-        api_metrics = metrics.get_all_stats()
-    else:
-        api_metrics = {
-            "user": user["name"],
-            "stats": metrics.get_key_stats(user["key"])
-        }
-    return {
-        "api": api_metrics,
-        "system": {
-            "memory_usage_mb": round(memory_mb, 2),
-            "cpu_percent": round(cpu_percent, 2)
-        },
-        "jobs": {
-            "active": job_manager.active_jobs,
-            "total": len(job_manager.jobs),
-            "max_concurrent": MAX_CONCURRENT_JOBS
-        }
-    }
 @app.delete("/jobs/{job_id}", tags=["Jobs"])
 async def delete_job(
     job_id: str,

 from fastapi import FastAPI, HTTPException, UploadFile, File, BackgroundTasks, Form, Depends, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
+from fastapi.responses import JSONResponse, HTMLResponse
 from pydantic import BaseModel, Field, field_validator
 from typing import Optional, List, Dict
 import os
 import hashlib
 from enum import Enum
 import logging
+from dotenv import load_dotenv
+load_dotenv()
 from rag import RAGService, process_pdf
 API_KEYS = {
     os.getenv("ADMIN_API_KEY", "your-admin-secret-key"): {"role": "admin", "name": "admin"},
+    os.getenv("USER_API_KEY_1", "your-user-secret-key-1"): {"role": "user", "name": "vercel"},
+    os.getenv("USER_API_KEY_2", "your-user-secret-key-2"): {"role": "user", "name": "dotpmm"},
+    os.getenv("USER_API_KEY_3", "your-user-secret-key-3"): {"role": "user", "name": "beta"},
 }
 MAX_FILE_SIZE = 10 * 1024 * 1024
             except Exception as e:
                 logger.error(f"Failed to delete temp file {file_path}: {e}")
+@app.get("/", tags=["Dashboard"])
+async def dashboard():
+    html_path = os.path.join(os.path.dirname(__file__), "..", "assets", "index.html")
+    with open(html_path, "r", encoding="utf-8") as f:
+        html_content = f.read()
+    return HTMLResponse(content=html_content)
+@app.get("/health", tags=["Health"])
 async def health_check():
+    process = psutil.Process(os.getpid())
+    memory_mb = process.memory_info().rss / 1024 / 1024
+    cpu_percent = process.cpu_percent(interval=0.1) / psutil.cpu_count()
+    uptime_seconds = (datetime.now() - metrics.start_time).total_seconds()
+    all_stats = metrics.get_all_stats()
+    total_calls = sum(user_stats.get('api_calls', 0) for user_stats in all_stats['per_key'].values())
+    total_questions = sum(user_stats.get('questions_asked', 0) for user_stats in all_stats['per_key'].values())
+    per_user = {}
+    for user_name, user_stats in all_stats['per_key'].items():
+        role = 'user'
+        for key, key_info in API_KEYS.items():
+            if key_info['name'] == user_name:
+                role = key_info['role']
+                break
+        per_user[user_name] = {
+            **user_stats,
+            'role': role
+        }
     return {
         "status": "healthy",
         "service": "AskBookie API",
         "version": "2.0.0",
         "rag_initialized": rag_service is not None,
         "active_jobs": job_manager.active_jobs,
+        "timestamp": datetime.now().isoformat(),
+        "uptime_hours": round(uptime_seconds / 3600, 2),
+        "total_api_calls": total_calls,
+        "total_questions": total_questions,
+        "memory_mb": round(memory_mb, 2),
+        "cpu_percent": round(cpu_percent, 2),
+        "per_user": per_user
     }
 @app.post("/embed", tags=["Documents"], status_code=202)
         "jobs": dict(sorted_jobs[:limit])
     }
 @app.delete("/jobs/{job_id}", tags=["Jobs"])
 async def delete_job(
     job_id: str,