Update app.py

app.py

@@ -124,30 +124,31 @@ antagonist = st.sidebar.text_input(
 st.title("StoryForge")
 st.write("StoryForge is your tool for crafting comprehensive Game Design Documents. Enter your game's details, and StoryForge will generate a structured document to guide your development process.")
 
-# Generate the content when the user has filled in all fields
-if environment and protagonist and antagonist:
-    # Generate content
-    generated_environment = generate_game_environment(environment)
-    generated_protagonist = generate_protagonist(protagonist)
-    generated_antagonist = generate_antagonist(antagonist)
-    generated_story = generate_game_story(generated_environment, generated_protagonist, generated_antagonist)
-
-    # Columns for Outputs
-    col1, col2 = st.columns(2)
-
-    with col1:
-        st.subheader("Game Environment")
-        st.write(generated_environment)
-
-        st.subheader("Protagonist")
-        st.write(generated_protagonist)
-
-    with col2:
-        st.subheader("Game Story")
-        st.write(generated_story)
-
-        st.subheader("Antagonist")
-        st.write(generated_antagonist)
-else:
-    st.write("Please fill in all the fields in the sidebar to generate your Game Design Document.")
+# Add a Generate button
+if st.button("Generate"):
+    if environment and protagonist and antagonist:
+        # Generate content
+        generated_environment = generate_game_environment(environment)
+        generated_protagonist = generate_protagonist(protagonist)
+        generated_antagonist = generate_antagonist(antagonist)
+        generated_story = generate_game_story(generated_environment, generated_protagonist, generated_antagonist)
+
+        # Columns for Outputs
+        col1, col2 = st.columns(2)
+
+        with col1:
+            st.subheader("Game Environment")
+            st.write(generated_environment)
+
+            st.subheader("Protagonist")
+            st.write(generated_protagonist)
+
+        with col2:
+            st.subheader("Game Story")
+            st.write(generated_story)
+
+            st.subheader("Antagonist")
+            st.write(generated_antagonist)
+    else:
+        st.write("Please fill in all the fields in the sidebar to generate your Game Design Document.")