Update app.py

app.py

@@ -6,6 +6,12 @@ import time
 import os
 import datetime
 from datetime import datetime
+from PIL import Image
+from PIL import ImageOps
+from PIL import Image, ImageDraw, ImageFont
+from textwrap import wrap
+import json
+from io import BytesIO
 
 print('for  update')
 
@@ -28,7 +34,33 @@ except requests.exceptions.ReadTimeout as e:
 
 prompt_text=str(r.content, 'UTF-8')
 print(prompt_text)
+data={"prompt_type":'ad_image_prompt',"decodem_token":DECODEM_TOKEN}
+try:
+    r = requests.post(url_decodemprompts, data=json.dumps(data), headers=headers)
+except requests.exceptions.ReadTimeout as e:
+    print(e)
+#print(r.content)
+
+prompt_image=str(r.content, 'UTF-8')
+print(prompt_image)
+
+ENDPOINT_URL="https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1" # url of your endpoint
+#ENDPOINT_URL="https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-1-5" # url of your endpoint
+HF_TOKEN=API_TOKEN# token where you deployed your endpoint
+
+def generate_image(prompt_SD:str):
+    payload = {"inputs": prompt_SD,}
+    headers = {
+        "Authorization": f"Bearer {HF_TOKEN}",
+        "Content-Type": "application/json",
+        "Accept": "image/png" # important to get an image back
+    }
+    response = r.post(ENDPOINT_URL, headers=headers, json=payload)
+    print(response.content)
+    img = Image.open(BytesIO(response.content))
 
+    return img
+    
 def infer(prompt,
           max_length = 250,
           top_k = 0,
@@ -100,19 +132,137 @@ def getadline(text_inp):
         print(topic)
     return(topic)
 
+def getadvertisement(topic):
+    input_keyword=topic
+    backdrop=['surrounded by water droplets','in front of a brick wall','in front of a wooden wall','in a white boho style studio','with nature backdrop','with water splash','laying on a wooden table',]
+    whichitem=random.randint(0,len(backdrop)-1)
+    prompt_SD='product photograph of '+input_keyword+' '+backdrop[whichitem]+prompt_image
+    # generate image
+    image = generate_image(prompt_SD)
+    
+    # save to disk
+    image.save("generation.png")
+    
+    
+    # Set the font to be used
+    req = requests.get("https://github.com/openmaptiles/fonts/raw/master/roboto/Roboto-Regular.ttf")
+    
+    FONT_USER_INFO = ImageFont.truetype(BytesIO(req.content), 75, encoding="utf-8")
+    FONT_TEXT = ImageFont.truetype(BytesIO(req.content), 75, encoding="utf-8")
+    TITLE_TEXT = ImageFont.truetype(BytesIO(req.content), 75, encoding="utf-8")
+    
+    #FONT_USER_INFO = ImageFont.load_default()
+    #FONT_TEXT = ImageFont.load_default()
+    
+    # Image dimensions (pixels)
+    WIDTH = 768
+    HEIGHT = 768
+    # Color scheme
+    COLOR_BG = 'white'
+    COLOR_NAME = 'black'
+    COLOR_TAG = (64, 64, 64)
+    COLOR_TEXT = 'black'
+    # Write coordinates
+    COORD_PHOTO = (10, 40)
+    COORD_NAME = (10, 200)
+    COORD_TAG = (10, 280)
+    COORD_TEXT = (10, 128)
+    # Extra space to add in between lines of text
+    LINE_MARGIN = 5
+    # -----------------------------------------------------------------------------
+    
+    # Information for the image
+    # -----------------------------------------------------------------------------
+    text = getadline(input_keyword)
+    print(text)
+    img_name = "textimage"
+    # -----------------------------------------------------------------------------
+    
+    # Setup of variables and calculations
+    # -----------------------------------------------------------------------------
+    # Break the text string into smaller strings, each having a maximum of 37\
+    # characters (a.k.a. create the lines of text for the image)
+    text_string_lines = wrap(text, 10)
+    
+    # Horizontal position at which to start drawing each line of the tweet body
+    x = COORD_TEXT[0]
+    
+    # Current vertical position of drawing (starts as the first vertical drawing\
+    # position of the tweet body)
+    y = COORD_TEXT[1]
+    
+    # Create an Image object to be used as a means of extracting the height needed\
+    # to draw each line of text
+    temp_img = Image.new('RGB', (0, 0))
+    temp_img_draw_interf = ImageDraw.Draw(temp_img)
+    
+    # List with the height (pixels) needed to draw each line of the tweet body
+    # Loop through each line of text, and extract the height needed to draw it,\
+    # using our font settings
+    line_height = [
+        temp_img_draw_interf.textsize(text_string_lines[i], font=FONT_TEXT )[1]
+        for i in range(len(text_string_lines))
+    ]
+    # -----------------------------------------------------------------------------
+    
+    # Image creation
+    # -----------------------------------------------------------------------------
+    # Create what will be the final image
+    img_final = Image.new('RGB', (WIDTH, HEIGHT), color='white')
+    # Create the drawing interface
+    draw_interf = ImageDraw.Draw(img_final)
+    
+    
+    # Draw each line of the tweet body. To find the height at which the next\
+    # line will be drawn, add the line height of the next line to the current\
+    # y position, along with a small margin
+    for index, line in enumerate(text_string_lines):
+        # Draw a line of text
+        draw_interf.text((x, y), line, font=FONT_USER_INFO,  fill=COLOR_TEXT)
+        # Increment y to draw the next line at the adequate height
+        y += line_height[index] + LINE_MARGIN
+    
+    # Load the user photo (read-mode). It should be a 250x250 circle 
+    #user_photo = Image.open('userprofilepic.png', 'r').convert("RGBA")
+    
+    # Paste the user photo into the working image. We also use the photo for\
+    # its own mask to keep the photo's transparencies
+    #img_final.paste(user_photo, COORD_PHOTO, mask=user_photo)
+    
+    # Finally, save the created image
+    img_final.save(f'{img_name}.png')
+    # -----------------------------------------------------------------------------
+    
+    im = Image.open(img_name+".png")
+    width_orig, height_orig = im.size
+    print(width_orig, height_orig)
+    
+    im_bar = Image.open("generation.png")
+    width_orig_x, height_orig_x = im_bar.size
+    print(width_orig_x, height_orig_x)
+    
+    im_bar = im_bar.resize((int(width_orig / 1), int(height_orig / 1)))
+    new_im = Image.new('RGB', (2*im.size[0],1*im_bar.size[1]), (250,250,250))
+    
+    new_im.paste(im, (0,0))
+    new_im.paste(im_bar, (im.size[0],0))
+    new_im.save('finalimage.png')
+    return 'finalimage.png'
+
+
 with gr.Blocks() as demo:
-    gr.Markdown("<h1><center>Market Sizing Framework for Your Business</center></h1>")
+    gr.Markdown("<h1><center>Ad for Your Business</center></h1>")
     gr.Markdown(
-        """ChatGPT based Insights from <a href="https://www.decodem.ai">Decodem.ai</a> for businesses.\nWhile ChatGPT has multiple use cases we have evolved specific use cases/ templates for businesses \n\n This template provides ideas on how a business can size a market they are entering. Enter a business area to size and get the results. Use examples as a guide. We use a equally powerful AI model bigscience/bloom."""
+        """ChatGPT based Insights from <a href="https://www.decodem.ai">Decodem.ai</a> for businesses.\nWhile ChatGPT has multiple use cases we have evolved specific use cases/ templates for businesses \n\n This template provides ideas on how a business can generate Advertisement ideas for a product. Enter a product area to size and get the results. Use examples as a guide. We use a equally powerful AI model bigscience/bloom."""
         )
-    textbox = gr.Textbox(placeholder="Enter market size focus for business here...", lines=1,label='Your business area')
+    textbox = gr.Textbox(placeholder="Enter product name...", lines=1,label='Your product')
     btn = gr.Button("Generate")    
     #output1 = gr.Textbox(lines=2,label='Market Sizing Framework')
-    output_image = gr.components.Image(label="Image")
+    output_image = gr.components.Image(label="Your Advertisement")
 
 
-    btn.click(getideas,inputs=[textbox], outputs=[output_image])
-    examples = gr.Examples(examples=['ice cream parlor in London','HR saas for fintech','book shops in NYC','Starbucks cafe in Bangalore','organic vegetables via ecommerce','grocery delivery'],
+    btn.click(getadvertisement,inputs=[textbox], outputs=[output_image])
+    examples = gr.Examples(examples=['spectacles','rice cooker','smart watch','coffee mug',],
                            inputs=[textbox])