Hugging Face's logo

Spaces:
KazukiWatanabe
/
FE_Dev
Sleeping

App Files Community
FE_Dev
File size: 4,448 Bytes
68f7925
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
 
import { parseApiResponse } from '@/api-client/utils/parseApiResponse';
import { rpcClient } from '@/api-client/utils/rpc-client';
import type { SwotData, SwotDataWithoutSummary } from '@/schema/pox';
import {
  generateContentsHtmlResponseSchema,
  generateFvHtmlResponseSchema,
  type ContentSection,
  type FvData,
  type GenerateContentsHtmlResponse,
  type GenerateFvHtmlResponse,
} from '@/schema/proposal';
import type { ThemeExtractionResult } from '@/schema/theme-extraction';

/**
 * FV提案のHTML生成
 */
export async function generateFvHtml(params: {
  tabName: string;
  fvData: FvData;
  provider?: 'openai' | 'gemini' | 'claude';
  referenceUrl?: string;
  screenshot?: string;
  dummyMode?: boolean;
  swotData?: SwotData | SwotDataWithoutSummary;
  userEmail?: string;
  sourcePage?: string;
  forceFallbackMode?: boolean;
  themeData?: ThemeExtractionResult | null;
}): Promise<GenerateFvHtmlResponse> {
  // テーマデータがある場合はログ出力のみ(将来のAPI対応用)
  if (params.themeData) {
    console.log('[FV HTML] テーマデータが適用されます:', params.themeData);
  }

  const response = await rpcClient.proposal['html-preview']['generate-fv-html'].$post({
    json: {
      tabName: params.tabName,
      fvData: params.fvData,
      provider: params.provider,
      referenceUrl: params.referenceUrl,
      screenshotUrl: params.screenshot, // パラメータ名をscreenshotUrlに変換
      dummyMode: params.dummyMode,
      swotData: params.swotData,
      userEmail: params.userEmail,
      sourcePage: params.sourcePage,
      forceFallbackMode: params.forceFallbackMode,
      themeData: params.themeData ?? undefined, // nullをundefinedに変換
    },
  });
  return parseApiResponse(response, generateFvHtmlResponseSchema);
}

/**
 * コンテンツ提案のHTML生成
 */
export async function generateContentsHtml(params: {
  tabName: string;
  cnData: ContentSection[];
  provider?: 'openai' | 'gemini' | 'claude';
  referenceUrl?: string;
  screenshot?: string;
  dummyMode?: boolean;
  userEmail?: string;
  themeData?: ThemeExtractionResult | null;
}): Promise<GenerateContentsHtmlResponse> {
  // テーマデータがある場合はログ出力のみ（将来のAPI対応用）
  if (params.themeData) {
    console.log('[CONTENTS HTML] テーマデータが適用されます:', params.themeData);
  }

  const response = await rpcClient.proposal['html-preview']['generate-contents-html'].$post({
    json: {
      tabName: params.tabName,
      cnData: params.cnData,
      provider: params.provider,
      referenceUrl: params.referenceUrl,
      screenshotUrl: params.screenshot, // パラメータ名をscreenshotUrlに変換
      dummyMode: params.dummyMode,
      userEmail: params.userEmail,
      themeData: params.themeData ?? undefined, // nullをundefinedに変換
    },
  });
  return parseApiResponse(response, generateContentsHtmlResponseSchema);
}

/**
 * 画像生成を開始
 */
export async function startImageGeneration(
  html: string,
  tabName: string,
  cnData: ContentSection[],
  provider: 'openai' | 'claude' = 'claude',
): Promise<{
  batchId: string;
  estimatedCompletionTime: string;
  message: string;
  status: string;
}> {
  const response = await fetch('/api/rpc/proposal/start-contents-image-generation', {
    method: 'POST',
    headers: {
      'Content-Type': 'application/json',
    },
    body: JSON.stringify({
      html,
      tabName,
      cnData,
      provider,
    }),
  });

  if (!response.ok) {
    throw new Error(`画像生成開始に失敗しました: ${response.statusText}`);
  }

  return response.json();
}

/**
 * 画像生成状況を確認
 */
export async function getImageGenerationStatus(batchId: string): Promise<{
  batchId: string;
  status: 'pending' | 'processing' | 'completed' | 'failed';
  jobs: Array<{
    jobId: string;
    imageId: string;
    status: 'pending' | 'generating' | 'completed' | 'failed';
    generatedImageUrl?: string;
    errorMessage?: string;
  }>;
  createdAt: string;
  completedAt?: string;
} | null> {
  const response = await fetch(`/api/rpc/proposal/image-generation-status/${batchId}`);

  if (!response.ok) {
    if (response.status === 404) {
      // バッチIDが存在しない場合は null を返す
      return null;
    }
    throw new Error(`ステータス確認に失敗しました: ${response.statusText}`);
  }

  return response.json();
}