Update app.py

app.py

@@ -16,58 +16,38 @@ client = Groq(
 
 # 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
 
 
@@ -81,33 +61,27 @@ game_environment = st.sidebar.text_input("Enter the Game Environment:", placehol
 protagonist = st.sidebar.text_input("Enter the Protagonist:", placeholder="e.g., A young hacker with a mysterious past")
 antagonist = st.sidebar.text_input("Enter the Antagonist:", placeholder="e.g., A corrupt government official controlling the city")
 
-# Main content layout
-col1, col2 = st.columns(2)
-
-# Column 1: Environment and Story
-with col1:
-    st.header("Game Environment")
-    if game_environment:
-        st.write(generate_game_environment(game_environment))
+# Generate button
+if st.sidebar.button("Generate"):
+    if not game_environment or not protagonist or not antagonist:
+        # If any of the fields are empty, show an error message
+        st.error("Please fill in all the fields (Game Environment, Protagonist, Antagonist) before generating the story.")
     else:
-        st.write("Please provide a game environment in the sidebar.")
+        # Main content layout
+        col1, col2 = st.columns(2)
 
-    st.header("Game Story")
-    if game_environment and protagonist and antagonist:
-        st.write(generate_game_story(game_environment, protagonist, antagonist))
-    else:
-        st.write("Please provide details in the sidebar.")
+        # Column 1: Environment and Story
+        with col1:
+            st.header("Game Environment")
+            st.write(generate_game_environment(game_environment))
 
-# Column 2: Protagonist and Antagonist
-with col2:
-    st.header("Protagonist")
-    if protagonist:
-        st.write(generate_protagonist(protagonist))
-    else:
-        st.write("Please provide a protagonist in the sidebar.")
+            st.header("Game Story")
+            st.write(generate_game_story(game_environment, protagonist, antagonist))
 
-    st.header("Antagonist")
-    if antagonist:
-        st.write(generate_antagonist(antagonist))
-    else:
-        st.write("Please provide an antagonist in the sidebar.")
+        # Column 2: Protagonist and Antagonist
+        with col2:
+            st.header("Protagonist")
+            st.write(generate_protagonist(protagonist))
+
+            st.header("Antagonist")
+            st.write(generate_antagonist(antagonist))