Update app.py

app.py

@@ -1,14 +1,22 @@
+import os
 import streamlit as st
 import anthropic
+from dotenv import load_dotenv
 
-# Initialize the Anthropic client
-client = anthropic.Anthropic()
+# 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=1000,
+        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=[
@@ -23,12 +31,12 @@ def generate_game_environment(environment_description):
             }
         ]
     )
-    return message.content
+    return message.content[0].text
 
 def generate_protagonist(protagonist_description):
     message = client.messages.create(
         model="claude-3-5-sonnet-20240620",
-        max_tokens=1000,
+        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=[
@@ -43,12 +51,12 @@ def generate_protagonist(protagonist_description):
             }
         ]
     )
-    return message.content
+    return message.content[0].text
 
 def generate_antagonist(antagonist_description):
     message = client.messages.create(
         model="claude-3-5-sonnet-20240620",
-        max_tokens=1000,
+        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=[
@@ -63,7 +71,7 @@ def generate_antagonist(antagonist_description):
             }
         ]
     )
-    return message.content
+    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"
@@ -72,7 +80,7 @@ def generate_game_story(environment, protagonist, antagonist):
                     f"Antagonist: {antagonist}")
     message = client.messages.create(
         model="claude-3-5-sonnet-20240620",
-        max_tokens=1000,
+        max_tokens= 150,
         temperature=0.7,
         system="You are a master storyteller. Generate a detailed game story based on the inputs provided.",
         messages=[
@@ -87,7 +95,7 @@ def generate_game_story(environment, protagonist, antagonist):
             }
         ]
     )
-    return message.content
+    return message.content[0].text
 
 # App Title
 st.title("StoryForge")
@@ -108,7 +116,7 @@ with st.sidebar:
         antagonist_description = generate_antagonist(antagonist)
         game_story = generate_game_story(game_environment, protagonist, antagonist)
         
-        # Display generated content
+        # 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