Update app.py

app.py

@@ -748,7 +748,7 @@ def _summarize(text):
     toks = tokenizer.encode( prefix + text, return_tensors="pt", truncation=False)
     gen = model.generate(
         toks,
-        length_penalty=0.5,
+        length_penalty=0.9,
         num_beams=8,
         early_stopping=True,
         max_length=512
@@ -787,31 +787,42 @@ def generate_random_string(length):
     characters = str(ascii_letters + digits)
     return ''.join(random.choice(characters) for _ in range(length))
 
-def add_song_cover_text(img,title,h,w):
+def add_song_cover_text(img,top_title=None,bottom_title=none):
 
-    draw = ImageDraw.Draw(img,mode="RGBA")
-
-    labels_distance = 1/3
+    if not working:
 
-    rows = len(song.split("\n"))
-    textheight=min(math.ceil( w / 10 ), math.ceil( h / 5 ))
-    font = ImageFont.truetype(r"Alef-Bold.ttf", textheight)
-    textwidth = draw.textlength(song,font)
-    x = math.ceil((w - textwidth) / 2)
-    y = h - (textheight * rows / 2) - (h / 2)
-    y = math.ceil(y - (h / 2 * labels_distance))
-    draw.text((x, y), song, (255,255,255,85), font=font, spacing=2, stroke_width=math.ceil(textheight/20), stroke_fill=(0,0,0,170))
+        working = True
 
-    rows = len(artist.split("\n"))
-    textheight=min(math.ceil( w / 10 ), math.ceil( h / 5 ))
-    font = ImageFont.truetype(r"Alef-Bold.ttf", textheight)
-    textwidth = draw.textlength(artist,font)
-    x = math.ceil((w - textwidth) / 2)
-    y = h - (textheight * rows / 2) - (h / 2)
-    y = math.ceil(y + (h / 2 * labels_distance))
-    draw.text((x, y), artist, (0,0,0,85), font=font, spacing=2, stroke_width=math.ceil(textheight/20), stroke_fill=(255,255,255,170))
+        h = _HEIGHT_
+        w = _WIDTH_
+        
+        draw = ImageDraw.Draw(img,mode="RGBA")
+    
+        labels_distance = 1/3
+    
+        if top_title:
+            rows = len(song.split("\n"))
+            textheight=min(math.ceil( w / 10 ), math.ceil( h / 5 ))
+            font = ImageFont.truetype(r"Alef-Bold.ttf", textheight)
+            textwidth = draw.textlength(top_title,font)
+            x = math.ceil((w - textwidth) / 2)
+            y = h - (textheight * rows / 2) - (h / 2)
+            y = math.ceil(y - (h / 2 * labels_distance))
+            draw.text((x, y), top_title, (255,255,255), font=font, spacing=2, stroke_width=math.ceil(textheight/20), stroke_fill=(0,0,0))
+    
+        if bottom_title:
+            rows = len(artist.split("\n"))
+            textheight=min(math.ceil( w / 10 ), math.ceil( h / 5 ))
+            font = ImageFont.truetype(r"Alef-Bold.ttf", textheight)
+            textwidth = draw.textlength(bottom_title,font)
+            x = math.ceil((w - textwidth) / 2)
+            y = h - (textheight * rows / 2) - (h / 2)
+            y = math.ceil(y + (h / 2 * labels_distance))
+            draw.text((x, y), bottom_title, (0,0,0), font=font, spacing=2, stroke_width=math.ceil(textheight/20), stroke_fill=(255,255,255))
 
-    return img
+        working = False
+        
+        return img
 
 google_translate_endpoint = "https://translate.google.com/m"
 language_codes = {
@@ -1523,8 +1534,22 @@ if __name__ == "__main__":
 
             with gr.Column():
                 cover = gr.Image(interactive=False,container=False,elem_classes="image-container", label="Result", show_label=True, type='pil', show_share_button=False)
-                upscale_now = gr.Button("Upscale",elem_classes="btn")
-
+                with gr.Column():
+                    upscale_now = gr.Button("Upscale",elem_classes="btn")
+                with gr.Column():
+                    top = gr.Textbox(
+                        placeholder="Top title",
+                        value="",
+                        container=False,
+                        max_lines=1
+                    )
+                    bottom = gr.Textbox(
+                        placeholder="Bottom title",
+                        value="",
+                        container=False,
+                        max_lines=1
+                    )
+                    add_titles = gr.Button("Add title(s)",elem_classes="btn")
         gr.on(
             triggers=[run.click],
             fn=handle_generation,
@@ -1536,5 +1561,11 @@ if __name__ == "__main__":
             inputs=[cover],
             outputs=[cover]
         )
+        add_titles.click(
+            fn=add_song_cover_text,
+            inputs=[cover,top,bottom],
+            outputs=[cover]
+        )
+        
 
     demo.queue().launch()