Create code/ss-to-kode.js

code/ss-to-kode.js

@@ -0,0 +1,49 @@
+const WebSocket = require('ws');
+
+async function ss2code(imageBuffer) {
+	return new Promise((resolve, reject) => {
+		const ws = new WebSocket('wss://imagetoappv2.ngrok.app/generate-code');
+		let finalCode = '';
+        
+		ws.on('open', () => {
+			console.log('Connect to WebSocket');
+			ws.send(JSON.stringify({
+				generationType: 'create',
+				image: `data:image/jpeg;base64,${imageBuffer.toString('base64')}`,
+				inputMode: 'image',
+				openAiApiKey: null,
+				openAiBaseURL: null,
+				anthropicApiKey: null,
+				screenshotOneApiKey: null,
+				isImageGenerationEnabled: true,
+				editorTheme: 'cobalt',
+				generatedCodeConfig: 'html_tailwind',
+				codeGenerationModel: 'gpt-4o-2024-05-13',
+				isTermOfServiceAccepted: false
+			}));
+		});
+
+		ws.on('message', (message) => {
+			const response = JSON.parse(message.toString());
+		    if (response.type === 'setCode') {
+		        finalCode = response.value;
+		    } else if (response.type === 'status') {
+		        console.log(response.value);
+		    }
+		});
+
+		ws.on('close', () => {
+			console.log('WebSocket connection closed');
+			resolve(finalCode.trim());
+		});
+
+		ws.on('error', (error) => {
+			reject(new Error(error.message));
+		});
+	});
+}
+
+// Usage:
+const fs = require('fs');
+const resp = await ss2code(fs.readFileSync('./ss.jpg'));
+console.log(resp);