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
+}