Ufoptg commited on
Commit
061dbff
·
verified ·
1 Parent(s): 2039e69

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -7
main.py CHANGED
@@ -475,6 +475,7 @@ async def bing_dalle(item: BingDalle):
475
  try:
476
  BingImages = await BingClient(image_provider=BingCreateImages)
477
  except Exception:
 
478
  BingImages = BingClient(image_provider=BingCreateImages)
479
 
480
  try:
@@ -489,7 +490,6 @@ async def bing_dalle(item: BingDalle):
489
 
490
  async def generate_image(prompt, model):
491
  try:
492
- # Call BingCreateImages.create() with the prompt argument
493
  response = await BingImages.images.generate(
494
  prompt=BingCreateImages.create(prompt),
495
  model=model,
@@ -501,13 +501,7 @@ async def bing_dalle(item: BingDalle):
501
 
502
  try:
503
  response = await generate_image(item.prompt, item.model)
504
- except BaseException as e:
505
- return {"status": "false", "message": f"Something went wrong: {e}"}
506
- try:
507
  image_url = response.data[0].url
508
- except BaseException as e:
509
- return {"status": "false", "message": f"No image URL found in the {response} Error: {e}"
510
-
511
  if image_url is not None:
512
  return SuccessResponse(status="True", randydev={"message": image_url})
513
  else:
 
475
  try:
476
  BingImages = await BingClient(image_provider=BingCreateImages)
477
  except Exception:
478
+ print("Not awaited")
479
  BingImages = BingClient(image_provider=BingCreateImages)
480
 
481
  try:
 
490
 
491
  async def generate_image(prompt, model):
492
  try:
 
493
  response = await BingImages.images.generate(
494
  prompt=BingCreateImages.create(prompt),
495
  model=model,
 
501
 
502
  try:
503
  response = await generate_image(item.prompt, item.model)
 
 
 
504
  image_url = response.data[0].url
 
 
 
505
  if image_url is not None:
506
  return SuccessResponse(status="True", randydev={"message": image_url})
507
  else: