Browse Source

Fixed error when image is already a jpg and edited the values to be a lil better in terms of output

tags/v1.8.0
Roxie Gibson 6 years ago
parent
commit
699d1b0c8c
1 changed files with 36 additions and 32 deletions
  1. +36
    -32
      roxbot/cogs/image.py

+ 36
- 32
roxbot/cogs/image.py View File

@@ -219,40 +219,44 @@ class CustomCommands:
image = self.image_lookup(ctx.message)
filename = await roxbot.http.download_file(image)

# Convert to jpg
jpg_name = filename.split(".")[0] + ".jpg"
img = Image.open(filename)
img = img.convert(mode="RGB")
img.save(jpg_name)
os.remove(filename)

# Brightness Enhance
img = Image.open(jpg_name)
ehn = ImageEnhance.Brightness(img)
ehn.enhance(1.5).save(jpg_name)

# Contrast Enhance
img = Image.open(jpg_name)
ehn = ImageEnhance.Contrast(img)
ehn.enhance(2).save(jpg_name)

# Sharpness Enhance
img = Image.open(jpg_name)
ehn = ImageEnhance.Sharpness(img)
ehn.enhance(20).save(jpg_name)

# Saturation Enhance
img = Image.open(jpg_name)
ehn = ImageEnhance.Color(img)
ehn.enhance(2).save(jpg_name)

# JPG-fy image
for x in range(10):
async with ctx.typing():
# Convert to jpg
if filename.split(".")[-1] != "jpg":
jpg_name = filename.split(".")[0] + ".jpg"
img = Image.open(filename)
img = img.convert(mode="RGB")
img.save(jpg_name)
os.remove(filename)
else:
jpg_name = filename

# Brightness Enhance
img = Image.open(jpg_name)
ehn = ImageEnhance.Brightness(img)
ehn.enhance(1.25).save(jpg_name)

# Contrast Enhance
img = Image.open(jpg_name)
ehn = ImageEnhance.Contrast(img)
ehn.enhance(1.5).save(jpg_name)

# Sharpness Enhance
img = Image.open(jpg_name)
ehn = ImageEnhance.Sharpness(img)
ehn.enhance(20).save(jpg_name)

# Saturation Enhance
img = Image.open(jpg_name)
img = img.convert(mode="RGB")
img.save(jpg_name)
ehn = ImageEnhance.Color(img)
ehn.enhance(2).save(jpg_name)

# JPG-fy image
for x in range(10):
img = Image.open(jpg_name)
img = img.convert(mode="RGB")
img.save(jpg_name)

await ctx.send(file=discord.File(jpg_name))
await ctx.send(file=discord.File(jpg_name))
os.remove(jpg_name)



Loading…
Cancel
Save