Create app.py

app.py

@@ -0,0 +1,119 @@
+
+import PIL
+import requests
+import torch
+from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
+from PIL import Image, ImageDraw, ImageFont,ImageOps
+import streamlit as st
+
+
+def generate_image(url,color):
+    model_id = "timbrooks/instruct-pix2pix"
+    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
+    pipe.to("cuda")
+    pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    image = download_image(url)
+    prompt = f"create a similar image, change the object color to {color}"
+    images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
+    generated_image = images[0].save("generated_image.jpg")
+    return images[0]
+
+
+def download_image(url):
+    image = Image.open(url)
+    image = ImageOps.exif_transpose(image)
+    image = image.convert("RGB")
+    return image
+
+def create_frame_with_resized_image(frame_size, image,logo_path, output_path, new_image_size,logo_size,text,button,color):
+    # Create an empty white frame
+    frame = Image.new("RGB", frame_size, "white")
+
+    # Open the image to be placed in the frame
+    original_image = image
+    logo = Image.open(logo_path)
+    # Resize the image to the desired dimensions
+    resized_image = original_image.resize(new_image_size)
+    resized_logo = logo.resize(logo_size)
+    # Calculate the position to center the resized image in the frame
+    x_position = (frame.width - resized_image.width) // 2
+    y_position = (frame.height - resized_image.height) // 2
+    
+    frame.paste(resized_logo,(200,0))
+    # Paste the resized image onto the frame
+    frame.paste(resized_image, (x_position, y_position))
+
+    draw = ImageDraw.Draw(frame)
+    font_size = 24
+    font = ImageFont.truetype("arial.ttf", font_size)
+    text_width, text_height = draw.textsize(text, font)
+    text_x = (frame.width - text_width) // 2
+    
+    button_width, button_height = draw.textsize(button, font)
+    button_x = (frame.width - button_width) // 2
+    button_y = 500
+    # Add text to the image
+    draw.text((text_x, 450), text, font=font, fill= f"{color}")
+    
+    # Save the result
+    
+    frame_thickness = 5
+    draw.rectangle([(button_x - frame_thickness, button_y - frame_thickness),
+                    (button_x + button_width + frame_thickness, button_y + button_height + frame_thickness)],
+                    fill=f"{color}")
+    
+    draw.text((button_x,500),button,font=font, fill= "white")
+
+    frame.save(output_path)
+    
+
+def main():
+    st.set_page_config(page_title="img 2 audio story")
+    st.header("Turn img into ad template")
+    uploaded_file = st.file_uploader("Choose an image...", type="jpg")
+    color_input = st.text_input("Enter the color you want in the image")
+    st.write("You entered:",color_input)
+    
+    if uploaded_file is not None and color_input is not None:
+        print(uploaded_file)
+        bytes_data = uploaded_file.getvalue()
+        with open(uploaded_file.name, "wb") as file:
+            file.write(bytes_data)
+        st.image(uploaded_file,caption="Uploaded Image.",
+                use_column_width=True)
+        
+        
+    #st.image(generated_image,caption="Generated Image.", use_column_width=True)
+    
+    #prompt_input = st.text_input("Enter the prompt:")
+    #st.write("You entered:", prompt_input)
+    punchline_input = st.text_input("Enter the punchline:")
+    st.write("You entered:", punchline_input)
+    
+    button_input = st.text_input("Enter the button text:")
+    st.write("You entered:", button_input)
+    
+    button_color_input = st.text_input("Enter the text and button color:")
+    st.write("You entered:",button_color_input)
+    
+    uploaded_logo = st.file_uploader("Upload the logo", type="jpg")
+    if  button_color_input is not None and uploaded_logo is not None and button_input is not None and punchline_input is not None:
+        frame_size = (600, 600)
+        logo_data = uploaded_logo.getvalue()
+        with open(uploaded_logo.name,  "wb") as file:
+            file.write(logo_data)
+            
+        output_path = "result.jpg"
+        new_image_size = (400, 300)  # Adjust these dimensions as needed
+        logo_size = (200,150)
+        image = generate_image(uploaded_file.name,color_input)
+        logo_path = uploaded_logo.name
+        create_frame_with_resized_image(frame_size, image,logo_path, output_path, new_image_size,logo_size,punchline_input,button_input,button_color_input)
+        ad_template = Image.open(output_path)
+        st.image(ad_template,caption="Generated Ad Template", use_column_width=True)
+        
+if __name__ == "__main__":
+    main()
+
+
+