This commit is contained in:
Jaime Freire 2023-12-02 19:17:51 +01:00
parent 059c99c70f
commit 8eb8b69759
8 changed files with 165 additions and 75 deletions

BIN
app.zip Normal file

Binary file not shown.

View File

@ -1,14 +1,11 @@
// ./app/api/chat/route.js
import OpenAI from 'openai'
import { OpenAIStream, StreamingTextResponse } from 'ai'
const USER_PROMPT = 'Generate code a web page that looks exactly like this'
const USER_PROMPT = 'Generate code for a web page that looks exactly like this'
const SYSTEM_PROMPT= `You are an expert Tailwind developer
const SYSTEM_PROMPT = `You are an expert Tailwind developer
You take screenshots of a reference web page from the user, and then build single page apps
using Tailwind, HTML and JS.
You might also be given a screenshot(The second image) of a web page that you have already built, and asked to
update it to look more like the reference image(The first image).
- Make sure the app looks exactly like the screenshot.
- Pay close attention to background color, text color, font size, font family,
@ -19,23 +16,24 @@ padding, margin, border, etc. Match the colors and sizes exactly.
- For images, use placeholder images from https://placehold.co and include a detailed description of the image in the alt text so that an image generation AI can generate the image later.
In terms of libraries,
- Use this script to include Tailwind: <script src="https://cdn.tailwindcss.com"></script>
- You can use Google Fonts
- Font Awesome for icons: <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.3/css/all.min.css"></link>
Return only the full code in <html></html> tags.
Return first the background hexadecimals, put a ||| separator, and then all the code.
Do not include markdown "\`\`\`" or "\`\`\`html" at the start or end.`
const openai = new OpenAI({
apiKey: process.env.OPENAI_API_KEY,
apiKey: process.env.OPENAI_API_TOKEN,
})
export const runtime = 'edge' //vs cloud. Edge is faster but more expensive
export const runtime = 'edge'
export async function POST(req: Request) {
const { url } = await req.json()
const { url, img } = await req.json()
const imageUrl = url ?? img
const response = await openai.chat.completions.create({
model: 'gpt-4-vision-preview',
stream: true,
@ -54,12 +52,12 @@ export async function POST(req: Request) {
},
{
type: 'image_url',
image_url: url
image_url: imageUrl,
},
],
}
},
],
});
})
const stream = OpenAIStream(response)
return new StreamingTextResponse(stream)

22
app/draganddrop.tsx Normal file
View File

@ -0,0 +1,22 @@
'use client'
import { Dropzone, ExtFile } from '@files-ui/react'
import { useState } from 'react'
export const DragAndDrop = ({ transformImageToCode }: { transformImageToCode: (file: File) => Promise<void> }) => {
const updateFiles = (files: ExtFile[]) => {
const file = files[0].file
if (file != null) transformImageToCode(file)
}
return (
<Dropzone
header={false}
footer={false}
maxFiles={1}
label="Arrastra aquí tu captura de pantalla"
accept="image/*"
onChange={updateFiles}
/>
)
}

View File

@ -3,26 +3,23 @@
import { Button } from '@/components/ui/button'
import { Input } from '@/components/ui/input'
import { Label } from '@/components/ui/label'
import { FormEvent } from 'react'
export const Form = ({transformUrlToCode} : {transformUrlToCode: (url: string) => void}) =>{
const handleSubmit = (evt: FormEvent) => {
const form = evt.currentTarget as HTMLFormElement;
const url = form.elements.namedItem('url') as HTMLInputElement;
transformUrlToCode(url.value);
}
export const Form = ({ transformUrlToCode }: { transformUrlToCode: (url: string) => void }) => {
return (
<form className="flex flex-col gap-5"
<form
className="flex flex-col gap-4"
onSubmit={(evt) => {
evt.preventDefault()
handleSubmit(evt)
const form = evt.currentTarget as HTMLFormElement
const url = form.elements.namedItem('url') as HTMLInputElement
transformUrlToCode(url.value)
}}
>
<Label htmlFor="Introduce to URL de la imagen" />
<Input type="text" name="url" id="url" placeholder="https://tu-screnshot.jpg"/>
<Button>Generar imágen</Button>
<Label htmlFor="url">Introduce tu URL de la imagen:</Label>
<Input name="url" id="url" type="url" placeholder="https://tu-screenshot/image.jpg" />
<Button>Generar código de la imagen</Button>
</form>
)
}

View File

@ -9,11 +9,7 @@ export const metadata: Metadata = {
description: 'Generated by create next app',
}
export default function RootLayout({
children,
}: {
children: React.ReactNode
}) {
export default function RootLayout({ children }: { children: React.ReactNode }) {
return (
<html lang="es">
<body className={`${inter.className} dark`}>{children}</body>

View File

@ -3,6 +3,7 @@
import { Button } from '@/components/ui/button'
import { Form } from './form'
import { useState } from 'react'
import { DragAndDrop } from './draganddrop'
const STEPS = {
INITIAL: 'INITIAL',
@ -11,76 +12,125 @@ const STEPS = {
ERROR: 'ERROR',
}
const toBase64 = (file: File) => {
return new Promise<string>((resolve, reject) => {
const reader = new FileReader()
reader.readAsDataURL(file)
reader.onload = () => resolve(reader.result as string)
reader.onerror = (error) => reject(error)
})
}
async function* streamReader(res: Response) {
const reader = res.body?.getReader()
const decoder = new TextDecoder()
if (reader == null) return
while (true) {
const { done, value } = await reader.read()
const chunk = decoder.decode(value)
yield chunk
if (done) break
}
}
export default function Home() {
const [result, setResult] = useState('')
const [step, setStep] = useState(STEPS.INITIAL)
const transformUrlToCode = async (url: string) => {
const transformToCode = async (body: string) => {
setStep(STEPS.LOADING)
const res = await fetch('/api/generate-code-from-image', {
method: 'POST',
body: JSON.stringify({ url }),
body,
headers: {
'Content-Type': 'application/json',
},
})
if (!res.ok || res.body === null) {
throw new Error('Error al generar el código')
if (!res.ok || res.body == null) {
setStep(STEPS.ERROR)
throw new Error('Error al generar el código')
}
setStep(STEPS.PREVIEW)
const reader = res.body.getReader()
const decoder = new TextDecoder('utf-8')
while (true) {
const { done, value } = await reader.read()
const chunk = decoder.decode(value)
setResult((prevResult) => prevResult + chunk)
console.log(chunk)
if (done) break
// leer el streaming de datos
for await (const chunk of streamReader(res)) {
setResult((prev) => prev + chunk)
}
}
const transformImageToCode = async (file: File) => {
const img = await toBase64(file)
await transformToCode(JSON.stringify({ img }))
}
const transformUrlToCode = async (url: string) => {
await transformToCode(JSON.stringify({ url }))
}
const [background, html = ''] = result.split('|||')
return (
<div className="grid grid-cols-[400px_1fr]">
<aside className="flex flex-col justify-between min-h-scren p-4 bg-gray-900">
<aside className="flex flex-col justify-between min-h-screen p-4 bg-gray-900">
<header className="text-center">
<h1 className="text-3xl font-semibold">IMAGE 2 CODE</h1>
<h1 className="text-3xl font-semibold">Image 2 Code</h1>
<h2 className="text-sm opacity-75">Pasa tus imágenes a código en segundos</h2>
</header>
<section>{/* Aquí irán los filtros... */}</section>
<footer>Desarrollado por rdev</footer>
<footer>Desarrollo por rdev</footer>
</aside>
<main className="bg-gray-950">
<section className="max-w-5xl w-full mx-auto p-10">
{step == STEPS.INITIAL && (
<div>
{step === STEPS.LOADING && (
<div className="flex justify-center items-center">
<div role="status">
<svg
aria-hidden="true"
className="w-8 h-8 text-gray-200 animate-spin dark:text-gray-600 fill-blue-600"
viewBox="0 0 100 101"
fill="none"
xmlns="http://www.w3.org/2000/svg"
>
<path
d="M100 50.5908C100 78.2051 77.6142 100.591 50 100.591C22.3858 100.591 0 78.2051 0 50.5908C0 22.9766 22.3858 0.59082 50 0.59082C77.6142 0.59082 100 22.9766 100 50.5908ZM9.08144 50.5908C9.08144 73.1895 27.4013 91.5094 50 91.5094C72.5987 91.5094 90.9186 73.1895 90.9186 50.5908C90.9186 27.9921 72.5987 9.67226 50 9.67226C27.4013 9.67226 9.08144 27.9921 9.08144 50.5908Z"
fill="currentColor"
/>
<path
d="M93.9676 39.0409C96.393 38.4038 97.8624 35.9116 97.0079 33.5539C95.2932 28.8227 92.871 24.3692 89.8167 20.348C85.8452 15.1192 80.8826 10.7238 75.2124 7.41289C69.5422 4.10194 63.2754 1.94025 56.7698 1.05124C51.7666 0.367541 46.6976 0.446843 41.7345 1.27873C39.2613 1.69328 37.813 4.19778 38.4501 6.62326C39.0873 9.04874 41.5694 10.4717 44.0505 10.1071C47.8511 9.54855 51.7191 9.52689 55.5402 10.0491C60.8642 10.7766 65.9928 12.5457 70.6331 15.2552C75.2735 17.9648 79.3347 21.5619 82.5849 25.841C84.9175 28.9121 86.7997 32.2913 88.1811 35.8758C89.083 38.2158 91.5421 39.6781 93.9676 39.0409Z"
fill="currentFill"
/>
</svg>
<span className="sr-only">Loading...</span>
</div>
</div>
)}
{step === STEPS.INITIAL && (
<div className="flex flex-col gap-4">
<DragAndDrop transformImageToCode={transformImageToCode} />
<Form transformUrlToCode={transformUrlToCode} />
</div>
)}
{step == STEPS.LOADING && (
<div className="flex flex-col items-center justify-center">
<h2 className="text-2xl font-semibold">Generando código...</h2>
<div className="w-24 h-24 border-4 border-gray-900 rounded-full animate-spin"></div>
</div>
)}
{step == STEPS.PREVIEW && (
{step === STEPS.PREVIEW && (
<div className="rounded flex flex-col gap-4">
<iframe srcDoc={result} className="w-full h-full border-4 rounded border-gray-700 aspect-video" />
<pre className='pt-10'>
<code>{result}</code>
<div
className="w-full h-full border-4 rounded border-gray-700 aspect-video"
style={{ backgroundColor: `#${background ? background.trim() : 'fff'}` }}
>
<iframe srcDoc={html} className="w-full h-full" />
</div>
<pre className="pt-10">
<code>{html}</code>
</pre>
</div>
)}
{step == STEPS.ERROR && 'ERROR'}
</section>
</main>
</div>

View File

@ -9,6 +9,7 @@
"lint": "next lint"
},
"dependencies": {
"@files-ui/react": "^1.0.8",
"@radix-ui/react-label": "^2.0.2",
"@radix-ui/react-slot": "^1.0.2",
"ai": "^2.2.27",

26
pnpm-lock.yaml generated
View File

@ -5,6 +5,9 @@ settings:
excludeLinksFromLockfile: false
dependencies:
'@files-ui/react':
specifier: ^1.0.8
version: 1.0.8(@types/react@18.2.40)(react-dom@18.2.0)(react@18.2.0)
'@radix-ui/react-label':
specifier: ^2.0.2
version: 2.0.2(@types/react-dom@18.2.17)(@types/react@18.2.40)(react-dom@18.2.0)(react@18.2.0)
@ -126,6 +129,11 @@ packages:
to-fast-properties: 2.0.0
dev: false
/@dynamicss/dynamicss@2.2.8:
resolution: {integrity: sha512-e6hrGUydr8f+c9E/9fHFSG5LoSLdq/MdZXXfbzEDWIVuzKF2hcdxZE7nHNqUNF2htw1mZ17Pyoshu3A6kFEeFA==}
requiresBuild: true
dev: false
/@eslint-community/eslint-utils@4.4.0(eslint@8.55.0):
resolution: {integrity: sha512-1/sA4dwrzBAyeUoQ6oxahHKmrZvsnLCg4RfxW3ZFGGmQkSNQPFNLV9CUEFQP1x9EYXHTo5p6xdhZM1Ne9p/AfA==}
engines: {node: ^12.22.0 || ^14.17.0 || >=16.0.0}
@ -163,6 +171,24 @@ packages:
engines: {node: ^12.22.0 || ^14.17.0 || >=16.0.0}
dev: true
/@files-ui/core@1.0.4:
resolution: {integrity: sha512-CaSyHSwjeWEkctfALoK4MwAxq3Zr7+Ozyk5AMoB3s1UBIVgh2618wvNuUIHSVhWFKeB28bOoYxCzM2PEXO4QgA==}
dev: false
/@files-ui/react@1.0.8(@types/react@18.2.40)(react-dom@18.2.0)(react@18.2.0):
resolution: {integrity: sha512-yveLiDziZ0WOf1rqRwnQGH6NErvNjkkOURtnK4YyiD2jgTMYlUNhghegciqzHjHd1P5UiD86W+++QYDvtS/rnQ==}
peerDependencies:
'@types/react': ^17.0.0 || ^18.0.0
react: ^17.0.2 || ^18.0.0
react-dom: ^17.0.2 || ^18.0.0
dependencies:
'@dynamicss/dynamicss': 2.2.8
'@files-ui/core': 1.0.4
'@types/react': 18.2.40
react: 18.2.0
react-dom: 18.2.0(react@18.2.0)
dev: false
/@humanwhocodes/config-array@0.11.13:
resolution: {integrity: sha512-JSBDMiDKSzQVngfRjOdFXgFfklaXI4K9nLF49Auh21lmBWRLIK3+xTErTWD4KU54pb6coM6ESE7Awz/FNU3zgQ==}
engines: {node: '>=10.10.0'}