2.0 beta

Author: Jazee6

components/ChatList.vue

@@ -27,7 +27,7 @@ const md: MarkdownIt = markdownit({
         <USkeleton class="loading-item"/>
       </template>
       <template v-else>
-        <li v-if="i.type==='text'" class="chat-item slide-top prose"
+        <li v-if="i.type==='text'||i.type==='image-prompt'" class="chat-item slide-top prose"
             :class="[i.role==='user'?'send':'reply-text', index+1===history.length && loading ?  'loading':'' ]"
             v-html="i.role === 'user'? i.content: md.render(i.content)"/>
         <li v-else-if="i.type === 'image'">

pages/index.vue

@@ -121,7 +121,7 @@ async function handleSend(input: string, addHistory: boolean) {
     session,
     role: 'user',
     content: input,
-    type: 'text'
+    type: type === 'chat' ? 'text' : 'image-prompt'
   } as HistoryItem
   const id = await DB.history.add(historyItem) as number
   history.value.push({

utils/db.ts

@@ -133,6 +133,11 @@ export const imageGenModels: Model[] = [{
     name: 'stable-diffusion-xl-base-1.0',
     provider: 'workers-ai-image',
     type: 'text-to-image'
+}, {
+    id: '@cf/bytedance/stable-diffusion-xl-lightning',
+    name: 'stable-diffusion-xl-lightning',
+    provider: 'workers-ai-image',
+    type: 'text-to-image'
 }]
 
 export const models: Model[] = [...textGenModels, ...imageGenModels]