Update app.py

app.py

@@ -1,11 +1,21 @@
+import os
+from dotenv import load_dotenv
 import gradio as gr
 import anthropic
 
-# Initialize the Anthropic client
-client = anthropic.Anthropic()
+# Load environment variables from .env file
+load_dotenv()
+
+# Get the API key from environment variables
+def get_client():
+    api_key = os.getenv("ANTHROPIC_API_KEY")
+    if not api_key:
+        raise ValueError("API key not found. Please set the ANTHROPIC_API_KEY environment variable.")
+    return anthropic.Anthropic(api_key=api_key)
 
 def generate_game_environment(description_prompt):
-    message = client.messages.create(
+    client = get_client()
+    response = client.messages.create(
         model="claude-3-5-sonnet-20240620",
         max_tokens=500,
         temperature=0.7,
@@ -22,10 +32,11 @@ def generate_game_environment(description_prompt):
             }
         ]
     )
-    return message.content
+    return response['choices'][0]['text'].strip()
 
 def generate_protagonist(description_prompt):
-    message = client.messages.create(
+    client = get_client()
+    response = client.messages.create(
         model="claude-3-5-sonnet-20240620",
         max_tokens=500,
         temperature=0.7,
@@ -42,10 +53,11 @@ def generate_protagonist(description_prompt):
             }
         ]
     )
-    return message.content
+    return response['choices'][0]['text'].strip()
 
 def generate_antagonist(description_prompt):
-    message = client.messages.create(
+    client = get_client()
+    response = client.messages.create(
         model="claude-3-5-sonnet-20240620",
         max_tokens=500,
         temperature=0.7,
@@ -62,7 +74,7 @@ def generate_antagonist(description_prompt):
             }
         ]
     )
-    return message.content
+    return response['choices'][0]['text'].strip()
 
 def generate_game_story(environment, protagonist, antagonist):
     description_prompt = (
@@ -72,7 +84,8 @@ def generate_game_story(environment, protagonist, antagonist):
         f"Antagonist: {antagonist}\n"
         "Please provide a narrative that ties these elements together in an engaging way."
     )
-    message = client.messages.create(
+    client = get_client()
+    response = client.messages.create(
         model="claude-3-5-sonnet-20240620",
         max_tokens=1000,
         temperature=0.7,
@@ -89,7 +102,7 @@ def generate_game_story(environment, protagonist, antagonist):
             }
         ]
     )
-    return message.content
+    return response['choices'][0]['text'].strip()
 
 def generate_game_design(environment, protagonist, antagonist):
     # Generate each part of the game design
@@ -127,6 +140,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-
-            
-