ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-18 20:27:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

updated midjourney

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-08-25 04:15:23 +02:00
parent 62fff368de
commit 92109353b8
1 changed files with 27 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
lollms/services/midjourney/lollms_midjourney.py
View File

@ -180,7 +180,31 @@ class LollmsMidjourney(LollmsTTI):
else:
print(f"Failed to download image. Status code: {response.status_code}")
return None
def get_nearest_aspect_ratio(self, width: int, height: int) -> str:
# Define the available aspect ratios
aspect_ratios = {
"1:2": 0.5,
"2:3": 0.6667,
"3:4": 0.75,
"4:5": 0.8,
"1:1": 1,
"5:4": 1.25,
"4:3": 1.3333,
"3:2": 1.5,
"16:9": 1.7778,
"7:4": 1.75,
"2:1": 2
}
# Calculate the input aspect ratio
input_ratio = width / height
# Find the nearest aspect ratio
nearest_ratio = min(aspect_ratios.items(), key=lambda x: abs(x[1] - input_ratio))
# Return the formatted string
return f"--ar {nearest_ratio[0]}"
def paint(
self,
positive_prompt,
@ -200,6 +224,7 @@ class LollmsMidjourney(LollmsTTI):
try:
# Send prompt and get initial response
positive_prompt += self.get_nearest_aspect_ratio(width, height)
initial_response = self.send_prompt_with_retry(positive_prompt, self.retries)
message_id = initial_response.get("messageId")
if not message_id:
@ -210,7 +235,7 @@ class LollmsMidjourney(LollmsTTI):
if "error" in progress_response:
raise ValueError(progress_response["error"])
if width<1024:
if width<=1024:
file_name = self.download_image(progress_response["uri"], output_path)
return file_name, {"prompt":positive_prompt, "negative_prompt":negative_prompt}