Update app.py

app.py

@@ -1,27 +1,27 @@
 import streamlit as st
 
 # Define the function that generates the game design document
-def generate_document(game_environments, protagonist, antagonist):
+def generate_document(game_environments, protagonist, antagonist, game_story):
     return (
         f"**Game Environments:**\n{game_environments}\n\n"
         f"**Protagonist:**\n{protagonist}\n\n"
         f"**Antagonist:**\n{antagonist}\n\n"
-        f"**Game Story:**\n"
-        "To be added based on the environments, protagonist, and antagonist."
+        f"**Game Story:**\n{game_story}"
     )
 
 # Streamlit app
 def main():
     st.title("StoryForge")
-    
+
     st.write(
         "StoryForge is your ultimate companion for crafting comprehensive game design documents. "
         "By providing structured inputs about game environments, protagonists, and antagonists, "
         "StoryForge helps you organize your ideas into a clear and cohesive design document."
     )
-    
+
     st.header("Input Details")
-    
+
+    # Input fields for the user
     game_environments = st.text_area(
         "Game Environments",
         "Enter the different settings or worlds where the game takes place."
@@ -37,14 +37,35 @@ def main():
         "Detail the main antagonist of the game."
     )
     
+    game_story = st.text_area(
+        "Game Story",
+        "Provide an overview of the game's narrative arc."
+    )
+
     if st.button("Generate Document"):
-        if game_environments and protagonist and antagonist:
-            document = generate_document(game_environments, protagonist, antagonist)
+        if game_environments and protagonist and antagonist and game_story:
+            document = generate_document(game_environments, protagonist, antagonist, game_story)
             st.subheader("Generated Game Design Document")
-            st.markdown(document)
+            
+            # Two-column layout
+            col1, col2 = st.columns(2)
+
+            with col1:
+                st.write("**Game Environments:**")
+                st.write(game_environments)
+                st.write("**Protagonist:**")
+                st.write(protagonist)
+            
+            with col2:
+                st.write("**Antagonist:**")
+                st.write(antagonist)
+                st.write("**Game Story:**")
+                st.write(game_story)
+                
         else:
             st.error("Please fill out all fields.")
 
 if __name__ == "__main__":
     main()
 
+