syntax

app.py

@@ -23,25 +23,12 @@ def generate_story(user_input):
 def start_storytelling():
     print("Welcome to the Storytelling bot.")
 
-    story_prompt = input("What would you like your story to be about?")
+    story_prompt = st.text_input("What would you like your story to be about?")
 
 
     story_part = generate_story(story_prompt)
     print("\nHere's the beginning of your story:")
     print(story_part)
 
-    # Continuation of story (generate part two)
-    while True:
-        continue_story = input("\nDo you want to continue the story? (yes/no): ")
-
-        if continue_story.lower() == 'yes':
-            user_input = input("\nWhat should happen next?")
-            story_part = generate_story(user_input)
-            print("\nContinued:")
-            print(story_part)
-        else:
-            print("\nEnjoy!")
-            break
-
 # Function start
 start_storytelling()