107 lines
3.3 KiB
TypeScript
107 lines
3.3 KiB
TypeScript
|
/**
|
|||
|
|
* AI 圖像生成 Composable
|
|||
|
|
* 用於生成人物三視圖和場景圖片
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
import { useAIProvider } from './useAIProvider'
|
|||
|
|
import { getImageGenerationModel } from '~/utils/storage'
|
|||
|
|
import { DEFAULT_IMAGE_GENERATION_MODEL } from '~/utils/clients/gemini-models'
|
|||
|
|
import { getModelProvider } from '~/utils/clients/all-models'
|
|||
|
|
|
|||
|
|
export function useAIImageGeneration() {
|
|||
|
|
const isLoading = ref(false)
|
|||
|
|
const error = ref<string | null>(null)
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 生成圖像
|
|||
|
|
* 注意：目前 Gemini 和 Grok 的圖像生成 API 可能需要不同的實現
|
|||
|
|
* 這裡先提供一個基礎框架，實際 API 調用需要根據各 provider 的文檔調整
|
|||
|
|
*/
|
|||
|
|
async function generateImage(prompt: string): Promise<string> {
|
|||
|
|
isLoading.value = true
|
|||
|
|
error.value = null
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
const modelName = getImageGenerationModel() || DEFAULT_IMAGE_GENERATION_MODEL
|
|||
|
|
const provider = useAIProvider(modelName)
|
|||
|
|
const providerType = getModelProvider(modelName)
|
|||
|
|
|
|||
|
|
// 根據不同的 provider 使用不同的圖像生成方式
|
|||
|
|
// 目前 Gemini 和 Grok 可能沒有直接的圖像生成 API
|
|||
|
|
// 這裡使用文字生成模型來生成圖像描述的 prompt，然後由前端處理
|
|||
|
|
// 實際應用中，可能需要整合專門的圖像生成服務（如 DALL-E, Midjourney 等）
|
|||
|
|
|
|||
|
|
// 暫時返回一個提示，說明需要整合圖像生成服務
|
|||
|
|
// 實際實作時，這裡應該調用圖像生成 API
|
|||
|
|
throw new Error('圖像生成功能需要整合專門的圖像生成服務。目前可以使用文字模型生成圖像描述，然後手動生成圖像。')
|
|||
|
|
} catch (err) {
|
|||
|
|
const errorMessage = err instanceof Error ? err.message : '未知錯誤'
|
|||
|
|
error.value = errorMessage
|
|||
|
|
throw err
|
|||
|
|
} finally {
|
|||
|
|
isLoading.value = false
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 生成人物三視圖
|
|||
|
|
*/
|
|||
|
|
async function generateCharacterView(
|
|||
|
|
characterName: string,
|
|||
|
|
characterDescription: string,
|
|||
|
|
view: 'front' | 'side' | 'back',
|
|||
|
|
style?: string
|
|||
|
|
): Promise<string> {
|
|||
|
|
const viewNames = {
|
|||
|
|
front: '正面',
|
|||
|
|
side: '側面',
|
|||
|
|
back: '背面'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const prompt = `請生成一個${viewNames[view]}視圖的人物圖片，要求：
|
|||
|
|
- 角色名稱：${characterName}
|
|||
|
|
- 角色描述：${characterDescription}
|
|||
|
|
- 視圖角度：${viewNames[view]}
|
|||
|
|
- 背景：純白色
|
|||
|
|
- 風格：${style || '寫實風格'}
|
|||
|
|
- 圖片格式：PNG，透明背景或白色背景
|
|||
|
|
- 解析度：512x512 或更高
|
|||
|
|
|
|||
|
|
請確保圖片清晰，角色特徵明顯。`
|
|||
|
|
|
|||
|
|
// 目前返回提示文字，實際應該返回圖像 URL 或 base64
|
|||
|
|
// 這裡可以整合實際的圖像生成 API
|
|||
|
|
return await generateImage(prompt)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 生成場景圖片
|
|||
|
|
*/
|
|||
|
|
async function generateSceneImage(
|
|||
|
|
sceneName: string,
|
|||
|
|
sceneDescription: string,
|
|||
|
|
style?: string
|
|||
|
|
): Promise<string> {
|
|||
|
|
const prompt = `請生成一個場景圖片，要求：
|
|||
|
|
- 場景名稱：${sceneName}
|
|||
|
|
- 場景描述：${sceneDescription}
|
|||
|
|
- 背景：純白色或透明
|
|||
|
|
- 風格：${style || '寫實風格'}
|
|||
|
|
- 圖片格式：PNG
|
|||
|
|
- 解析度：1024x1024 或更高
|
|||
|
|
|
|||
|
|
請確保場景清晰，細節豐富。`
|
|||
|
|
|
|||
|
|
return await generateImage(prompt)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
generateImage,
|
|||
|
|
generateCharacterView,
|
|||
|
|
generateSceneImage,
|
|||
|
|
isLoading: readonly(isLoading),
|
|||
|
|
error: readonly(error)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|