diff --git a/app/services/image_service.py b/app/services/image_service.py index 6da7eb2..7d78c27 100644 --- a/app/services/image_service.py +++ b/app/services/image_service.py @@ -15,37 +15,61 @@ container_name = "greefile" AZURE_ACCOUNT_KEY = os.getenv("AZURE_ACCOUNT_KEY") connection_string = f'DefaultEndpointsProtocol=https;AccountName=greedotstorage;AccountKey={AZURE_ACCOUNT_KEY};EndpointSuffix=core.windows.net' - promptDict = { 1: """ - please make cute + **normal** please make cute This picture was painted by a child. Please change this picture a little bit more cute. I want to keep most of the original. Arms and legs must be in the form of characters. Please don't put your face and body in. I just wish I had one character. And the parts other than the character outline must be a white background, so please make this important. - white background. + transparent png download, + --no background, shadow, surrounding friend, surrounding objects + --quality 0.5 """, 2: """ - please make cool - This picture was painted by a child. - Please change this picture a little bit more cute. - I want to keep most of the original. - Arms and legs must be in the form of characters. - Please don't put your face and body in. I just wish I had one character. - And the parts other than the character outline must be a white background, so please make this important. - white background. + **sketch** + in the style of childs drawing, + transparent png download, + simple, + cute and colorful, + hd, + full body shot, + wearing shoes, + --no background, shadow, surrounding friend, surrounding objects + --quality 0.5 + --stylize 50 + """, + 3: """ + **anime** + anime, + transparent png download, + simple, + cute and colorful, + hd, + full body shot, + wearing shoes, + --no background, shadow, surrounding friend, surrounding objects + --quality 0.5 + """, + 4: """ + **dizney** + Disney style, + 3d animation, + transparent png download, + simple, + cute and colorful, + hd, + full body shot, + wearing shoes, + --no background, shadow, surrounding friend, surrounding objects """ } async def create_image(promptSelect: int, raw_img_url: str) -> dict: - prompt = promptDict[promptSelect] + "\n" + raw_img_url - prompt += """ - --no background, shadow, surrounding friend, surrounding objects - --quality 0.25 - --stylize 50 - """ + prompt = raw_img_url + ' ' + prompt += promptDict[promptSelect] headers = { 'Authorization': f'Bearer {MID_API_KEY}',