Update app.py

app.py

@@ -26,7 +26,7 @@ def generate_caption(image):
     response = openai.ChatCompletion.create(
                 model="gpt-4",
                 messages=[
-                    {"role": "system", "content": "Escribe un meme chistoso para una imagen a partir en la descripción dada por el usuario. No uses emojis. El meme tiene que ser corto y gracioso. El output del asistente solo debe ser el meme. Asegúrate que el meme sea tan bueno que se vuelva viral!"},
+                    {"role": "system", "content": "Escribe un meme chistoso en español para una imagen a partir en la descripción dada por el usuario. No uses emojis. El meme tiene que ser corto y gracioso. El output del asistente solo debe ser el meme. Asegúrate que el meme sea tan bueno que se vuelva viral!"},
                     {"role": "user", "content": caption}
                 ],
                 temperature=0.6
@@ -35,19 +35,19 @@ def generate_caption(image):
     meme_text = response.choices[0].message.content
     print(meme_text)
 
-     # put the meme text on the image
+    # Put the meme text on the image
     draw = ImageDraw.Draw(pil_image)
 
-    # Calculate font size
-    font_size = int(pil_image.width * 0.8)  # Adjust the multiplier to change the font size based on image width
+    # Determine font size
+    max_width = int(pil_image.width * 0.9)
+    font_size = int(max_width / (len(meme_text) / 2))
     font = ImageFont.truetype("impact.ttf", font_size)
-    
-    # Wrap the text to fit within the image width
-    wrapped_text = textwrap.fill(meme_text, width=int(pil_image.width / font_size * 0.8))
-    
-    # Calculate the position to place the text at the top and center horizontally
+
+    # Wrap the text to fit within the image width and have a maximum of 2 lines
+    wrapped_text = textwrap.fill(meme_text, width=int(max_width / font.getsize("A")[0]))
     text_lines = wrapped_text.split('\n')
-    y = 10  # Adjust this value to add more or less padding from the top
+
+    y = 10
     for line in text_lines:
         line_width, line_height = draw.textsize(line, font=font)
         x = (pil_image.width - line_width) // 2