in src/image_to_code/Home.tsx [59:85]
async function generateCodeFromImage(imageBase64, prompt, userInput) {
const image = {
inlineData: {
data: imageBase64.split(',')[1],
mimeType: 'image/jpeg',
},
};
const finalPrompt = userInput.trim()
? `${prompt}\n\nUser input: ${userInput}`
: prompt;
const result = await ai.models.generateContent({
model: MODEL_NAME,
contents: [finalPrompt, image],
});
const response = result.text;
const regex = /```(?:javascript|js)?\s*([\s\S]*?)```/g;
const match = regex.exec(response);
const extractedCode = match ? match[1].trim() : response;
return {
fullResponse: response,
code: extractedCode,
};
}