diff --git a/modules/aiutils.py b/modules/aiutils.py index 48dab0e..56aaa1f 100644 --- a/modules/aiutils.py +++ b/modules/aiutils.py @@ -36,45 +36,44 @@ async def vdxl(c: Client, message: Message): return data = { - "model": "juggernaut-xl-V5", - "prompt": prompt, - "negative_prompt": "", - "image_count": 1, - "token": vca_api_key, - "width": 1024, - "height": 768, - "enhance": True, - "cfg_scale": 8, - "watermark": False -} - # Send the request to generate images - response = requests.post(f"{api_url}/generate-xl", json=data, verify=False) + "model": "juggernaut-xl-V5", + "prompt": prompt, + "negative_prompt": "", + "token": vca_api_key, + "width": 1024, + "height": 768, + "cfg_scale": 8, + "steps": 30, + "watermark": False + } + # Send the request to generate images + response = requests.post(f"{api_url}/generate-xl", json=data) + + # Extract the request id from the response job_id = response.json()["job_id"] - + + # Check the generation process while True: response = requests.post("https://visioncraft-rs24.koyeb.app/job-status", json={"job_id": job_id}) if response.json()["image"]: - image_url = response.json()["image"] - break - - # Download and save the generated images - for i, image_url in enumerate(image_urls): - # Get the image data from the URL - response = requests.get(image_url) - # Save the image locally - with open(f"generated_image_{i}.png", "wb") as f: - f.write(response.content) + image_url = response.json()["image"] + break + + # Get the image data from the URL + response = requests.get(image_url) + # Save the image locally + with open(f"generated_image.png", "wb") as f: + f.write(response.content) await message.delete() #for i, image_url in enumerate(image_urls): - await c.send_photo(chat_id, photo=f"generated_image_{i}.png", caption=f"Prompt:{prompt}") + await c.send_photo(chat_id, photo=f"generated_image.png", caption=f"Prompt:{prompt}") except Exception as e: await message.edit_text(f"An error occurred: {format_exc(e)}") finally: - for i, image_url in enumerate(image_urls): - os.remove(f"generated_image_{i}.png") + os.remove(f"generated_image.png") @Client.on_message(filters.command("upscale", prefix) & filters.me)