view data fixes

apps/website/lib/view-models/AvatarGenerationViewModel.ts

@@ -1,18 +1,22 @@
-import { RequestAvatarGenerationOutputDTO } from '@/lib/types/generated/RequestAvatarGenerationOutputDTO';
+import { ViewModel } from "../contracts/view-models/ViewModel";
+import { AvatarGenerationViewData } from "../view-data/AvatarGenerationViewData";
 
 /**
  * AvatarGenerationViewModel
  *
- * View model for avatar generation process
+ * View model for avatar generation process.
+ *
+ * Accepts AvatarGenerationViewData as input and produces UI-ready data.
  */
-export class AvatarGenerationViewModel {
+export class AvatarGenerationViewModel extends ViewModel {
   readonly success: boolean;
   readonly avatarUrls: string[];
   readonly errorMessage?: string;
 
-  constructor(dto: RequestAvatarGenerationOutputDTO) {
-    this.success = dto.success;
-    this.avatarUrls = dto.avatarUrls || [];
-    this.errorMessage = dto.errorMessage;
+  constructor(viewData: AvatarGenerationViewData) {
+    super();
+    this.success = viewData.success;
+    this.avatarUrls = viewData.avatarUrls;
+    this.errorMessage = viewData.errorMessage;
   }
 }