Update app.py

app.py

@@ -97,15 +97,15 @@ def generate_game_story(environment, protagonist, antagonist):
     )
     return message.content[0].text
 
-# App Title
-st.title("StoryScribe")
+# App Title with a bit of styling
+st.markdown("<h1 style='text-align: center; color: #4CAF50;'>StoryScribe</h1>", unsafe_allow_html=True)
 
 # App Description
-st.write("StoryForge helps game developers generate comprehensive Game Design Documents. Input details about your game environment, protagonist, and antagonist to create a structured design document.")
+st.markdown("<p style='text-align: center; color: #888;'>StoryForge helps game developers generate comprehensive Game Design Documents. Input details about your game environment, protagonist, and antagonist to create a structured design document.</p>", unsafe_allow_html=True)
 
 # Sidebar Inputs
 with st.sidebar:
-    st.header("Game Details")
+    st.markdown("<h2 style='color: #FF6347;'>Game Details</h2>", unsafe_allow_html=True)
     game_environment = st.text_input("Game Environment", "Describe the setting of your game")
     protagonist = st.text_input("Protagonist", "Describe the main character")
     antagonist = st.text_input("Antagonist", "Describe the main villain or opposing force")
@@ -125,30 +125,34 @@ with st.sidebar:
 # Layout with two columns
 col1, col2 = st.columns(2)
 
+# Game Environment
 with col1:
-    st.header("Game Environment")
+    st.markdown("<h2 style='color: #007BFF;'>Game Environment</h2>", unsafe_allow_html=True)
     if 'env_description' in st.session_state:
         st.write(st.session_state.env_description)
     else:
         st.write(game_environment)
 
+# Game Story
 with col2:
-    st.header("Game Story")
+    st.markdown("<h2 style='color: #007BFF;'>Game Story</h2>", unsafe_allow_html=True)
     if 'game_story' in st.session_state:
         st.write(st.session_state.game_story)
     else:
         st.write("Your game story will be generated based on the inputs provided.")
 
+# Protagonist
 with col1:
-    st.header("Protagonist")
+    st.markdown("<h2 style='color: #FF4500;'>Protagonist</h2>", unsafe_allow_html=True)
     if 'protagonist_description' in st.session_state:
         st.write(st.session_state.protagonist_description)
     else:
         st.write(protagonist)
 
+# Antagonist
 with col2:
-    st.header("Antagonist")
+    st.markdown("<h2 style='color: #FF4500;'>Antagonist</h2>", unsafe_allow_html=True)
     if 'antagonist_description' in st.session_state:
         st.write(st.session_state.antagonist_description)
     else:
-        st.write(antagonist)
+        st.write(antagonist)