Function predictImage

src/modelUtils.tsx:1–46

(
  mimeType: string,
  imageData: string,
  prompt: string,
  activeModel: "flash" | "pro",
)

export const predictImage = async (
  mimeType: string,
  imageData: string,
  prompt: string,
  activeModel: "flash" | "pro",
) => {
  const url =
    activeModel === "flash"
      ? "/api/flashGenerateResponseToTextAndImage"
      : "/api/proGenerateResponseToTextAndImage";
  try {
    const response = await fetch(url, {
      headers: {
        "Content-Type": "application/json",
      },
      method: "POST",
      body: JSON.stringify({
        prompt: prompt,
        imageData: imageData,
        mimeType: mimeType,
      }),
    });
    const result = await response.json();
    if (response.ok) {
      return result;
    } else {
      const fetchError = new Error(result.error.message);
      fetchError.fetchResult = result;
      throw fetchError;
    }
  } catch (error) {
    console.log(error)
    if (error.fetchResult) {
      console.error(
        `HTTP request failed with status code ${error.fetchResult.error.code}`,
        error.fetchResult,
      );
      return error.fetchResult;
    }
    return {
      error: {
        message: error.message,
      },
    };
  }
};