Upload 5 files

Procfile

@@ -0,0 +1 @@
+web: streamlit run home.py

ad_Template_generator.py

@@ -0,0 +1,79 @@
+import streamlit as st
+from PIL import Image, ImageDraw, ImageFont
+
+def create_ad_Template(punchline, button_text, logo_image, main_image, punchline_font_size, button_font_size):
+    ad_Template_width, ad_Template_height = 800, 1000
+    ad_Template = Image.new("RGB", (ad_Template_width, ad_Template_height), "white")
+
+    main_image = main_image.resize((300, 300))
+    ad_Template.paste(main_image, (250, 300))
+
+
+    logo_size = (200, 150)
+    logo_image = logo_image.resize(logo_size)
+
+    if logo_image.mode != 'RGBA':
+        logo_image = logo_image.convert('RGBA')
+
+    logo_position = (300, 50)
+    ad_Template.paste(logo_image, logo_position, logo_image)
+
+    draw = ImageDraw.Draw(ad_Template)
+
+    font_path = 'fonts/arial.ttf'
+    punchline_font = ImageFont.truetype(font_path, punchline_font_size)
+    button_font = ImageFont.truetype(font_path, button_font_size)
+
+    draw.text((400, 650), punchline, fill="black", font=punchline_font, anchor="mm", spacing=10, align="center", width=400)
+
+    #button size based on text length and font
+    button_text_size = button_font.getmask(button_text).getbbox()[2:]
+    button_width = button_text_size[0] + 40
+    button_height = button_text_size[1] + 20
+
+    button_y_position = 800
+    button_x_position = 400 - button_width // 2
+
+    button_position = (button_x_position, button_y_position)
+    corner_radius = 10
+    draw.rounded_rectangle([button_position, (button_position[0] + button_width, button_position[1] + button_height)],
+                           fill=(144, 238, 144), outline="black", radius=corner_radius)
+
+    text_position = (button_x_position + button_width // 2, button_y_position + button_height // 2)
+    draw.text(text_position, button_text, fill="black", font=button_font, anchor="mm")
+
+    return ad_Template
+
+
+def show_ad_Template_generator_page():
+    st.title("Ad Template Generator")
+
+    punchline = st.text_input("Enter the punchline text for the Ad Template:")
+    punchline_font_size = st.select_slider("Select the font size for the punchline:", options=[i for i in range(1, 101)])
+
+
+    button_text = st.text_input("Enter the button text:")
+    button_font_size = st.select_slider("Select the font size for the button text:", options=[i for i in range(1, 101)])
+
+    # Option to use the generated image or upload a new one
+    use_generated_image = st.checkbox("Use the image generated by Stable Diffusion")
+
+    logo_image = st.file_uploader("Choose a logo image...", type=["jpg", "jpeg", "png"])
+    if logo_image is not None:
+        logo_image = Image.open(logo_image)
+
+    main_image = None
+    if use_generated_image and 'generated_image' in st.session_state:
+        main_image = st.session_state['generated_image']
+        st.image(main_image, caption="Generated Image", use_column_width=True)
+    else:
+        uploaded_main_image = st.file_uploader("Choose the main image...", type=["jpg", "jpeg", "png"])
+        if uploaded_main_image is not None:
+            main_image = Image.open(uploaded_main_image)
+
+    generate_button = st.button("Generate Ad Template")
+
+    if generate_button and punchline and button_text and logo_image and main_image:
+        ad_Template = create_ad_Template(punchline, button_text, logo_image, main_image, punchline_font_size, button_font_size)
+
+        st.image(ad_Template, caption="Generated Ad Template", use_column_width=True)

home.py

@@ -0,0 +1,28 @@
+import streamlit as st
+from stable_diffusion import show_stable_diffusion_page
+from ad_Template_generator import show_ad_Template_generator_page
+
+st.set_page_config(page_title="Multi-Page App", layout="wide")
+
+#the navigation structure
+PAGES = {
+   "Home": lambda: st.markdown("""
+   # Welcome to the Ad Teamplate Generator App 👋 !
+
+   This application allows you to generate ad templates with customizable elements such as headlines, button texts, and images.
+   You can create your ad template with an image created with stable diffusion or with an image you upload yourself.
+
+   To get started,my suggestion is to first go to the Stable Diffusion page and create your image. Afterward, navigate to the Ad Template Generator page to prepare your template.
+
+   Enjoy creating your own ad templates!
+   """),
+   "Stable Diffusion": show_stable_diffusion_page,
+   "Ad Template Generator": show_ad_Template_generator_page
+}
+
+
+st.sidebar.title("Navigation")
+selection = st.sidebar.radio("Go to", list(PAGES.keys()))
+
+page = PAGES[selection]
+page()

requirements.txt

@@ -0,0 +1,4 @@
+streamlit
+diffusers
+torch
+PIL

stable_diffusion.py

@@ -0,0 +1,31 @@
+import streamlit as st
+import torch
+from PIL import Image
+
+from diffusers import StableDiffusionImg2ImgPipeline
+
+def show_stable_diffusion_page():
+    st.title("Generate Image Using Stable Diffusion Img2Img")
+
+    # Initialize the stable diffusion pipeline
+    model_id_or_path = "runwayml/stable-diffusion-v1-5"
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float32)
+    pipe.to(device)
+
+
+    uploaded_main_image = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
+
+    if uploaded_main_image is not None:
+        main_image = Image.open(uploaded_main_image).convert("RGB")
+        main_image = main_image.resize((768, 512))
+
+
+        prompt = st.text_input("Enter a prompt:")
+        #st.write(f"User entered prompt: {prompt}")
+
+
+        if st.button("Generate Image"):
+            result_images = pipe(prompt=prompt, image=main_image, strength=0.75, guidance_scale=7.5).images
+            st.session_state['generated_image'] = result_images[0]
+            st.image(result_images[0], caption="Generated Image", use_column_width=True)