Delete utils/collections.py

utils/collections.py

@@ -1,128 +0,0 @@
-import sqlite3
-import streamlit as st
-from typing import Dict, List, Optional, Any
-from datetime import datetime
-from threading import Lock
-
-# Use the same lock from database
-from .database import conn_lock
-
-def create_collection_tables(conn: sqlite3.Connection) -> None:
-    """Create collection-related tables."""
-    try:
-        with conn_lock:
-            # Collections table
-            conn.execute('''
-                CREATE TABLE IF NOT EXISTS collections (
-                    id INTEGER PRIMARY KEY AUTOINCREMENT,
-                    name TEXT NOT NULL UNIQUE,
-                    description TEXT,
-                    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-                )
-            ''')
-            
-            # Document-Collections relationship table
-            conn.execute('''
-                CREATE TABLE IF NOT EXISTS document_collections (
-                    document_id INTEGER,
-                    collection_id INTEGER,
-                    added_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-                    PRIMARY KEY (document_id, collection_id),
-                    FOREIGN KEY (document_id) REFERENCES documents (id) ON DELETE CASCADE,
-                    FOREIGN KEY (collection_id) REFERENCES collections (id) ON DELETE CASCADE
-                )
-            ''')
-            
-            conn.commit()
-    except sqlite3.Error as e:
-        st.error(f"Error creating collection tables: {e}")
-
-def get_collections(conn: sqlite3.Connection) -> List[Dict]:
-    """Get all collections with their document counts."""
-    try:
-        with conn_lock:
-            cursor = conn.cursor()
-            cursor.execute('''
-                SELECT 
-                    c.id,
-                    c.name,
-                    c.description,
-                    c.created_at,
-                    COUNT(DISTINCT dc.document_id) as doc_count
-                FROM collections c
-                LEFT JOIN document_collections dc ON c.id = dc.collection_id
-                GROUP BY c.id
-                ORDER BY c.name
-            ''')
-            
-            return [{
-                'id': row[0],
-                'name': row[1],
-                'description': row[2],
-                'created_at': row[3],
-                'doc_count': row[4]
-            } for row in cursor.fetchall()]
-            
-    except sqlite3.Error as e:
-        st.error(f"Error retrieving collections: {e}")
-        return []
-
-def create_collection(conn: sqlite3.Connection, name: str, description: str = "") -> Optional[int]:
-    """Create a new collection."""
-    try:
-        with conn_lock:
-            cursor = conn.cursor()
-            cursor.execute('''
-                INSERT INTO collections (name, description)
-                VALUES (?, ?)
-            ''', (name, description))
-            conn.commit()
-            return cursor.lastrowid
-            
-    except sqlite3.Error as e:
-        st.error(f"Error creating collection: {e}")
-        return None
-
-def add_document_to_collection(conn: sqlite3.Connection, document_id: int, collection_id: int) -> bool:
-    """Add a document to a collection."""
-    try:
-        with conn_lock:
-            cursor = conn.cursor()
-            cursor.execute('''
-                INSERT OR IGNORE INTO document_collections (document_id, collection_id)
-                VALUES (?, ?)
-            ''', (document_id, collection_id))
-            conn.commit()
-            return True
-            
-    except sqlite3.Error as e:
-        st.error(f"Error adding document to collection: {e}")
-        return False
-
-def get_collection_documents(conn: sqlite3.Connection, collection_id: int) -> List[Dict]:
-    """Get all documents in a specific collection."""
-    try:
-        with conn_lock:
-            cursor = conn.cursor()
-            cursor.execute('''
-                SELECT 
-                    d.id,
-                    d.name,
-                    d.content,
-                    d.upload_date
-                FROM documents d
-                JOIN document_collections dc ON d.id = dc.document_id
-                WHERE dc.collection_id = ?
-                ORDER BY d.upload_date DESC
-            ''', (collection_id,))
-            
-            return [{
-                'id': row[0],
-                'name': row[1],
-                'content': row[2],
-                'upload_date': row[3]
-            } for row in cursor.fetchall()]
-            
-    except sqlite3.Error as e:
-        st.error(f"Error retrieving collection documents: {e}")
-        return []