Update app.py

app.py

@@ -1,12 +1,76 @@
 import os
 import streamlit as st
+from dotenv import load_dotenv
 from groq import Groq
 
-# Initialize Groq client
+# Load environment variables from .env file
+load_dotenv()
+
+# Fetch API key from environment variables
+api_key = os.getenv("groq_api_key")
+
+# Initialize Groq client using the API key from .env
 client = Groq(
-    api_key=os.environ.get("GROQ_API_KEY"),
+    api_key=api_key,  # API key fetched securely from .env
 )
 
+# Function to generate the game environment
+def generate_game_environment(environment):
+    if not environment:
+        return "Please provide a valid game environment."
+    
+    message = f"Describe the setting of a game in the environment: {environment}."
+    
+    chat_completion = client.chat.completions.create(
+        messages=[{"role": "user", "content": message}],
+        model="llama3-8b-8192",
+    )
+    
+    return chat_completion.choices[0].message.content
+
+# Function to generate the protagonist
+def generate_protagonist(protagonist):
+    if not protagonist:
+        return "Please provide a valid protagonist."
+    
+    message = f"Create a detailed description of the protagonist: {protagonist}."
+    
+    chat_completion = client.chat.completions.create(
+        messages=[{"role": "user", "content": message}],
+        model="llama3-8b-8192",
+    )
+    
+    return chat_completion.choices[0].message.content
+
+# Function to generate the antagonist
+def generate_antagonist(antagonist):
+    if not antagonist:
+        return "Please provide a valid antagonist."
+    
+    message = f"Create a detailed description of the antagonist: {antagonist}."
+    
+    chat_completion = client.chat.completions.create(
+        messages=[{"role": "user", "content": message}],
+        model="llama3-8b-8192",
+    )
+    
+    return chat_completion.choices[0].message.content
+
+# Function to generate the game story
+def generate_game_story(environment, protagonist, antagonist):
+    if not (environment and protagonist and antagonist):
+        return "Please provide all the required details (environment, protagonist, antagonist)."
+    
+    message = f"Create a compelling game story set in {environment}, featuring {protagonist} as the protagonist and {antagonist} as the antagonist."
+    
+    chat_completion = client.chat.completions.create(
+        messages=[{"role": "user", "content": message}],
+        model="llama3-8b-8192",
+    )
+    
+    return chat_completion.choices[0].message.content
+
+
 # Streamlit app title and description
 st.title("StoryForge")
 st.write("StoryForge is an app designed to help game developers generate comprehensive Game Design Documents by providing essential details about their game's environment, protagonist, and antagonist.")
@@ -20,37 +84,30 @@ antagonist = st.sidebar.text_input("Enter the Antagonist:", placeholder="e.g., A
 # Main content layout
 col1, col2 = st.columns(2)
 
-# Column 1
+# Column 1: Environment and Story
 with col1:
     st.header("Game Environment")
-    st.write(game_environment)
+    if game_environment:
+        st.write(generate_game_environment(game_environment))
+    else:
+        st.write("Please provide a game environment in the sidebar.")
 
     st.header("Game Story")
     if game_environment and protagonist and antagonist:
-        # Create a message for Groq to generate a game story based on user inputs
-        message = f"Create a compelling game story set in {game_environment}, featuring {protagonist} as the protagonist and {antagonist} as the antagonist."
-        
-        # Get completion from Groq
-        chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": message,
-                }
-            ],
-            model="llama3-8b-8192",
-        )
-        
-        # Return generated story content
-        story = chat_completion.choices[0].message.content
-        st.write(story)
+        st.write(generate_game_story(game_environment, protagonist, antagonist))
     else:
         st.write("Please provide details in the sidebar.")
 
-# Column 2
+# Column 2: Protagonist and Antagonist
 with col2:
     st.header("Protagonist")
-    st.write(protagonist)
+    if protagonist:
+        st.write(generate_protagonist(protagonist))
+    else:
+        st.write("Please provide a protagonist in the sidebar.")
 
     st.header("Antagonist")
-    st.write(antagonist)
+    if antagonist:
+        st.write(generate_antagonist(antagonist))
+    else:
+        st.write("Please provide an antagonist in the sidebar.")