From 25d51590443561e28b5b32fdbbf006ca6a3eb5df Mon Sep 17 00:00:00 2001
From: BFlat <bombayapirat@gmail.com>
Date: Thu, 18 Jan 2024 23:44:13 +0700
Subject: [PATCH] sd inference panel width and height update

---
 .../inference_panels/SdInferencePanel.tsx     | 24 +++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/packages/fakeyou.com/src/v2/view/pages/weight/inference_panels/SdInferencePanel.tsx b/packages/fakeyou.com/src/v2/view/pages/weight/inference_panels/SdInferencePanel.tsx
index 3a49b4fa22..fd259e5c72 100644
--- a/packages/fakeyou.com/src/v2/view/pages/weight/inference_panels/SdInferencePanel.tsx
+++ b/packages/fakeyou.com/src/v2/view/pages/weight/inference_panels/SdInferencePanel.tsx
@@ -128,6 +128,26 @@ export default function SdInferencePanel({
     { label: "Custom", value: "custom" },
   ];
 
+  let imageWidth: number;
+  let imageHeight: number;
+
+  switch (aspectRatio) {
+    case "square":
+      imageWidth = 512;
+      imageHeight = 512;
+      break;
+    case "landscape":
+      imageWidth = 768;
+      imageHeight = 512;
+      break;
+    case "portrait":
+      imageWidth = 512;
+      imageHeight = 768;
+      break;
+    default:
+      throw new Error("Invalid aspect ratio");
+  }
+
   const handlePromptChange = (
     event: React.ChangeEvent<HTMLTextAreaElement>
   ) => {
@@ -215,8 +235,8 @@ export default function SdInferencePanel({
       maybe_prompt: prompt,
       maybe_n_prompt: negativePrompt,
       maybe_seed: internalSeed.current,
-      maybe_width: 512,
-      maybe_height: 512,
+      maybe_width: imageWidth,
+      maybe_height: imageHeight,
       maybe_sampler: sampler,
       maybe_cfg_scale: cfgScale,
       maybe_number_of_samples: samples,