Update app.py

app.py

@@ -1,101 +1,69 @@
 import os
 import streamlit as st
-import anthropic
 from dotenv import load_dotenv
-
+from groq import Groq
 # Load environment variables from .env file
 load_dotenv()
 
 # Retrieve the API key from environment variables
-api_key = os.getenv("Claude_api_key")
+api_key = os.getenv("groq_api_key")
 
 # Initialize the Anthropic client with the API key
-client = anthropic.Anthropic(api_key=api_key)
+client = Groq(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.",
+    message = client.chat.completions.create(
         messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": f"Create a detailed description of a game environment based on this input: {environment_description}"
-                    }
-                ]
-            }
-        ]
+        {
+            "role": "user",
+            "content": f"You are an expert in world-building. Generate a detailed description of a game environment based on the input. Create a detailed description of a game environment based on this input: {environment_description}",
+        }
+        ],
+        model="llama-3.3-70b-versatile",
     )
-    return message.content[0].text
+    return message.choices[0].message.content
 
 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.",
+    message = client.chat.completions.create(
         messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": f"Create a detailed description of a game protagonist based on this input: {protagonist_description}"
-                    }
-                ]
-            }
-        ]
+        {
+            "role": "user",
+            "content": f"You are an expert in character creation. Generate a detailed description of a game protagonist based on the input. Create a detailed description of a game protagonist based on this input: {protagonist_description}",
+        }
+        ],
+        model="llama-3.3-70b-versatile",
     )
-    return message.content[0].text
+    return message.choices[0].message.content
 
 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.",
+    message = client.chat.completions.create(
         messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": f"Create a detailed description of a game antagonist based on this input: {antagonist_description}"
-                    }
-                ]
-            }
-        ]
+        {
+            "role": "user",
+            "content": f"You are an expert in villain creation. Generate a detailed description of a game antagonist based on the input. Create a detailed description of a game antagonist based on this input: {antagonist_description}.",
+        }
+        ],
+        model="llama-3.3-70b-versatile",
     )
-    return message.content[0].text
+    return message.choices[0].message.content
 
 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.",
+
+    message = client.chat.completions.create(
         messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": story_prompt
-                    }
-                ]
-            }
-        ]
+        {
+            "role": "user",
+            "content": f"You are a master storyteller. Generate a detailed game story based on the inputs provided.\n {story_prompt}",
+        }
+        ],
+        model="llama-3.3-70b-versatile",
     )
-    return message.content[0].text
+    return message.choices[0].message.content
 
 # App Title
 st.title("StoryForge")