Function: Image Generation History

Author: ShenKSPZ

src/services/database-integration.ts

@@ -3,6 +3,7 @@ import { SettingsService } from './settings-service';
 import { Conversation, Message, ConversationFolder, MessageContent, MessageContentType, FileJsonData } from '../types/chat';
 import { v4 as uuidv4 } from 'uuid';
 import { FileData } from '../types/file';
+import { ImageGenerationResult } from '../types/image';
 
 const SYSTEM_MESSAGE_CONTENT: MessageContent[] = [
     {
@@ -350,6 +351,17 @@ export class DatabaseIntegrationService {
         };
     }
 
+    public async saveImageGenerationResult(imageResult: ImageGenerationResult){
+        try {
+            const resultID = await this.dbService.saveImageGenerationResult(imageResult);
+            return resultID;
+        } catch (error) {
+            console.error('Error getting files:', error);
+            return null;
+        }
+    }
+
+
     public async saveFile(fileData: FileJsonData, arrayBuffer: ArrayBuffer): Promise<string> {
         const fileId = uuidv4();
         const file: FileData = {

src/services/database.ts

@@ -3,12 +3,13 @@ import { Conversation, Message, ConversationFolder } from '../types/chat';
 import { v4 as uuidv4 } from 'uuid';
 import { UserSettings } from '../types/settings';
 import { FileData } from '../types/file';
+import { ImageGenerationResult } from '../types/image';
 
 // database.ts
 export class DatabaseService {
     private db: IDBDatabase | null = null;
     private readonly DB_NAME = 'tensorblock_db';
-    private readonly DB_VERSION = 3; // Increase version to trigger upgrade
+    private readonly DB_VERSION = 4; // Increase version to trigger upgrade
     private readonly ENCRYPTION_KEY = 'your-secure-encryption-key'; // In production, use a secure key management system
 
     private isInitialized: boolean = false;
@@ -70,6 +71,12 @@ export class DatabaseService {
                     chatStore.createIndex('messageId', 'messageId');
                 }
 
+                // Create image generation results store
+                if (!db.objectStoreNames.contains('imageGenerationResults')) {
+                    db.createObjectStore('imageGenerationResults', {
+                        keyPath: 'imageResultId'
+                    });
+                }
                 // Create API settings store
                 if (!db.objectStoreNames.contains('apiSettings')) {
                     db.createObjectStore('apiSettings', {
@@ -547,6 +554,19 @@ export class DatabaseService {
         });
     }
 
+    public async saveImageGenerationResult(imageGenerationResult: ImageGenerationResult): Promise<string> {
+        return new Promise((resolve, reject) => {
+            if (!this.db) throw new Error('Database not initialized');
+
+            const transaction = this.db.transaction('imageGenerationResults', 'readwrite');
+            const store = transaction.objectStore('imageGenerationResults');
+            const request = store.add(imageGenerationResult);
+
+            request.onsuccess = () => resolve(request.result as string);
+            request.onerror = () => reject(request.error);
+        });
+    }
+
     /**
      * Get all files from the database
      * @returns List of files

src/types/chat.ts

@@ -53,16 +53,3 @@ export interface ConversationFolder{
   updatedAt: Date;
   colorFlag: string;
 }
-
-export interface ImageGenerationResult {
-  id: string;
-  prompt: string;
-  negativePrompt: string;
-  seed: string;
-  number: number;
-  status: string;
-  aspectRatio: string;
-  provider: string;
-  model: string;
-  images: MessageContent[];
-}

src/types/image.ts

@@ -0,0 +1,14 @@
+import { MessageContent } from "./chat";
+
+export interface ImageGenerationResult {
+    imageResultId: string;
+    prompt: string;
+    negativePrompt: string;
+    seed: string;
+    number: number;
+    status: string;
+    aspectRatio: string;
+    provider: string;
+    model: string;
+    images: MessageContent[];
+}