Spaces:
Runtime error
Runtime error
| 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", | |
| ) | |