Update app.py

app.py

@@ -1,104 +1,154 @@
+import os
 import streamlit as st
 import anthropic
-import os
 from dotenv import load_dotenv
 
 # Load environment variables from .env file
 load_dotenv()
 
-# API keys
-claude_api_key = os.getenv("ANTHROPIC_API_KEY")
-if claude_api_key is None:
-    raise ValueError("Claude API Key is not set. Please check your .env file.")
+# 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)
 
-print(f"Claude API Key: {claude_api_key}")  # Debugging line to check if the key is loaded
+# 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
 
-# Initialize Claude AI client
-client = anthropic.Anthropic(api_key=claude_api_key)
+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
 
-# 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}"
-        ),
+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": prompt}]}
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": f"Create a detailed description of a game antagonist based on this input: {antagonist_description}"
+                    }
+                ]
+            }
         ]
     )
-    return response.content[0].text
+    return message.content[0].text
 
-# Main Function
-def main():
-    st.set_page_config(page_title="StoryForge", layout="wide")
-    st.title("StoryForge")
-    st.markdown(
-        """
-        **StoryForge** helps game developers generate comprehensive Game Design Documents. 
-        Input details about your game environment, protagonist, and antagonist to create a structured design document.
-        """
+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")
 
-    # Sidebar for user input
-    with st.sidebar:
-        st.header("Game Details")
-        image_description = st.text_area(
-            "Game Environment", 
-            placeholder="Describe the setting of your game",
-            height=100
-        )
-        protagonist_description = st.text_area(
-            "Protagonist", 
-            placeholder="Describe the main character",
-            height=100
-        )
-        antagonist_description = st.text_area(
-            "Antagonist", 
-            placeholder="Describe the main villain or opposing force",
-            height=100
-        )
-        generate_text_btn = st.button("Generate Document")
+# 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.")
 
-    col1, col2 = st.columns(2)
+# 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
 
-    with col1:
-        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)
+# Layout with two columns
+col1, col2 = st.columns(2)
 
-        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)
+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.subheader("Game 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.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.")
 
-        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 col1:
+    st.header("Protagonist")
+    if 'protagonist_description' in st.session_state:
+        st.write(st.session_state.protagonist_description)
+    else:
+        st.write(protagonist)
 
-# Run Main Function 
-if __name__ == "__main__":
-    main()
+with col2:
+    st.header("Antagonist")
+    if 'antagonist_description' in st.session_state:
+        st.write(st.session_state.antagonist_description)
+    else:
+        st.write(antagonist)