fix url

src/lib/db/trainerScanning.ts

@@ -36,7 +36,11 @@ export async function initializeTrainerScanProgress(imagePaths: string[]): Promi
       
       const imageMatch = imageFile.match(/image_(\d+)\.jpg/);
       const imageIndex = imageMatch ? parseInt(imageMatch[1]) : 1;
-      const remoteUrl = `https://huggingface.co/datasets/Fraser/piclets/resolve/main/${imagePath}`;
+      
+      // Build remote URL without trainer_images folder
+      // From: trainer_images/001_Willow_Snap/image_001.jpg
+      // To: https://huggingface.co/datasets/Fraser/piclets/resolve/main/001_Willow_Snap/image_001.jpg
+      const remoteUrl = `https://huggingface.co/datasets/Fraser/piclets/resolve/main/${trainerName}/${imageFile}`;
       
       // Check if this path already exists to avoid duplicates
       const existing = await db.trainerScanProgress.get(imagePath);

src/lib/services/trainerScanService.ts

@@ -51,26 +51,43 @@ export class TrainerScanService {
   }
   
   // Notify all subscribers of state changes
-  private notifyStateChange(state: Partial<TrainerScanState>) {
-    const fullState = { ...this.getCurrentState(), ...state };
+  private async notifyStateChange(state: Partial<TrainerScanState>) {
+    const currentState = await this.getCurrentState();
+    const fullState = { ...currentState, ...state };
     this.stateCallbacks.forEach(callback => callback(fullState));
   }
   
   // Get current scanning state
   private async getCurrentState(): Promise<TrainerScanState> {
-    const stats = await getScanningStats();
-    return {
-      isScanning: this.isScanning,
-      currentImage: null,
-      currentTrainer: null,
-      progress: {
-        total: stats.total,
-        completed: stats.completed,
-        failed: stats.failed,
-        pending: stats.pending
-      },
-      error: null
-    };
+    try {
+      const stats = await getScanningStats();
+      return {
+        isScanning: this.isScanning,
+        currentImage: null,
+        currentTrainer: null,
+        progress: {
+          total: stats?.total || 0,
+          completed: stats?.completed || 0,
+          failed: stats?.failed || 0,
+          pending: stats?.pending || 0
+        },
+        error: null
+      };
+    } catch (error) {
+      console.error('Failed to get current state:', error);
+      return {
+        isScanning: this.isScanning,
+        currentImage: null,
+        currentTrainer: null,
+        progress: {
+          total: 0,
+          completed: 0,
+          failed: 0,
+          pending: 0
+        },
+        error: 'Failed to load progress stats'
+      };
+    }
   }
   
   // Initialize scanning database with image paths from file
@@ -97,7 +114,7 @@ export class TrainerScanService {
       }
       
       await initializeTrainerScanProgress(imagePaths);
-      this.notifyStateChange(await this.getCurrentState());
+      await this.notifyStateChange({});
     } catch (error) {
       console.error('Failed to initialize trainer scan progress:', error);
       throw new Error('Failed to load trainer image paths');
@@ -125,16 +142,16 @@ export class TrainerScanService {
     
     this.isScanning = true;
     this.shouldStop = false;
-    this.notifyStateChange({ isScanning: true, error: null });
+    await this.notifyStateChange({ isScanning: true, error: null });
     
     try {
       await this.processingLoop();
     } catch (error) {
       console.error('Scanning error:', error);
-      this.notifyStateChange({ error: error instanceof Error ? error.message : 'Unknown error' });
+      await this.notifyStateChange({ error: error instanceof Error ? error.message : 'Unknown error' });
     } finally {
       this.isScanning = false;
-      this.notifyStateChange({ isScanning: false, currentImage: null, currentTrainer: null });
+      await this.notifyStateChange({ isScanning: false, currentImage: null, currentTrainer: null });
       
       // Log final summary
       const finalStats = await getScanningStats();
@@ -163,7 +180,7 @@ export class TrainerScanService {
         break;
       }
       
-      this.notifyStateChange({ 
+      await this.notifyStateChange({ 
         currentImage: nextImage.imagePath,
         currentTrainer: typeof nextImage.trainerName === 'string' ? nextImage.trainerName : null
       });
@@ -190,7 +207,7 @@ export class TrainerScanService {
       }
       
       // Update progress
-      this.notifyStateChange(await this.getCurrentState());
+      await this.notifyStateChange({});
     }
   }