Update app.py

app.py

@@ -28,17 +28,17 @@ global_chapters = ""
 # Generation functions
 def generate_synopsis(topic):
     global global_synopsis
-    prompt = f"Write a synopsis for a story about {topic}: "
-    global_synopsis = generate_text(prompt, max_length=50)
+    prompt = f"Write a brief synopsis for a story about {topic}: "
+    global_synopsis = generate_text(prompt, max_length=100)
     return global_synopsis
 
 def generate_chapters():
     global global_synopsis, global_chapters
     if not global_synopsis:
         return "Please generate a synopsis first."
-    prompt = f'''Based on this synopsis for a book: {global_synopsis}, divide the story into 6 chapters with brief descriptions for each. 
-    Numerate every chapter created and make the first sound like an introduction and the last may sound as the epilogue'''
-    global_chapters = generate_text(prompt, max_length=150)
+    prompt = f'''Based on this synopsis for a book: {global_synopsis}. Divide the story into 4 chapters with brief descriptions for each. 
+    Enumerate every chapter created followed by its description and make the first chapter sound like an introduction and the last may sound as the epilogue'''
+    global_chapters = generate_text(prompt, max_length=2000)
     return global_chapters
 
 def expand_chapter(chapter_number):
@@ -48,7 +48,8 @@ def expand_chapter(chapter_number):
     chapters = global_chapters.split("\n")
     if chapter_number <= 0 or chapter_number > len(chapters):
         return f"Select a number between 1 and {len(chapters)}."
-    prompt = f"Expand and describe Chapter {chapter_number} in more detail: {chapters[chapter_number - 1]}"
+    prompt = f'''Knowing this synopsis for a book: {global_synopsis}. Expand and describe Chapter {chapter_number} 
+    in more detail, the title and current brief description of this chapter is: {chapters[chapter_number - 1]}'''
     return generate_text(prompt, max_length=200)
 
 # Gradio interface