Upload 7 files

app.py

@@ -0,0 +1,111 @@
+import streamlit as st
+from constants import *
+from stqdm import stqdm
+from prompts import *
+from generator import *
+from utils import *
+
+
+st.set_page_config(
+    layout="wide",
+    page_title="AI Book Generator",
+    page_icon=":book:",
+)
+st.title("AI Book Generator")
+st.markdown("<h3>Select options </h3>", unsafe_allow_html=True)
+with st.expander("Educational value *"):
+    age_range = st.select_slider("Age range of the reader", options=AGE_RANGE)
+    skill_development = st.selectbox("Skill development", options=SKILL_DEVELOPMENT)
+    learning_obectives = st.selectbox(
+        "Learning objectives", options=LEARNING_OBJECTIVES
+    )
+with st.expander("Emotional value *"):
+    theme = st.selectbox("Theme", options=THEME)
+    mood = st.selectbox("Moood of story", options=MODD_OF_STORY)
+    positive_messaging = st.selectbox("Skill development", options=POSITIVE_MESSAGNG)
+with st.expander("Personal *"):
+    theme = st.selectbox("Gender", options=GENDER)
+    fvrt_book = st.text_input("Favorite book")
+with st.expander("Book Details * "):
+    chapters = st.number_input(
+        "How many chapters should the book have?", min_value=3, max_value=100, value=5
+    )
+
+    title = st.text_input("Title of the book")
+    genre = st.selectbox("Genre", options=GENRE)
+    topic = st.selectbox("Topic ", options=TOPIC)
+    main_name = st.text_input("Name of main character")
+    type_of_main_character = st.selectbox(
+        "Type of main character", TYPE_OF_MAIN_CHARACTER
+    )
+    antagonist_name = st.text_input("Antagonist name")
+    antagonsit_type = st.selectbox("Antagonist type", options=ANTAGONIST_TYPE)
+    suuporting_character_name = st.text_input("Supporting character name (if any)")
+    suporting_character_type = st.selectbox(
+        "Supporting character type", options=SUPPORTING_CHARACTER_TYPE
+    )
+    settings = st.selectbox("Setting ", options=SETTINGS)
+    resolution = st.selectbox("Resolution", options=RESOLUTION)
+
+btn = st.button("Generate Book")
+if btn:
+    content = []
+    for x in stqdm(range(chapters), desc="Generating book"):
+        if x == 0:
+            prmpt = get_initial_prompts(
+                genre,
+                type_of_main_character,
+                main_name,
+                skill_development,
+                learning_obectives,
+                theme,
+                topic,
+            )
+            content.append(complete_with_gpt(prmpt, 200, "gpt2", 1500, 0.7, 1.5))
+        if x == 1:
+            prmpt = story_setting_prompt(
+                genre,
+                type_of_main_character,
+                main_name,
+                skill_development,
+                learning_obectives,
+                theme,
+                mood,
+                antagonist_name,
+                antagonsit_type,
+            )
+            previous = " ".join(x for x in content)
+            prmpt = previous + " " + prmpt
+            content.append(complete_with_gpt(prmpt, 200, "gpt2", 1500, 0.7, 1.5))
+
+        if x % 3 == 0:
+            prmpt = supporting_character_inclusion(
+                genre,
+                suuporting_character_name,
+                suporting_character_type,
+                positive_messaging,
+            )
+            previous = " ".join(x for x in content)
+            prmpt = previous + " " + prmpt
+            content.append(complete_with_gpt(prmpt, 200, "gpt2", 1500, 0.7, 1.5))
+        if x == chapters - 1:
+            prmpt = ending_scene(genre, resolution, main_name, positive_messaging)
+            previous = " ".join(x for x in content)
+            prmpt = previous + " " + prmpt
+            content.append(complete_with_gpt(prmpt, 200, "gpt2", 1500, 0.7, 1.5))
+        else:
+            previous = " ".join(x for x in content)
+            prmpt = previous
+            content.append(complete_with_gpt(prmpt, 200, "gpt2", 1500, 0.7, 1.5))
+
+    st.write(content)
+    filenamee = to_pdf(convert(create_md(text=content, title=title)))
+    with open(filenamee, "rb") as pdf_file:
+        PDFbyte = pdf_file.read()
+
+    st.download_button(
+        label="Download Book",
+        data=PDFbyte,
+        file_name=filenamee,
+        mime="application/octet-stream",
+    )

constants.py

@@ -0,0 +1,216 @@
+# Educational values
+AGE_RANGE = [3, 4, 5]
+SKILL_DEVELOPMENT = [
+    "Curiosity",
+    "Imagination",
+    "Communication",
+    "Observation",
+    "Adaptability",
+    "Persistence",
+    "Collaboration",
+    "Emotional Intelligence",
+    "Self-awareness",
+    "Memory",
+]
+LEARNING_OBJECTIVES = [
+    "Science exploration",
+    "Nature observation",
+    "History and social studies",
+    "Language development",
+    "Math concepts",
+    "Fine motor skills" "Gross motor skills",
+]
+
+# emotionl values
+THEME = [
+    "Empathy",
+    "Kindness",
+    "Friendship",
+    "Courage",
+    "Diversity",
+    "Imagination",
+    "Perseverance",
+    "Self-confidence",
+    "Responsibility" "Gratitude",
+]
+MODD_OF_STORY = [
+    "Lighthearted",
+    "Funny",
+    "Adventure",
+    "Comforting",
+    "Calm",
+    "Educational",
+    "Inspiring",
+]
+POSITIVE_MESSAGNG = [
+    "Kindness and compassion",
+    "Friendship and cooperation",
+    "Respect for others and diversity",
+    "Self-confidence and self-esteem",
+    "Perseverance and resilience",
+    "Honesty and integrity",
+    "Responsibility and accountability",
+    "Empathy and understanding",
+    "Mindfulness and emotional regulation",
+    "Creativity and imagination",
+]
+
+# Personal
+GENDER = ["Male", "Female"]
+GENRE = [
+    "Fairy tales",
+    "Biography",
+    "Animal stories",
+    "Adventure stories",
+    "Rhyming books",
+    "Concept books",
+    "Folktales",
+    "Nursery rhymes",
+    "Science fiction",
+    "Humorous books",
+]
+TOPIC = [
+    "Animals (fiction and non-fiction)",
+    "Friendship and relationships",
+    "Family and home life",
+    "Bedtime stories",
+    "Colors and shapes",
+    "Numbers and counting",
+    "Alphabet and phonics",
+    "Fairy tales and nursery rhymes",
+    "Adventure and exploration",
+    "Imagination and creativity",
+]
+
+TYPE_OF_MAIN_CHARACTER = [
+    "Human",
+    "Monkey",
+    "Cat",
+    "Caterpillar",
+    "Dog",
+    "Teddy bear",
+    "Elephant",
+    "Lions",
+    "Penguins",
+    "Monkeys",
+    "Ducks",
+    "Frogs",
+    "Giraffes",
+    "Tigers",
+    "Whales",
+    "Zebras",
+    "Superhero",
+    "Dinosaur",
+    "Robot",
+    "Aliens",
+    "Dragons",
+    "Pirates",
+    "Princess",
+    "Fairy",
+    "Dragons",
+    "Cyclops",
+    "Yeti",
+    "Bigfoot",
+    "Goblins",
+    "Trolls",
+    "Ghosts",
+    "Fire trucks",
+    "Trains",
+    "Race cars",
+    "Construction vehicle",
+    "Airplanes",
+    "Helicopters",
+    "Boats",
+    "Spaceships",
+    "Motorcycles",
+    "Buses",
+]
+ANTAGONIST_TYPE = [
+    "Witch",
+    "Wizard",
+    "Evil queen",
+    "Evil king",
+    "Monster",
+    "Pirate",
+    "Giant",
+    "Robot",
+    "Evil machine",
+    "Human",
+    "Monkey",
+    "Cat",
+    "Caterpillar",
+    "Dog",
+    "Teddy bear",
+    "Elephant",
+    "Lions",
+    "Penguins",
+    "Monkeys",
+    "Ducks",
+    "Frogs",
+    "Giraffes",
+    "Tigers",
+    "Whales",
+    "Zebras",
+    "Superhero",
+    "Dinosaur",
+    "Robot",
+    "Aliens",
+    "Dragons",
+    "Pirates",
+    "Princess",
+    "Fairy",
+    "Dragons",
+    "Cyclops",
+    "Yeti",
+    "Bigfoot",
+    "Goblins",
+    "Trolls",
+    "Ghosts",
+    "Fire trucks",
+    "Trains",
+    "Race cars",
+    "Construction vehicle",
+    "Airplanes",
+    "Helicopters",
+    "Boats",
+    "Spaceships",
+    "Motorcycles",
+    "Buses",
+]
+SUPPORTING_CHARACTER_TYPE = ANTAGONIST_TYPE
+SETTINGS = [
+    "Forests",
+    "Underwater",
+    "Imaginary world",
+    "Cityscapes",
+    "Countryside",
+    "Outer space",
+    "Castle",
+    "School",
+    "Magical kingdom",
+    "Enchanted forest",
+    "Circus",
+    "City",
+]
+RESOLUTION = [
+    "Apologizing and making amends",
+    "Learning to share and take turns",
+    "Standing up for oneself and others",
+    "Overcoming fear and facing challenges",
+    "Solving a problem or mystery",
+    "Learning a lesson or moral value",
+    "Forgiving and moving on",
+    "Accepting differences and celebrating diversity",
+    "Helping others and showing kindness",
+    "Making a new friend or finding common ground",
+    "Discovering a new passion or talent",
+    "Resolving a conflict peacefully and fairly",
+    "Making a sacrifice for the greater good",
+    "Learning from mistakes and taking responsibility",
+    "Understanding the consequences of one's actions",
+    "Finding a creative or innovative solution",
+    "Embracing change and adapting to new situations",
+    "Discovering the importance of teamwork and cooperation",
+    "Making a tough decision and sticking to it",
+    "Embracing the power of imagination and creativity",
+]

generator.py

@@ -0,0 +1,34 @@
+from transformers import pipeline
+
+
+def generate(text, the_model, max_length, temperature, repetition_penalty):
+    generator = pipeline("text-generation", model=the_model)
+    result = generator(
+        text,
+        num_return_sequences=1,
+        max_length=max_length,
+        temperature=temperature,
+        repetition_penalty=repetition_penalty,
+        no_repeat_ngram_size=2,
+        early_stopping=False,
+    )
+    return result[0]["generated_text"]
+
+
+def complete_with_gpt(
+    text, context, the_model, max_length, temperature, repetition_penalty
+):
+    # Use the last [context] characters of the text as context
+    max_length = max_length + context
+    return generate(
+        text[-context:], the_model, max_length, temperature, repetition_penalty
+    )
+
+
+# complete_with_gpt("I am a language model",200,'gpt2-medium',1500,0.7,1.5)
+# def generate(genre:str,topic:str,main_name:str,type_of_main_character:str,antagonist_name:str,
+#              antagonist_type:str,supporting_character_name:str,supporting_char_type:str,settings:str,
+#              resoluton:str,chapter:int
+#              ):
+#     chapter_text={}
+#     for x in range(chapter):0

prompts.py

@@ -0,0 +1,46 @@
+def get_initial_prompts(
+    key,
+    type_of_main_character,
+    main_character_name,
+    skill_development,
+    learning_objectives,
+    theme,
+    topic,
+):
+
+    prompts = {
+        "Fairy tales": f"I am a land far far away there was a {type_of_main_character} who is famous by the name {main_character_name} . {main_character_name} always had a  good{skill_development} and the story revolves around {learning_objectives} ,once can learn about {theme}. This is a famous topic consisting of {topic} for childrens "
+    }
+
+    return prompts.get(key)
+
+
+def story_setting_prompt(
+    key,
+    type_of_main_character,
+    main_character_name,
+    skill_development,
+    learning_objectives,
+    theme,
+    mood_of_story,
+    ant_name,
+    ant_type,
+):
+    prompts = {
+        "Fairy tales": f"{main_character_name} was always inclined towards {mood_of_story} but there was a {ant_type} famoulsy known as {ant_name} the {ant_type} who did not like {main_character_name}"
+    }
+    return prompts.get(key)
+
+
+def supporting_character_inclusion(key, sup_name, sup_type, positive_messaging):
+    prompts = {
+        "Fairy tales": f"Our hero who was always a supporter of {positive_messaging} had an friend named {sup_name} who was a {sup_type}"
+    }
+    return prompts.get(key)
+
+
+def ending_scene(key, resolution, main_char, positive_mess):
+    prompts = {
+        "Fairy tales": f"So this concludes that {main_char} was focused on getting {positive_mess} out for the world which resulted in {resolution}"
+    }
+    return prompts.get(key)

requirements.txt

@@ -0,0 +1,6 @@
+transformers
+streamlit
+markdown
+pdfkit
+pandas
+stqdm

utils.py

@@ -0,0 +1,54 @@
+import pandas as pd
+import random
+import markdown
+import pdfkit
+import platform
+
+
+def get_book_genre(book_name: str) -> str:
+
+    df = pd.read_csv("Goodreads_books_with_genres.csv")
+    genre = df[df.Title.str.contains(book_name)]["genres"].values.tolist()[0].split(";")
+    genre = genre[random.randint(0, len(genre))]
+    return genre
+
+
+def create_md(text, title):
+    with open(f"{title}.md", "w") as f:
+        f.write(f"# Title: {title}")
+        f.write(r"\n")
+        for x in range(len(text)):
+            f.write("\n")
+            f.write(f"## Chapter {x+1}")
+            f.write("\n")
+            f.write(text[x])
+    return title
+
+
+def convert(filename):
+    with open(f"{filename}.md", "r") as f:
+        text = f.read()
+        html = markdown.markdown(text)
+
+    with open(f"{filename}.html", "w", encoding="utf-8") as f:
+        f.write(html)
+    return f"{filename}"
+
+
+def to_pdf(filename):
+    if platform.system() == "Windows":
+        path_to_wkhtmltopdf = r"C:\Program Files\wkhtmltopdf\bin\wkhtmltopdf.exe"
+
+        # Point pdfkit configuration to wkhtmltopdf.exe
+        config = pdfkit.configuration(wkhtmltopdf=path_to_wkhtmltopdf)
+
+        # Convert HTML file to PDF
+        pdfkit.from_file(
+            f"{filename}.html", output_path=f"{filename}.pdf", configuration=config
+        )
+        return f"{filename}.pdf"
+    else:
+        pdfkit.from_file(
+            f"{filename}.html", output_path=f"{filename}.pdf", configuration=config
+        )
+        return f"{filename}.pdf"