superbrain-server 1.0.2-beta.0

package/payload/stop-docker.sh

@@ -0,0 +1,16 @@
+#!/bin/bash
+
+# SuperBrain Docker Stop Script
+
+echo "🛑 Stopping SuperBrain..."
+
+# Use docker compose if available, otherwise docker-compose
+if command -v docker compose &> /dev/null; then
+    DOCKER_COMPOSE="docker compose"
+else
+    DOCKER_COMPOSE="docker-compose"
+fi
+
+$DOCKER_COMPOSE down
+
+echo "✅ SuperBrain stopped."

package/payload/utils/db_stats.py

@@ -0,0 +1,108 @@
+#!/usr/bin/env python3
+"""
+SQLite Database Statistics
+Shows storage usage, document count, and collection info
+"""
+
+import sys
+from pathlib import Path
+
+# Ensure backend root is in sys.path (needed when run directly)
+sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
+
+from core.database import get_db
+
+
+def format_bytes(bytes_size):
+    """Convert bytes to human-readable format"""
+    for unit in ['B', 'KB', 'MB', 'GB']:
+        if bytes_size < 1024.0:
+            return f"{bytes_size:.2f} {unit}"
+        bytes_size /= 1024.0
+    return f"{bytes_size:.2f} TB"
+
+
+def get_database_stats():
+    """Get comprehensive database statistics"""
+
+    print("=" * 80)
+    print("📊 SUPERBRAIN DATABASE STATISTICS (SQLite)")
+    print("=" * 80)
+    print()
+
+    db = get_db()
+
+    if not db.is_connected():
+        print("❌ Not connected to database")
+        return
+
+    try:
+        stats = db.get_stats()
+        doc_count = stats['document_count']
+        storage_mb = stats['storage_mb']
+        category_counts = stats['categories']
+        storage_bytes = storage_mb * 1024 * 1024
+
+        # ── Storage ──────────────────────────────────────────────────────
+        print("💾 STORAGE USAGE")
+        print("-" * 80)
+        print(f"Database file:       {db.db_path}")
+        print(f"File size:           {format_bytes(storage_bytes)}")
+        print(f"Storage (MB):        {storage_mb} MB")
+        print()
+
+        # ── Documents ────────────────────────────────────────────────────
+        print("📄 DOCUMENT INFORMATION")
+        print("-" * 80)
+        print(f"Total Posts:         {doc_count}")
+        if doc_count > 0 and storage_bytes > 0:
+            print(f"Average Post Size:   {format_bytes(storage_bytes / doc_count)}")
+        print()
+
+        # ── Schema ───────────────────────────────────────────────────────
+        print("📋 COLUMNS IN analyses TABLE")
+        print("-" * 80)
+        columns = [
+            "shortcode", "url", "username", "analyzed_at", "updated_at",
+            "post_date", "likes", "title", "summary", "tags",
+            "music", "category", "visual_analysis", "audio_transcription", "text_analysis"
+        ]
+        for i, col in enumerate(columns, 1):
+            print(f"{i:2}. {col}")
+        print()
+
+        # ── Categories ───────────────────────────────────────────────────
+        print("📂 CATEGORY BREAKDOWN")
+        print("-" * 80)
+        if category_counts:
+            for name, count in sorted(category_counts.items(), key=lambda x: -x[1]):
+                print(f"{name:<20} {count:>5} posts")
+        else:
+            print("No posts yet")
+        print()
+
+        # ── Recent ───────────────────────────────────────────────────────
+        print("📅 RECENT POSTS (Last 5)")
+        print("-" * 80)
+        recent = db.get_recent(5)
+        if recent:
+            for i, doc in enumerate(recent, 1):
+                title = (doc.get('title') or 'No title')[:50]
+                date = doc.get('analyzed_at', 'Unknown')
+                category = doc.get('category', 'N/A')
+                print(f"{i}. [{category}] {title}")
+                print(f"   Analyzed: {date}")
+        else:
+            print("No posts yet")
+
+        print()
+        print("=" * 80)
+
+    except Exception as e:
+        print(f"❌ Error getting stats: {e}")
+        import traceback
+        traceback.print_exc()
+
+
+if __name__ == "__main__":
+    get_database_stats()

package/payload/utils/manage_token.py

@@ -0,0 +1,91 @@
+"""
+SuperBrain API Token Manager
+Manage your API authentication token
+"""
+import sys
+import os
+from pathlib import Path
+import secrets
+import string
+
+TOKEN_FILE = Path(__file__).parent.parent / "token.txt"
+
+def generate_token(length=8):
+    """Generate an 8-character alphanumeric Access Token"""
+    alphabet = string.ascii_uppercase + string.digits
+    return ''.join(secrets.choice(alphabet) for _ in range(length))
+
+
+def is_valid_token(token: str) -> bool:
+    """Validate token format: exactly 8 alphanumeric characters."""
+    return len(token) == 8 and token.isalnum()
+
+def load_token():
+    """Load existing token"""
+    if TOKEN_FILE.exists():
+        with open(TOKEN_FILE, 'r') as f:
+            return f.read().strip()
+    return None
+
+def save_token(token):
+    """Save token to file"""
+    with open(TOKEN_FILE, 'w') as f:
+        f.write(token)
+    print(f"\n✅ Token saved to: {TOKEN_FILE}")
+
+def main():
+    print("\n" + "="*80)
+    print("🔐 SuperBrain Access Token Manager")
+    print("="*80)
+    
+    current_token = load_token()
+    
+    if current_token:
+        print(f"\n📋 Current Token: {current_token}")
+    else:
+        print("\n⚠️  No token found!")
+    
+    print("\nOptions:")
+    print("  1. Generate new random token")
+    print("  2. Set custom token")
+    print("  3. View current token")
+    print("  4. Exit")
+    
+    choice = input("\nEnter choice (1-4): ").strip()
+    
+    if choice == "1":
+        new_token = generate_token()
+        save_token(new_token)
+        print(f"\n🔑 New Token: {new_token}")
+        print("\n💡 Use this in API requests:")
+        print(f"   Header: X-API-Key: {new_token}")
+        print("\n⚠️  Restart the API server for changes to take effect!")
+        
+    elif choice == "2":
+        custom_token = input("\nEnter custom 8-character token: ").strip().upper()
+        if not is_valid_token(custom_token):
+            print("\n❌ Invalid token! Use exactly 8 alphanumeric characters.")
+            return
+        save_token(custom_token)
+        print(f"\n🔑 Token set to: {custom_token}")
+        print("\n⚠️  Restart the API server for changes to take effect!")
+        
+    elif choice == "3":
+        if current_token:
+            print(f"\n🔑 Current Token: {current_token}")
+            print("\n💡 Use in API requests:")
+            print(f"   Header: X-API-Key: {current_token}")
+        else:
+            print("\n⚠️  No token found!")
+            
+    elif choice == "4":
+        print("\n👋 Goodbye!")
+        return
+        
+    else:
+        print("\n❌ Invalid choice!")
+    
+    print("\n" + "="*80 + "\n")
+
+if __name__ == "__main__":
+    main()