diff --git a/modules/aiutils.py b/modules/aiutils.py index 2ecda24..602a4b8 100644 --- a/modules/aiutils.py +++ b/modules/aiutils.py @@ -8,7 +8,7 @@ from utils.misc import modules_help, prefix from utils.scripts import format_exc, edit_or_reply # Define the API endpoint -api_url = "https://visioncraft-aiapi.koyeb.app/" +api_url = "https://visioncraft-aiapi.koyeb.app" def upscale_request(image): b = base64.b64encode(image).decode('utf-8') @@ -39,25 +39,26 @@ async def vdxl(c: Client, message: Message): return data = { - "model": "sdxl-turbo", - "prompt": prompt, - "negative_prompt": "", - "image_count": 1, - "token": vca_api_key, - "width": 1024, - "height": 1024, - "enhance": False - } - + "model": "sdxl-turbo", + "prompt": prompt, + "negative_prompt": "bad quality", + "image_count": 1, + "token": vca_api_key, + "width": 1024, + "height": 768, + "enhance": False +} # Send the request to generate images response = requests.post(f"{api_url}/generate-xl", json=data, verify=False) + image_urls = response.json()["images"] + # Extract the image URLs from the response - if "images" in response.json(): - image_urls = response.json()["images"] - else: - await message.edit_text("No images found in the response... Possibly Server is Down") - return + #if "images" in response.json(): + # image_urls = response.json()["images"] + #else: + # await message.edit_text("No images found in the response... Possibly Server is Down") + # return # Download and save the generated images for i, image_url in enumerate(image_urls): @@ -67,8 +68,8 @@ async def vdxl(c: Client, message: Message): with open(f"generated_image_{i}.png", "wb") as f: f.write(response.content) - await message.delete() - for i, image_url in enumerate(image_urls): + 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}") except Exception as e: @@ -100,4 +101,4 @@ async def upscale(client: Client, message: Message): modules_help["aiutils"] = { "vdxl [prompt/reply to promot]*": "Text to Image with SDXL model", "upscale [cap/reply to image]*": "As the name says", -} \ No newline at end of file +}