shaheerawan3 commited on
Commit
d568509
·
verified ·
1 Parent(s): 9a8bff9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -2
app.py CHANGED
@@ -289,9 +289,17 @@ class EnhancedPDFFormatter(FPDF):
289
  return None
290
 
291
  def initialize_session_state():
292
- """Initialize enhanced session state"""
293
  if 'book_settings' not in st.session_state:
294
- st.session_state.book_settings = {}
 
 
 
 
 
 
 
 
295
  if 'generated_book' not in st.session_state:
296
  st.session_state.generated_book = None
297
  if 'current_step' not in st.session_state:
@@ -300,9 +308,14 @@ def initialize_session_state():
300
  st.session_state.user_prompt = ""
301
  if 'author_name' not in st.session_state:
302
  st.session_state.author_name = ""
 
 
303
 
304
  def main():
305
  st.set_page_config(page_title="AI Book Generator", layout="wide")
 
 
 
306
 
307
  # Book templates
308
  book_templates = {
@@ -346,6 +359,7 @@ def main():
346
  ["Book Settings", "Generate Content", "Preview & Download"]
347
  )
348
  st.session_state.current_step = selected_step
 
349
  st.title("AI Book Generator")
350
 
351
  # Main content area
 
289
  return None
290
 
291
  def initialize_session_state():
292
+ """Initialize enhanced session state with default values"""
293
  if 'book_settings' not in st.session_state:
294
+ st.session_state.book_settings = {
295
+ 'title': '',
296
+ 'template': '',
297
+ 'genre': '',
298
+ 'num_chapters': 0,
299
+ 'words_per_chapter': 0,
300
+ 'chapter_titles': [],
301
+ 'total_words': 0
302
+ }
303
  if 'generated_book' not in st.session_state:
304
  st.session_state.generated_book = None
305
  if 'current_step' not in st.session_state:
 
308
  st.session_state.user_prompt = ""
309
  if 'author_name' not in st.session_state:
310
  st.session_state.author_name = ""
311
+ if 'book_generator' not in st.session_state:
312
+ st.session_state.book_generator = EnhancedBookGenerator()
313
 
314
  def main():
315
  st.set_page_config(page_title="AI Book Generator", layout="wide")
316
+
317
+ # Initialize session state at the start
318
+ initialize_session_state()
319
 
320
  # Book templates
321
  book_templates = {
 
359
  ["Book Settings", "Generate Content", "Preview & Download"]
360
  )
361
  st.session_state.current_step = selected_step
362
+
363
  st.title("AI Book Generator")
364
 
365
  # Main content area