Update utils/database.py

utils/database.py

@@ -175,7 +175,24 @@ def create_tables(conn: sqlite3.Connection) -> None:
     except Exception as e:
         st.error(f"Unexpected error in create_tables: {e}")
         raise
-
+def search_documents_in_collection(conn: sqlite3.Connection, collection_id: int, query: str) -> List[Dict]:
+    """Search for documents within a collection."""
+    try:
+        with conn_lock:
+            cursor = conn.cursor()
+            cursor.execute('''
+                SELECT d.* 
+                FROM documents d
+                JOIN document_collections dc ON d.id = dc.document_id
+                WHERE dc.collection_id = ?
+                AND (d.name LIKE ? OR d.content LIKE ?)
+            ''', (collection_id, f'%{query}%', f'%{query}%'))
+            
+            return [dict(row) for row in cursor.fetchall()]
+            
+    except sqlite3.Error as e:
+        st.error(f"Error searching documents: {e}")
+        return []
 def force_recreate_collections_tables(conn: sqlite3.Connection) -> bool:
     """Force recreate collections tables if they're missing."""
     try: