Update app.py

app.py

@@ -18,6 +18,14 @@ rmbgkey = os.getenv("RMBGKEY")
 
 #fore, prompt, intensity, mode, refprompt, isrmbg
 outputs_list = [["examples/out1.png"],["examples/out3.png"], ["examples/out4.png"],["examples/out5.png"],["examples/out6.png"],["examples/out7.png"],["examples/out2.png"]]
+prompt_list = ["A Perfume Bottle, resting on a wooden table, surrounded by lavender leaves with droplets of morning dew glistening, in a serene garden, under the soft glow of a twilight sky, encircled by delicate purple petals",
+               "A PAir of Shoes, resting on a sleek white platform under a bright spotlight, with subtle pastel-colored reflections and delicate flower petals scattered around, creating a cheerful and elegant setting",
+               "A Perfume bottle, perched delicately on a rock by the ocean, with a breathtaking sunset casting a golden glow over the waves",
+               "A Bottle, placed on a rustic wooden table, viewed from a slight above angle, overlooking a serene beach, with seashells and driftwood artfully scattered around, bathed in the warm glow of the setting sun",
+               "A Sofa, In a snug, inviting room, a large sofa accompanied by plush cushions, a thick rug, book-filled shelves, potted plants, framed art, a coffee table with candles, a floor lamp, and gentle, warm lighting",
+               "A car, on a high-tech city street at night, surrounded by neon lights and holographic billboards, with sleek skyscrapers and flying cars in the background, a glowing cyber-road beneath",
+                "Headphones, hovering gracefully against a swirling pastel galaxy, accented by glimmers of stardust, ethereal light beams, and floating musical notes, under a dreamy aurora borealis, with no visible wires"
+              ]
 example_list = [
         [
             "examples/in1.png",
@@ -28,19 +36,10 @@ example_list = [
             False,
             "examples/out1.png"
         ],
-        [
-            "examples/in3.png",
-            "A PAir of Shoes, resting on a sleek white platform under a bright spotlight, with subtle pastel-colored reflections and delicate flower petals scattered around, creating a cheerful and elegant setting",
-            2.0,
-            "full",
-            "(( A Pair of Brown Shoes )), vibrant, Shadow",
-            False,
-            "examples/out3.png"
-        ],
         [
             "examples/in4.png",
             "A Bottle, placed on a rustic wooden table, viewed from a slight above angle, overlooking a serene beach, with seashells and driftwood artfully scattered around, bathed in the warm glow of the setting sun",
-            2.50,
+            3.50,
             "full",
             "A transparent Bottle with a black cap and a semi transparent liquide, Sunset",
             False,
@@ -49,16 +48,16 @@ example_list = [
         [
             "examples/in5.png",
             "A Perfume bottle, perched delicately on a rock by the ocean, with a breathtaking sunset casting a golden glow over the waves",
-            4.0,
+            4.5,
             "full",
-            "A glass perfume Bottle, with a yellow lequide and a black cap, sunset, transparent",
+            "(( A transparent perfume Bottle )), with a yellow lequide and a black cap, sunset, transparent",
             False,
             "examples/out5.png"
         ],
         [
             "examples/in6.png",
             "A Sofa, In a snug, inviting room, a large sofa accompanied by plush cushions, a thick rug, book-filled shelves, potted plants, framed art, a coffee table with candles, a floor lamp, and gentle, warm lighting",
-            3.0,
+            3.5,
             "full",
             "A light grey Sofa, studio light, shadows",
             False,
@@ -69,18 +68,27 @@ example_list = [
             "A car, on a high-tech city street at night, surrounded by neon lights and holographic billboards, with sleek skyscrapers and flying cars in the background, a glowing cyber-road beneath",
             3.0,
             "full",
-            "A light grey Car, vibrant, reflections",
+            "(( A light grey Car )), vibrant, reflections",
             False,
             "examples/out7.png"
         ],
         [
             "examples/in2.png",
             "Headphones, hovering gracefully against a swirling pastel galaxy, accented by glimmers of stardust, ethereal light beams, and floating musical notes, under a dreamy aurora borealis, with no visible wires",
-            2.50,
+            3.0,
             "full",
             "white headphones, vibrant, colorful, ",
             False,
             "examples/out2.png"
+        ],
+        [
+            "examples/in3.png",
+            "A PAir of Shoes, resting on a sleek white platform under a bright spotlight, with subtle pastel-colored reflections and delicate flower petals scattered around, creating a cheerful and elegant setting",
+            3.5,
+            "full",
+            "(( A Pair of Brown Shoes )), vibrant, Shadow",
+            False,
+            "examples/out3.png"
         ]
     ]
 
@@ -206,7 +214,10 @@ def resize_to_fit(max_size, original_size):
 
     
 def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg): 
-    
+
+    max_size = (1500,1500)
+    if prompt in prompt_list:
+        max_size = (1800,1800)
     if isrmbg:
         try:
             rmbgfore = rmbg(fore)
@@ -227,7 +238,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))
+        fore.resize(max_size)
 
     
         
@@ -238,8 +249,8 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg):
         "prompt" : prompt,
         "mode" : mode,
         "intensity" : float(intensity),
-        "width" : 1500,
-        "height" : 1500,
+        "width" : max_size[0],
+        "height" : max_size[1],
         "refprompt" : refprompt
     }
     #print(f"DATA: {data}")