优化界面

src/services/geminiService.ts

@@ -20,13 +20,19 @@ export interface EditRequest {
   seed?: number;
 }
 
+export interface UsageMetadata {
+  totalTokenCount?: number;
+  promptTokenCount?: number;
+  candidatesTokenCount?: number;
+}
+
 export interface SegmentationRequest {
   image: string; // base64
   query: string; // "像素(x,y)处的对象" 或 "红色汽车"
 }
 
 export class GeminiService {
-  async generateImage(request: GenerationRequest): Promise<string[]> {
+  async generateImage(request: GenerationRequest): Promise<{images: string[], usageMetadata?: any}> {
     try {
       const contents: any[] = [{ text: request.prompt }];
       
@@ -67,7 +73,10 @@ export class GeminiService {
         }
       }
 
-      return images;
+      // 获取usageMetadata（如果存在）
+      const usageMetadata = response.usageMetadata;
+
+      return { images, usageMetadata };
     } catch (error) {
       console.error('生成图像时出错:', error);
       if (error instanceof Error && error.message) {
@@ -77,7 +86,7 @@ export class GeminiService {
     }
   }
 
-  async editImage(request: EditRequest): Promise<string[]> {
+  async editImage(request: EditRequest): Promise<{images: string[], usageMetadata?: any}> {
     try {
       const contents = [
         { text: this.buildEditPrompt(request) },
@@ -135,7 +144,10 @@ export class GeminiService {
         }
       }
 
-      return images;
+      // 获取usageMetadata（如果存在）
+      const usageMetadata = response.usageMetadata;
+
+      return { images, usageMetadata };
     } catch (error) {
       console.error('编辑图像时出错:', error);
       if (error instanceof Error && error.message) {