Upload 4 files

app.py

@@ -1,12 +1,16 @@
 import streamlit as st
 from streamlit_option_menu import option_menu
 from generate_plot import generate_plot, generate_storybook, generate_book_cover
+from generate_horror_plot import generate_horror_plot, generate_horror_storybook
 from markup import app_intro, how_use_intro
 
 
 if "generated_code" not in st.session_state:
     st.session_state["generated_code"] = ""
 
+if "generated_horror_code" not in st.session_state:
+    st.session_state["generated_horror_code"] = ""
+
 def tab1():
 
     col1, col2 = st.columns([1, 2])
@@ -76,7 +80,55 @@ def tab2():
             st.markdown(f"<Page {page_number}>")
             st.image(image, use_column_width=True)
 
+
 def tab3():
+    st.header("🕷️ Horror Story Generator 🏚️")
+    st.markdown("Create a spine-chilling horror story with custom length and illustrations!")
+    st.write("Follow the eerie instructions below:")
+
+    #openai_api_key = st.text_input("🔑 Enter your OpenAI API key:", type='password')
+
+    col1, col2 = st.columns(2)
+    with col1:
+
+        st.subheader("Step 1: Choose the Number of Pages 📖")
+        st.write("Select the number of pages for your bone-chilling horror story (between 2 and 7).")
+        number_of_pages = st.number_input("Enter the Number of Pages:",  min_value=2, max_value=7, value=3)
+
+    with col2:
+
+        st.subheader("Step 2: Choose the Art Style 🎨")
+        st.write("Select the art style that will add terror to your horror story illustrations.")
+        art_styles = ["Style 1", "Style 2", "Style 3"]
+        selected_style = st.selectbox("Select Art Style:", art_styles)
+
+    result = None
+
+    if st.button("Generate Plot"):
+
+        with st.spinner('Generating plot...'):
+            result = generate_horror_plot(number_of_pages, selected_style)
+        st.code(result)
+        st.session_state.generated_code = result
+
+    if st.button("Generate Horror Storybook"):
+        if not st.session_state.generated_horror_code:
+            st.warning("Please generate the plot first before creating the horror storybook.")
+            return
+
+        with st.spinner('Generating horror storybook...'):
+            horror_storybook = generate_horror_storybook(st.session_state.generated_horror_code)
+
+        if len(horror_storybook) == 0:
+            st.error("Error: Unable to generate horror storybook.")
+            return
+
+        for page_number, image in horror_storybook:
+            st.markdown(f"<Page {page_number}>")
+            st.image(image, use_column_width=True)
+
+
+def tab4():
     st.header("Book Cover Page Generator")
     st.markdown("Auto Generate a book cover for your children's storybook.")
 
@@ -111,7 +163,7 @@ def tab3():
 
 def main():
     st.set_page_config(page_title="Children storybook generator", page_icon=":memo:", layout="wide")
-    tabs = ["Intro", "Generate book", "Generate Cover"]
+    tabs = ["Intro", "Children", "Horror"]
 
     with st.sidebar:
 
@@ -119,8 +171,8 @@ def main():
 
     tab_functions = {
     "Intro": tab1,
-    "Generate book": tab2,
-    "Generate Cover": tab3,
+    "Children": tab2,
+    "Horror": tab3,
     }
 
     if current_tab in tab_functions:

generate_horror_plot.py

@@ -0,0 +1,141 @@
+import os
+import requests
+import io
+from PIL import Image
+from langchain import PromptTemplate, LLMChain
+from PIL import Image, ImageDraw, ImageFont, ImageFilter
+from langchain.llms import OpenAI
+import openai
+from g4f import Provider, Model
+from langchain_g4f import G4FLLM
+
+def set_openai_api_key(api_key):
+    openai.api_key = api_key
+    os.environ["OPENAI_API_KEY"] = openai.api_key
+
+template = template = """Write a very short and unsettling {number_of_pages}-sentence horror story with images that will give you chills.  
+
+
+Your answer should be structured like this with <Text> and <Image> tags.
+<Text> first sentence of the horror story </Text>
+<Image> describe a matching eerie or spooky image for first sentence here without including names so that prompt can be used to generate an image using an ML model.</Image>
+<Text> second sentence of the horror story </Text>
+<Image> describe a matching eerie or spooky image for second sentence here without including names so that prompt can be used to generate an image using an ML model.</Image>
+
+for all {number_of_pages} sentences.
+=============
+Answer:"""
+
+prompt = PromptTemplate(template=template, input_variables=["number_of_pages"])
+
+def query(payload):
+    #API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
+    API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/all-526-animated"
+    headers = {"Authorization": "Bearer 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"}
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.content
+
+def query_alt(payload):
+    API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/anything-v5"
+    headers = {"Authorization": "Bearer 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"}
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.content
+
+def generate_horror_plot(number_of_pages, selected_style):
+    llm = G4FLLM(
+        model=Model.gpt_35_turbo,
+        provider=Provider.GetGpt,
+    )
+    
+    llm_chain = LLMChain(prompt=prompt, llm=llm)
+    response = llm_chain.run(number_of_pages=number_of_pages)
+    pages = response.split("<Text>")
+    plot_result = []
+
+    additional_texts = {
+        "Style 1": " chilling horror illustration, dark and mysterious, haunting shadows, eerie atmosphere, spooky vector art, 8k, artist unknown",
+        "Style 2": " horror story illustration, monochromatic, deep shadows, creepy background, unsettling, digital art, artist unknown",
+        "Style 3": " horror art, dark and creepy, foggy night, ghostly presence, terrifying, trending on artstation, artist unknown"
+    }
+
+    for i, page in enumerate(pages[1:]):
+        text, img_text = page.split("<Image>", 1)
+        selected_additional_text = additional_texts.get(selected_style, "")
+        
+        new_img_text = img_text.strip() + " " + selected_additional_text
+        text = text.replace("</Text>", "")
+        
+        new_img_text = new_img_text.replace("</Image>", "")
+        
+        plot_result.append((i + 1, "<Text>" + text.strip() + "</Text>", "<Image>" + new_img_text + "</Image>"))
+    return plot_result
+
+
+
+def generate_horror_storybook(plot_result):
+    storybook = []
+    for page_number, text, image_text in plot_result:
+        image_bytes = query({"inputs": image_text})
+        image = Image.open(io.BytesIO(image_bytes))
+        blurred_image = image.filter(ImageFilter.GaussianBlur(8))
+
+        draw = ImageDraw.Draw(blurred_image)
+
+        font_size = 30
+        font = ImageFont.truetype("Birada!.ttf", font_size)
+
+        first_letter_font_size = 60
+        first_letter_font = ImageFont.truetype("Birada!.ttf", first_letter_font_size)
+
+        text_x, text_y = 50, 50
+        max_width = blurred_image.width - text_x * 2
+
+        text = text.replace("<Text>", "").replace("</Text>", "")
+        wrapped_text = ""
+        words = text.split()
+        for i, word in enumerate(words):
+            if i == 0:  
+                first_letter_width = draw.textsize(word[0], font=first_letter_font)[0]
+                draw.text((text_x, text_y), word[0], fill="white", font=first_letter_font, stroke_width=2, stroke_fill="black")
+                text_x += first_letter_width + 10
+                wrapped_text += word[1:] + " "  
+            elif draw.textsize(wrapped_text + word, font=font)[0] < max_width:
+                wrapped_text += word + " "
+            else:
+                draw.text((text_x, text_y), wrapped_text.strip(), fill="white", font=font, stroke_width=2, stroke_fill="black")
+                text_y += font.getsize(wrapped_text)[1] + 10
+                wrapped_text = word + " "
+
+        draw.text((text_x, text_y), wrapped_text.strip(), fill="white", font=font, stroke_width=2, stroke_fill="black")
+        combined_image = Image.new('RGB', (image.width * 2, image.height))
+        combined_image.paste(blurred_image, (0, 0))
+        combined_image.paste(image, (image.width, 0))
+
+        storybook.append((page_number, combined_image))
+
+    return storybook
+
+
+def generate_book_cover(title, author, image_text):
+    book_cover = []
+
+    image_bytes = query({"inputs": image_text})
+    image = Image.open(io.BytesIO(image_bytes))
+
+    cover_image = Image.new('RGB', (image.width * 2, image.height), color='white')
+    cover_image.paste(image, (0, 0))
+
+    draw = ImageDraw.Draw(cover_image)
+    font_size = 50
+    title_font = ImageFont.truetype("DeliusSwashCaps-Regular.ttf", font_size)
+    author_font = ImageFont.truetype("DeliusSwashCaps-Regular.ttf", 30)
+
+    title_x, title_y = image.width + 50, 50
+    author_x, author_y = image.width + 50, title_y + 100
+
+    draw.text((title_x, title_y), title, fill="black", font=title_font)
+    draw.text((author_x, author_y), "By " + author, fill="black", font=author_font)
+
+    book_cover.append(cover_image)
+
+    return book_cover

markup.py

@@ -1,15 +1,15 @@
 def app_intro():
     return """
     <div style='text-align: left;'>
-    <h2 style='text-align: center;'>Welcome to the AI Children Storybook Generator! 📘🎨</h2>
+    <h2 style='text-align: center;'>Welcome to the AI Storybook Generator! 📚🎨</h2>
     <h3 style='text-align: center;'>Unleash Your Imagination</h3>
             
-    <p>Get ready for magical adventures and delightful tales! This AI-powered app lets you create wonderful storybooks for children of all ages, complete with enchanting art and captivating plots!</p>
+    <p>Get ready for magical adventures and spine-chilling tales! This AI-powered app lets you create wonderful storybooks for children of all ages, from delightful fairytales to bone-chilling horror tales, complete with enchanting art and captivating plots!</p>
 
     <h4>What You Can Do:</h4>
     <ul>
-        <li><b>Custom Storybooks:</b> Tell us the age group and the number of pages you want, then enter the names of your favorite characters like Robot, Puppy, or anything you like! Pick a cute art style to bring your story to life.</li>
-        <li><b>Magical Plot Generation:</b> Need inspiration? Just hit the "Generate Plot" button, and our AI will conjure up a whimsical plot for your storybook, filled with surprises and excitement!</li>
+        <li><b>Custom Storybooks:</b> Tell us the age group and the number of pages you want for your storybook. Enter the names of your favorite characters, whether it's friendly fairies or terrifying monsters! Pick a suitable art style to bring your story to life.</li>
+        <li><b>Magical Plot Generation:</b> Need inspiration? Just hit the "Generate Plot" button, and our AI will conjure up a whimsical or spine-chilling plot for your storybook, filled with surprises and excitement!</li>
     </ul>
     </div>          
     """
@@ -21,9 +21,9 @@ def how_use_intro():
     <br>
     <h4>How to Use:</h4>
     <ul>
-    <li><b>Create Your Story:</b> First, tell us the age group and the number of pages you want for your storybook. Then, enter the names of the adorable characters who will star in your tale!</li>
-    <li><b>Art Style:</b> Choose a magical art style that suits your story best! Whether it's dreamy watercolors or vibrant cartoons, the choice is yours!</li>
-    <li><b>Plot Magic:</b> Want a thrilling plot? Just press the "Generate Plot" button, and our AI will weave an enchanting story filled with wonder and joy!</li>
+    <li><b>Create Your Story:</b> First, tell us the age group and the number of pages you want for your storybook. Then, enter the names of the adorable characters for your children's story or the terrifying entities for your horror tale!</li>
+    <li><b>Art Style:</b> Choose a magical art style for children's stories or a haunting art style for horror stories that suits your tale best! Whether it's dreamy watercolors or dark and mysterious illustrations, the choice is yours!</li>
+    <li><b>Plot Magic:</b> Want a thrilling plot for your horror story or a heartwarming adventure for your children's tale? Just press the "Generate Plot" button, and our AI will weave an enchanting or chilling story filled with wonder and suspense!</li>
     <li><b>Cover Creation:</b> Want a beautiful cover for your masterpiece? We've got you covered! Select the "Generate Cover" option to create a captivating cover for your storybook.</li>
     </ul>
     <br>