Upload database.py with huggingface_hub

database.py

@@ -0,0 +1,329 @@
+"""
+Supabase Database Integration for Naveed AI
+Handles all database operations — chat history, analytics, feedback.
+"""
+
+import os
+import uuid
+import time
+from datetime import datetime
+from typing import Optional, List, Dict
+
+# Supabase config
+SUPABASE_URL = os.environ.get(
+    "SUPABASE_URL",
+    "https://kfvpwpjqdvfsmyldaqau.supabase.co"
+)
+SUPABASE_KEY = os.environ.get(
+    "SUPABASE_KEY",
+    "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImtmdnB3cGpxZHZmc215bGRhcWF1Iiwicm9sZSI6ImFub24iLCJpYXQiOjE3NzEzNDgzMTEsImV4cCI6MjA4NjkyNDMxMX0.P9YujsB_WS2d3zVBYD3pY3J8I2RCvWkd4FRqDqrHCaM"
+)
+
+# Global client (lazy init)
+_supabase = None
+_db_available = False
+
+
+def _get_client():
+    """Get or create Supabase client (lazy initialization)."""
+    global _supabase, _db_available
+    if _supabase is not None:
+        return _supabase
+
+    try:
+        from supabase import create_client
+        _supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
+        _db_available = True
+        print("✅ Supabase connected!")
+        return _supabase
+    except ImportError:
+        print("⚠️ supabase package not installed — database disabled")
+        _db_available = False
+        return None
+    except Exception as e:
+        print(f"⚠️ Supabase connection failed: {e} — database disabled")
+        _db_available = False
+        return None
+
+
+def is_available() -> bool:
+    """Check if database is available."""
+    _get_client()
+    return _db_available
+
+
+def generate_session_id() -> str:
+    """Generate a unique session ID for anonymous users."""
+    return str(uuid.uuid4())[:16]
+
+
+# ─────────────────────────────────────────────────────────────
+# CONVERSATIONS & MESSAGES
+# ─────────────────────────────────────────────────────────────
+
+def create_conversation(session_id: str, title: str = "New Chat") -> Optional[str]:
+    """Create a new conversation. Returns conversation ID."""
+    client = _get_client()
+    if not client:
+        return None
+    try:
+        result = client.table("conversations").insert({
+            "session_id": session_id,
+            "title": title[:100],
+        }).execute()
+        if result.data:
+            _increment_analytics("chat")
+            return result.data[0]["id"]
+    except Exception as e:
+        print(f"DB error (create_conversation): {e}")
+    return None
+
+
+def save_message(
+    conversation_id: str,
+    role: str,
+    content: str,
+    thinking_mode: bool = False,
+    tokens_used: int = 0,
+    response_time_ms: int = 0,
+) -> Optional[str]:
+    """Save a message to the database. Returns message ID."""
+    client = _get_client()
+    if not client or not conversation_id:
+        return None
+    try:
+        result = client.table("messages").insert({
+            "conversation_id": conversation_id,
+            "role": role,
+            "content": content[:10000],  # limit content size
+            "thinking_mode": thinking_mode,
+            "tokens_used": tokens_used,
+            "response_time_ms": response_time_ms,
+        }).execute()
+
+        # Update conversation message count
+        client.table("conversations").update({
+            "message_count": client.table("messages")
+            .select("id", count="exact")
+            .eq("conversation_id", conversation_id)
+            .execute().count or 0
+        }).eq("id", conversation_id).execute()
+
+        _increment_analytics("message")
+
+        if result.data:
+            return result.data[0]["id"]
+    except Exception as e:
+        print(f"DB error (save_message): {e}")
+    return None
+
+
+def get_conversation_history(conversation_id: str, limit: int = 50) -> List[Dict]:
+    """Get messages for a conversation."""
+    client = _get_client()
+    if not client or not conversation_id:
+        return []
+    try:
+        result = (
+            client.table("messages")
+            .select("role, content, thinking_mode, created_at")
+            .eq("conversation_id", conversation_id)
+            .order("created_at", desc=False)
+            .limit(limit)
+            .execute()
+        )
+        return result.data or []
+    except Exception as e:
+        print(f"DB error (get_history): {e}")
+        return []
+
+
+def get_recent_conversations(session_id: str, limit: int = 20) -> List[Dict]:
+    """Get recent conversations for a session."""
+    client = _get_client()
+    if not client:
+        return []
+    try:
+        result = (
+            client.table("conversations")
+            .select("id, title, message_count, created_at")
+            .eq("session_id", session_id)
+            .order("created_at", desc=True)
+            .limit(limit)
+            .execute()
+        )
+        return result.data or []
+    except Exception as e:
+        print(f"DB error (get_conversations): {e}")
+        return []
+
+
+# ─────────────────────────────────────────────────────────────
+# IMAGE GENERATION TRACKING
+# ─────────────────────────────────────────────────────────────
+
+def log_image_generation(
+    session_id: str,
+    prompt: str,
+    style: str = "Default",
+    negative_prompt: str = "",
+    success: bool = True,
+    error_message: str = "",
+):
+    """Log an image generation event."""
+    client = _get_client()
+    if not client:
+        return
+    try:
+        client.table("image_generations").insert({
+            "session_id": session_id,
+            "prompt": prompt[:500],
+            "style": style,
+            "negative_prompt": negative_prompt[:500] if negative_prompt else None,
+            "success": success,
+            "error_message": error_message[:500] if error_message else None,
+        }).execute()
+        _increment_analytics("image_gen")
+    except Exception as e:
+        print(f"DB error (log_image_gen): {e}")
+
+
+# ─────────────────────────────────────────────────────────────
+# SPEECH-TO-TEXT TRACKING
+# ─────────────────────────────────────────────────────────────
+
+def log_stt(
+    session_id: str,
+    transcribed_text: str = "",
+    success: bool = True,
+    error_message: str = "",
+):
+    """Log a speech-to-text event."""
+    client = _get_client()
+    if not client:
+        return
+    try:
+        client.table("speech_transcriptions").insert({
+            "session_id": session_id,
+            "transcribed_text": transcribed_text[:5000] if transcribed_text else None,
+            "success": success,
+            "error_message": error_message[:500] if error_message else None,
+        }).execute()
+        _increment_analytics("stt")
+    except Exception as e:
+        print(f"DB error (log_stt): {e}")
+
+
+# ─────────────────────────────────────────────────────────────
+# IMAGE READING / VISION TRACKING
+# ─────────────────────────────────────────────────────────────
+
+def log_vision(
+    session_id: str,
+    question: str = "",
+    analysis_result: str = "",
+    success: bool = True,
+    error_message: str = "",
+):
+    """Log an image reading/vision event."""
+    client = _get_client()
+    if not client:
+        return
+    try:
+        client.table("image_readings").insert({
+            "session_id": session_id,
+            "question": question[:500] if question else None,
+            "analysis_result": analysis_result[:5000] if analysis_result else None,
+            "success": success,
+            "error_message": error_message[:500] if error_message else None,
+        }).execute()
+        _increment_analytics("vision")
+    except Exception as e:
+        print(f"DB error (log_vision): {e}")
+
+
+# ─────────────────────────────────────────────────────────────
+# FEEDBACK
+# ─────────────────────────────────────────────────────────────
+
+def save_feedback(
+    session_id: str,
+    rating: int,
+    comment: str = "",
+    feature: str = "general",
+):
+    """Save user feedback/rating."""
+    client = _get_client()
+    if not client:
+        return
+    try:
+        client.table("feedback").insert({
+            "session_id": session_id,
+            "rating": max(1, min(5, rating)),
+            "comment": comment[:1000] if comment else None,
+            "feature": feature,
+        }).execute()
+    except Exception as e:
+        print(f"DB error (save_feedback): {e}")
+
+
+# ─────────────────────────────────────────────────────────────
+# ANALYTICS
+# ─────────────────────────────────────────────────────���───────
+
+def _increment_analytics(feature: str):
+    """Increment daily analytics counter (internal)."""
+    client = _get_client()
+    if not client:
+        return
+    try:
+        # Use the database function
+        client.rpc("increment_analytics", {"p_feature": feature}).execute()
+    except Exception as e:
+        # Silently fail — analytics are not critical
+        pass
+
+
+def get_analytics(days: int = 7) -> List[Dict]:
+    """Get analytics for the last N days."""
+    client = _get_client()
+    if not client:
+        return []
+    try:
+        result = (
+            client.table("analytics")
+            .select("*")
+            .order("date", desc=True)
+            .limit(days)
+            .execute()
+        )
+        return result.data or []
+    except Exception as e:
+        print(f"DB error (get_analytics): {e}")
+        return []
+
+
+def get_total_stats() -> Dict:
+    """Get all-time total stats."""
+    client = _get_client()
+    if not client:
+        return {"total_chats": 0, "total_messages": 0, "total_image_gens": 0,
+                "total_stt": 0, "total_vision": 0}
+    try:
+        result = client.table("analytics").select(
+            "total_chats, total_messages, total_image_gens, total_stt_uses, total_vision_uses"
+        ).execute()
+
+        stats = {"total_chats": 0, "total_messages": 0, "total_image_gens": 0,
+                 "total_stt": 0, "total_vision": 0}
+        for row in (result.data or []):
+            stats["total_chats"] += row.get("total_chats", 0)
+            stats["total_messages"] += row.get("total_messages", 0)
+            stats["total_image_gens"] += row.get("total_image_gens", 0)
+            stats["total_stt"] += row.get("total_stt_uses", 0)
+            stats["total_vision"] += row.get("total_vision_uses", 0)
+        return stats
+    except Exception as e:
+        print(f"DB error (get_total_stats): {e}")
+        return {"total_chats": 0, "total_messages": 0, "total_image_gens": 0,
+                "total_stt": 0, "total_vision": 0}