Update app.py

app.py

@@ -31,7 +31,7 @@ def generate_game_environment(environment_description):
         ]
     )
     
-    return message['content']  # Return the relevant content
+    return message.content[0].text  # Return the relevant content
 
 def generate_game_story(environment, protagonist, antagonist):
     client = anthropic.Anthropic(api_key=API_KEY)
@@ -57,7 +57,7 @@ def generate_game_story(environment, protagonist, antagonist):
         ]
     )
     
-    return message['content']  # Return the relevant content
+    return message.content[0].text  # Return the relevant content
 
 def generate_protagonist(protagonist_description):
     client = anthropic.Anthropic(api_key=API_KEY)
@@ -80,7 +80,7 @@ def generate_protagonist(protagonist_description):
         ]
     )
     
-    return message['content']  # Return the relevant content
+    return message.content[0].text  # Return the relevant content
 
 def generate_antagonist(antagonist_description):
     client = anthropic.Anthropic(api_key=API_KEY)
@@ -103,7 +103,7 @@ def generate_antagonist(antagonist_description):
         ]
     )
     
-    return message['content']  # Return the relevant content
+    return message.content[0].text  # Return the relevant content
 
 # Sidebar Inputs
 st.sidebar.title("StoryForge Input")