async function generateCodeFromImage()

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,
  };
}