Update app.py

app.py

@@ -1,80 +1,106 @@
-  import streamlit as st
-  import anthropic
-  import os
-  from dotenv import load_dotenv
+import streamlit as st
+import anthropic
+import os
+from dotenv import load_dotenv
 
-  # Load environment variables
-  load_dotenv()
+# Load environment variables
+load_dotenv()
 
-  # API keys
-  claude_api_key = os.getenv("ANTHROPIC_API_KEY")
+# API keys
+claude_api_key = os.getenv("ANTHROPIC_API_KEY")
 
-  # Initialize Claude AI client
-  client = anthropic.Anthropic(api_key=claude_api_key)
+# Initialize Claude AI client
+client = anthropic.Anthropic(api_key=claude_api_key)
 
-  # Function to generate text using Claude AI
-  def generate_text(prompt, model_name="claude-3-opus-20240229"):
-      response = client.messages.create(
-          model=model_name,
-          max_tokens=500,
-          temperature=0.5,
-          system=f"You are a creative writer. Based on the following description, generate content for a game design document: {prompt}",
-          messages=[
-              {"role": "user", "content": [{"type": "text", "text": prompt}]}
-          ]
-      )
-      return response.content[0].text
+# Function to generate text using Claude AI
+def generate_text(prompt, model_name="claude-3-opus-20240229"):
+    response = client.messages.create(
+        model=model_name,
+        max_tokens=500,
+        temperature=0.5,
+        system=(
+            "You are a creative writer. Based on the following description, "
+            "generate content for a game design document: {prompt}"
+        ),
+        messages=[
+            {"role": "user", "content": [{"type": "text", "text": prompt}]}
+        ]
+    )
+    return response.content[0].text
 
-  # Main Function
-  def main():
-      st.set_page_config(page_title="StoryForge", layout="wide")
-      st.title("StoryForge")
+# Main Function
+def main():
+    st.set_page_config(page_title="StoryForge", layout="wide")
+    st.title("StoryForge")
 
-      # Sidebar for user input
-      with st.sidebar:
-          st.header("Controls")
-          image_description = st.text_area("Set the foundation for the game world by providing a detailed and imaginative description of the overall theme, setting, and gameplay elements.", height=100)
-          protagonist_description = st.text_area("Introduce the main character of the game. Describe their appearance, personality, strengths, and weaknesses. What makes them unique and interesting?", height=100)
-          antagonist_description = st.text_area("Describe the primary antagonist or enemy in the game. What are their motivations, abilities, and characteristics? How do they pose a challenge to the protagonist?", height=100)
+    # Sidebar for user input
+    with st.sidebar:
+        st.header("Controls")
+        image_description = st.text_area(
+            "Set the foundation for the game world by providing a detailed and imaginative description of the overall theme, setting, and gameplay elements.", 
+            height=100
+        )
+        protagonist_description = st.text_area(
+            "Introduce the main character of the game. Describe their appearance, personality, strengths, and weaknesses. What makes them unique and interesting?", 
+            height=100
+        )
+        antagonist_description = st.text_area(
+            "Describe the primary antagonist or enemy in the game. What are their motivations, abilities, and characteristics? How do they pose a challenge to the protagonist?", 
+            height=100
+        )
+        generate_text_btn = st.button("Generate Game Design")
 
-          generate_text_btn = st.button("Generate Game Design")
+    col1, col2 = st.columns(2)
 
-      col1, col2 = st.columns(2)
+    with col1:
+        st.markdown(
+            "*StoryForge* is a cutting-edge AI app designed to empower *game developers* and creators. "
+            "With StoryForge, you can effortlessly craft captivating game design documents. "
+            "From crafting immersive storylines and intricate plots to defining compelling protagonists and antagonists, "
+            "StoryForge unlocks your creative potential and streamlines the game design process."
+        )
 
-      with col1:
-          st.markdown("*StoryForge* is a cutting-edge AI app designed to empower *game developers* and creators. With StoryForge, you can effortlessly craft captivating game design documents. From crafting immersive storylines and intricate plots to defining compelling protagonists and antagonists, StoryForge unlocks your creative potential and streamlines the game design process.")
+        st.subheader("Game Environment")
+        if generate_text_btn and image_description:
+            with st.spinner("Generating game environment description..."):
+                generated_environment = generate_text(
+                    f"Generate a detailed description of a game environment based on this theme and setting: {image_description}"
+                )
+                st.markdown(generated_environment)
 
-          st.subheader("Game Environment")
-          if generate_text_btn and image_description:
-              with st.spinner("Generating game environment description..."):
-                  generated_environment = generate_text(f"Generate a detailed description of a game environment based on this theme and setting: {image_description}")
-                  st.markdown(generated_environment)
+        st.subheader("Antagonist")
+        if generate_text_btn and antagonist_description:
+            with st.spinner("Generating antagonist description..."):
+                generated_antagonist = generate_text(
+                    f"Generate a detailed description of a game antagonist based on this description: {antagonist_description}"
+                )
+                st.markdown(generated_antagonist)
 
-          st.subheader("Antagonist")
-          if generate_text_btn and antagonist_description:
-              with st.spinner("Generating antagonist description..."):
-                  generated_antagonist = generate_text(f"Generate a detailed description of a game antagonist based on this description: {antagonist_description}")
-                  st.markdown(generated_antagonist)
+    with col2:
+        st.subheader("Story")
+        if generate_text_btn and image_description:
+            with st.spinner("Generating game story..."):
+                generated_story = generate_text(
+                    f"Create a creative and engaging game story that includes a protagonist, antagonist, and a challenge based on this description: {image_description}"
+                )
+                st.markdown(generated_story)
 
-      with col2:
-          st.subheader("Story")
-          if generate_text_btn and image_description:
-              with st.spinner("Generating game story..."):
-                  generated_story = generate_text(f"Create a creative and engaging game story that includes a protagonist, antagonist, and a challenge based on this description: {image_description}")
-                  st.markdown(generated_story)
+        st.subheader("Protagonist")
+        if generate_text_btn and protagonist_description:
+            with st.spinner("Generating protagonist description..."):
+                generated_protagonist = generate_text(
+                    f"Generate a detailed description of a game protagonist based on this description: {protagonist_description}"
+                )
+                st.markdown(generated_protagonist)
 
-          st.subheader("Protagonist")
-          if generate_text_btn and protagonist_description:
-              with st.spinner("Generating protagonist description..."):
-                  generated_protagonist = generate_text(f"Generate a detailed description of a game protagonist based on this description: {protagonist_description}")
-                  st.markdown(generated_protagonist)
+        st.subheader("Game Plot")
+        if generate_text_btn and image_description:
+            with st.spinner("Generating game plot..."):
+                generated_plot = generate_text(
+                    f"Generate a short game plot with a hook, gameplay relation, sticky mechanics, and setting based on this description: {image_description}"
+                )
+                st.markdown(generated_plot)
 
-          st.subheader("Game Plot")
-          if generate_text_btn and image_description:
-              with st.spinner("Generating game plot..."):
-                  generated_plot = generate_text(f"Generate a short game plot with a hook, gameplay relation, sticky mechanics, and setting based on this description: {image_description}")
-                  st.markdown(generated_plot)
-
-  # Run Main Function 
-  if __name__ == "__main__":
-      main()
+# Run Main Function 
+if __name__ == "__main__":
+    main()