Update app.py

app.py

@@ -140,9 +140,9 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg):
     if isrmbg:
         try:
             rmbgfore = rmbg(fore)
-            print(f"Background removed!")
             if rmbgfore is not None:
-                fore = rmbgfore
+                fore = rmbgfore.convert("RGBA")
+                print(f"Background removed!")
         except:
             pass
          
@@ -157,6 +157,7 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg):
     if size[0]*size[1]>(1500*1500):
         gr.Warning("ℹ️ The input image size is too big, I will lower it!")
         image_width, image_height = resize_to_fit((1500,1500), (image_width, image_height))
+        fore.resize((1500,1500))
 
     
         
@@ -167,8 +168,8 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg):
         "prompt" : prompt,
         "mode" : mode,
         "intensity" : float(intensity),
-        "width" : int(image_width),
-        "height" : int(image_height),
+        "width" : 1500,
+        "height" : 1500,
         "refprompt" : refprompt
     }
     print(f"DATA: {data}")
@@ -263,6 +264,32 @@ with block:
             <li>🎨 <strong>Custom Background Control:</strong> Choose or generate backgrounds that perfectly match your creative vision.</li>
             <li>⚙️ <strong>Easy Integration:</strong> Integrates smoothly with your existing workflow and tools.</li>
         </ul>
+
+        <h2>Step-by-Step Instructions</h2>
+        <p>Follow these instructions to control the generation of backgrounds while keeping the foreground's shape and style consistent:</p>
+        <ul>
+            <li>
+                <span class="emoji">📝</span>
+                <strong>Step 1: Describe the Background</strong>
+                <p>Start by providing a detailed description of the background you want to create.</p>
+                <div class="example">
+                    <p><strong>Example:</strong> "A Perfume Bottle nestled on a crystalline cliff of glistening snow, overlooking a serene, moonlit valley."</p>
+                </div>
+            </li>
+            <li>
+                <span class="emoji">💡</span>
+                <strong>Step 2: Describe the Foreground</strong>
+                <p>Next, describe the texture, lighting, and style of the foreground element.</p>
+                <div class="example">
+                    <p><strong>Example:</strong> "A transparent glass perfume bottle, vibrant, sunset lighting reflecting off its surface."</p>
+                </div>
+            </li>
+            <li>
+                <span class="emoji">🎚️</span>
+                <strong>Step 3: Adjust the Intensity</strong>
+                <p>Decide how much change you want to apply to the image. Adjust the intensity to balance between keeping consistency and introducing new elements.</p>
+            </li>
+        </ul>
     </div>
         ''')
     gr.HTML("""