artintel235 commited on
Commit
c90930f
·
verified ·
1 Parent(s): 101da32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -35,7 +35,7 @@ async def generate_image_async(prompt, aspect_ratio):
35
  response.raise_for_status()
36
  response_data = await response.json()
37
  if "output" in response_data:
38
- return response_data["output"]
39
  elif "error" in response_data:
40
  return response_data['error']
41
  else:
@@ -69,11 +69,11 @@ async def generate_command(
69
  try:
70
  await interaction.response.defer()
71
 
72
- image_url_or_error = await generate_image_async(prompt, aspect_ratio.value)
73
  # await interaction.response.send_message(f"Error:{image_url_or_error}")
74
  if image_url_or_error.startswith("http"):
75
  await interaction.followup.send(
76
- f"Here's your generated image based on the prompt '{prompt}' with aspect ratio {aspect_ratio.name}:\n{image_url_or_error}"
77
  )
78
  else:
79
  await interaction.followup.send(
 
35
  response.raise_for_status()
36
  response_data = await response.json()
37
  if "output" in response_data:
38
+ return response_data["output"],response_data["outputFull"]
39
  elif "error" in response_data:
40
  return response_data['error']
41
  else:
 
69
  try:
70
  await interaction.response.defer()
71
 
72
+ image_url_or_error, additional_data = await generate_image_async(prompt, aspect_ratio.value)
73
  # await interaction.response.send_message(f"Error:{image_url_or_error}")
74
  if image_url_or_error.startswith("http"):
75
  await interaction.followup.send(
76
+ f"Here's your generated image based on the prompt '{prompt}' with aspect ratio {aspect_ratio.name}:\n{image_url_or_error}\n additional data:{additional_data}"
77
  )
78
  else:
79
  await interaction.followup.send(