Update app.py

app.py

@@ -1,4 +1,93 @@
 import streamlit as st
+import anthropic
+
+# Initialize the Anthropic client
+client = anthropic.Anthropic()
+
+# Define the functions to generate content
+def generate_game_environment(environment_description):
+    message = client.messages.create(
+        model="claude-3-5-sonnet-20240620",
+        max_tokens=1000,
+        temperature=0.7,
+        system="You are an expert in world-building. Generate a detailed description of a game environment based on the input.",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": f"Create a detailed description of a game environment based on this input: {environment_description}"
+                    }
+                ]
+            }
+        ]
+    )
+    return message.content
+
+def generate_protagonist(protagonist_description):
+    message = client.messages.create(
+        model="claude-3-5-sonnet-20240620",
+        max_tokens=1000,
+        temperature=0.7,
+        system="You are an expert in character creation. Generate a detailed description of a game protagonist based on the input.",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": f"Create a detailed description of a game protagonist based on this input: {protagonist_description}"
+                    }
+                ]
+            }
+        ]
+    )
+    return message.content
+
+def generate_antagonist(antagonist_description):
+    message = client.messages.create(
+        model="claude-3-5-sonnet-20240620",
+        max_tokens=1000,
+        temperature=0.7,
+        system="You are an expert in villain creation. Generate a detailed description of a game antagonist based on the input.",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": f"Create a detailed description of a game antagonist based on this input: {antagonist_description}"
+                    }
+                ]
+            }
+        ]
+    )
+    return message.content
+
+def generate_game_story(environment, protagonist, antagonist):
+    story_prompt = (f"Create a detailed game story based on the following inputs:\n"
+                    f"Game Environment: {environment}\n"
+                    f"Protagonist: {protagonist}\n"
+                    f"Antagonist: {antagonist}")
+    message = client.messages.create(
+        model="claude-3-5-sonnet-20240620",
+        max_tokens=1000,
+        temperature=0.7,
+        system="You are a master storyteller. Generate a detailed game story based on the inputs provided.",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": story_prompt
+                    }
+                ]
+            }
+        ]
+    )
+    return message.content
 
 # App Title
 st.title("StoryForge")
@@ -12,22 +101,46 @@ with st.sidebar:
     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")
+    if st.button("Generate Document"):
+        # Generate content based on user input
+        env_description = generate_game_environment(game_environment)
+        protagonist_description = generate_protagonist(protagonist)
+        antagonist_description = generate_antagonist(antagonist)
+        game_story = generate_game_story(game_environment, protagonist, antagonist)
+        
+        # Display generated content
+        st.session_state.env_description = env_description
+        st.session_state.protagonist_description = protagonist_description
+        st.session_state.antagonist_description = antagonist_description
+        st.session_state.game_story = game_story
 
 # Layout with two columns
 col1, col2 = st.columns(2)
 
 with col1:
     st.header("Game Environment")
-    st.write(game_environment)
+    if 'env_description' in st.session_state:
+        st.write(st.session_state.env_description)
+    else:
+        st.write(game_environment)
 
 with col2:
     st.header("Game Story")
-    st.write("Your game story will be generated based on the inputs provided.")
+    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.")
 
 with col1:
     st.header("Protagonist")
-    st.write(protagonist)
+    if 'protagonist_description' in st.session_state:
+        st.write(st.session_state.protagonist_description)
+    else:
+        st.write(protagonist)
 
 with col2:
     st.header("Antagonist")
-    st.write(antagonist)
+    if 'antagonist_description' in st.session_state:
+        st.write(st.session_state.antagonist_description)
+    else:
+        st.write(antagonist)