Spaces:
Build error
Build error
File size: 5,995 Bytes
1b3748a 92ba273 11d601a 17ba82d 1b3748a 5b9d6f1 1b3748a 17ba82d 1b3748a 11d601a 1b3748a 11d601a 1b3748a 92ba273 1b3748a 92ba273 1b3748a 11d601a 1b3748a ea8e80f 1b3748a 35f46a2 11d601a 1b3748a 35f46a2 11d601a 1b3748a 35f46a2 1b3748a 11d601a 1b3748a 11d601a 35f46a2 1b3748a 35f46a2 1b3748a 11d601a 35f46a2 1b3748a 35f46a2 1b3748a 11d601a 35f46a2 1b3748a 35f46a2 1b3748a 11d601a 35f46a2 1b3748a 35f46a2 1b3748a 35f46a2 |
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 156 157 158 159 |
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 with a bit of styling
st.markdown("<h1 style='text-align: center; color: #4CAF50;'>StoryScribe</h1>", unsafe_allow_html=True)
# App Description
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.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")
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)
# Game Environment
with col1:
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.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.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.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)
|