Update app.py

app.py

@@ -29,7 +29,7 @@ def generate_image(payload):
 	response = requests.post(API_URL_IMG, headers=headers, json=payload)
 	return response.content
 
-if inpt and file is not None:
+if inpt and file is not None and not("Generate" in inpt or "Draw" in inpt or "Imagine" in inpt):
     image = Image.open(file)
     image_bytes = io.BytesIO()
     image.save(image_bytes, format=image.format)
@@ -40,7 +40,7 @@ if inpt and file is not None:
     aimsg = msgs.chat_message("Assistant")
     aimsg.write(output)
 
-elif inpt and file is None:
+elif inpt and file is None and not("Generate" in inpt or "Draw" in inpt or "Imagine" in inpt):
     output = generate_answer(inpt)
     aimsg = msgs.chat_message("Assistant")
     aimsg.write(output)