Spaces:
Sleeping
Sleeping
File size: 5,463 Bytes
9c0792c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 |
import os
import streamlit as st
import anthropic
from dotenv import load_dotenv
# Load environment variables from .env file
load_dotenv()
# Retrieve the API key from environment variables
api_key = os.getenv("Claude_api_key")
# Initialize the Anthropic client with the API key
client = anthropic.Anthropic(api_key=api_key)
# Define the functions to generate content
def generate_game_environment(environment_description):
message = client.messages.create(
model="claude-3-5-sonnet-20240620",
max_tokens=150,
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[0].text
def generate_protagonist(protagonist_description):
message = client.messages.create(
model="claude-3-5-sonnet-20240620",
max_tokens=150,
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[0].text
def generate_antagonist(antagonist_description):
message = client.messages.create(
model="claude-3-5-sonnet-20240620",
max_tokens=150,
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[0].text
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= 150,
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[0].text
# App Title
st.title("StoryForge")
# 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.")
# Sidebar Inputs
with st.sidebar:
st.header("Game Details")
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)
# Store results in session state
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")
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")
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")
if 'protagonist_description' in st.session_state:
st.write(st.session_state.protagonist_description)
else:
st.write(protagonist)
with col2:
st.header("Antagonist")
if 'antagonist_description' in st.session_state:
st.write(st.session_state.antagonist_description)
else:
st.write(antagonist)
|