Atualização

app.py

@@ -30,6 +30,8 @@ sf_neg_prompt_2 = "animal"
 template1 = Image.open("templates/template1.png").resize((512, 512))
 template2 = Image.open("templates/template2.png").resize((512, 512))
 
+fontMain = ImageFont.truetype(font="fonts/arial.ttf", size=32)
+fontSecond = ImageFont.truetype(font="fonts/arial.ttf", size=18)
 
 def numpy_to_pil(images):
     if images.ndim == 3:
@@ -115,9 +117,8 @@ async def predict(prompt, negative_prompt, image, mask_img):
     return (img_final)
 
 
-async def predicts(sf_prompt_1, sf_neg_prompt_1, sf_prompt_2, sf_neg_prompt_2, image, image_numpy, mask_img):
-    testing_local = True
-    if testing_local:
+async def predicts(sf_prompt_1, sf_neg_prompt_1, sf_prompt_2, sf_neg_prompt_2, image, image_numpy, mask_img, only_test):
+    if only_test:
         img1 = Image.fromarray(image_numpy).convert("RGB").resize((512, 512))
         img2 = Image.fromarray(image_numpy).convert("RGB").resize((512, 512))
         return img1, img2
@@ -135,21 +136,24 @@ async def predicts(sf_prompt_1, sf_neg_prompt_1, sf_prompt_2, sf_neg_prompt_2, i
     return img1, img2
 
 
-def draw_text(img, template_coords, text1, text2):
-    font1 = ImageFont.truetype(font="fonts/arial.ttf", size=18)
-    font2 = ImageFont.truetype(font="fonts/arial.ttf", size=14)
-
+def draw_text(img, template_coords, main_text, second_text):
     x1 = template_coords['x1']
     y1 = template_coords['y1']
     x2 = template_coords['x2']
     y2 = template_coords['y2']
 
+    if '\\n' in main_text:
+        main_text = main_text.replace('\\n', '\n')
+
+    if '\\n' in second_text:
+        second_text = second_text.replace('\\n', '\n')
+
     draw = ImageDraw.Draw(img)
-    draw.text((x1, y1), text1, fill=(255, 0, 0), font=font1)
-    draw.text((x2, y2), text2, fill=(255, 0, 0), font=font2)
+    draw.text((x1, y1), main_text, fill=(255, 255, 255), font=fontMain)
+    draw.text((x2, y2), second_text, fill=(255, 255, 255), font=fontSecond)
 
 
-def inference(obj2mask, image_numpy, txt_1, txt_2):
+def inference(obj2mask, image_numpy, main_text, second_text, only_test):
     generator = torch.Generator()
     generator.manual_seed(int(52362))
 
@@ -157,7 +161,7 @@ def inference(obj2mask, image_numpy, txt_1, txt_2):
 
     mask_img = get_mask(obj2mask, image)
 
-    img1, img2 = asyncio.run(predicts(sf_prompt_1, sf_neg_prompt_1, sf_prompt_2, sf_neg_prompt_2, image, image_numpy, mask_img))
+    img1, img2 = asyncio.run(predicts(sf_prompt_1, sf_neg_prompt_1, sf_prompt_2, sf_neg_prompt_2, image, image_numpy, mask_img, only_test))
 
     img1_1 = add_template(img1.copy(), template1.copy())
     img1_2 = add_template(img1.copy(), template2.copy())
@@ -166,23 +170,23 @@ def inference(obj2mask, image_numpy, txt_1, txt_2):
     img2_2 = add_template(img2.copy(), template2.copy())
 
     template1_coords = {
-        'x1': 5,
-        'y1': 300,
-        'x2': 15,
-        'y2': 400
+        'x1': 700/2,
+        'y1': 630/2,
+        'x2': 420/2,
+        'y2': 800/2
     }
 
     template2_coords = {
-        'x1': 5,
-        'y1': 300,
-        'x2': 15,
-        'y2': 400
+        'x1': 30/2,
+        'y1': 800/2,
+        'x2': 430/2,
+        'y2': 780/2
     }
 
-    draw_text(img1_1, template1_coords, txt_1, txt_2)
-    draw_text(img1_2, template2_coords, txt_1, txt_2)
-    draw_text(img2_1, template1_coords, txt_1, txt_2)
-    draw_text(img2_2, template2_coords, txt_1, txt_2)
+    draw_text(img1_1, template1_coords, main_text, second_text)
+    draw_text(img1_2, template2_coords, main_text, second_text)
+    draw_text(img2_1, template1_coords, main_text, second_text)
+    draw_text(img2_2, template2_coords, main_text, second_text)
 
     return [img1_1, img1_2, img2_1, img2_2]
 
@@ -190,12 +194,13 @@ def inference(obj2mask, image_numpy, txt_1, txt_2):
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
-            txt_1 = gr.Textbox(label="Texto principal da propaganda", value="Promoção Imperdível")
-            txt_2 = gr.Textbox(label="Texto secundário da propaganda", value="Até 50% para alguns produtos")
+            txt_1 = gr.Textbox(label="Texto principal da propaganda", value="Promoção\nImperdível", lines=2)
+            txt_2 = gr.Textbox(label="Texto secundário da propaganda", value="Até 50% para alguns produtos\nEntre em contato com um dos\nnossos vendedores", lines=3)
 
             mask = gr.Textbox(label="Descrição da imagem", value="shoe")
             intput_img = gr.Image()
-            run = gr.Button(value="Generate")
+            run = gr.Button(value="Gerar")
+            chk_test = gr.Checkbox(label='Gerar Prévia')
     with gr.Row():
         with gr.Column():
             output_img1_1 = gr.Image()
@@ -212,7 +217,7 @@ with gr.Blocks() as demo:
 
     run.click(
         inference,
-        inputs=[mask, intput_img, txt_1, txt_2],
+        inputs=[mask, intput_img, txt_1, txt_2, chk_test],
         outputs=[output_img1_1, output_img1_2, output_img2_1, output_img2_2],
     )