Update app.py

app.py

@@ -1,143 +1,157 @@
+import os
 import streamlit as st
 import anthropic
-import os
 from dotenv import load_dotenv
 
-# Load API key from .env file
+# Load environment variables from .env file
 load_dotenv()
+
+# Retrieve the API key from environment variables
 api_key = os.getenv("Claude_api_key")
 
-# Initialize the Anthrop client
+# Initialize the Anthropic client with the API key
 client = anthropic.Anthropic(api_key=api_key)
 
-def extract_relevant_info(response):
-    """Extract relevant information from the API response."""
-    # Print response to understand its structure
-    print(response)
-    # Adjust this based on the actual structure of response
-    return response.content.get_text().strip() if hasattr(response.content, 'get_text') else "No content available"
-
-def generate_game_environment(description):
-    response = client.messages.create(
+# 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 a world-class game designer. Respond only with detailed and creative descriptions of game environments.",
+        system="You are a World Class Game Designer. Generate a detailed description of a game environment based on the input.",
         messages=[
             {
                 "role": "user",
                 "content": [
                     {
                         "type": "text",
-                        "text": f"Create a game environment based on the following description: {description}"
+                        "text": f"Create a detailed description of a game environment based on this input: {environment_description}"
                     }
                 ]
             }
         ]
     )
-    return extract_relevant_info(response)
+    return message.content[0].text
 
-def generate_protagonist(description):
-    response = client.messages.create(
+def generate_protagonist(protagonist_description):
+    message = client.messages.create(
         model="claude-3-5-sonnet-20240620",
         max_tokens=150,
         temperature=0.7,
-        system="You are a world-class game designer. Respond only with detailed and creative descriptions of game protagonists.",
+        system="You are an expert in character creation. Generate a detailed description of a game protagonist based on the input.",
         messages=[
             {
                 "role": "user",
                 "content": [
                     {
                         "type": "text",
-                        "text": f"Create a game protagonist based on the following description: {description}"
+                        "text": f"Create a detailed description of a game protagonist based on this input: {protagonist_description}"
                     }
                 ]
             }
         ]
     )
-    return extract_relevant_info(response)
+    return message.content[0].text
 
-def generate_antagonist(description):
-    response = client.messages.create(
+def generate_antagonist(antagonist_description):
+    message = client.messages.create(
         model="claude-3-5-sonnet-20240620",
         max_tokens=150,
         temperature=0.7,
-        system="You are a world-class game designer. Respond only with detailed and creative descriptions of game antagonists.",
+        system="You are an expert in villain creation. Generate a detailed description of a game antagonist based on the input.",
         messages=[
             {
                 "role": "user",
                 "content": [
                     {
                         "type": "text",
-                        "text": f"Create a game antagonist based on the following description: {description}"
+                        "text": f"Create a detailed description of a game antagonist based on this input: {antagonist_description}"
                     }
                 ]
             }
         ]
     )
-    return extract_relevant_info(response)
+    return message.content[0].text
 
 def generate_game_story(environment, protagonist, antagonist):
-    response = client.messages.create(
+    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=1000,
+        max_tokens=150,
         temperature=0.7,
-        system="You are a world-class game designer. Respond only with detailed and creative game story plots.",
+        system="You are a master storyteller. Generate a detailed game story based on the inputs provided.",
         messages=[
             {
                 "role": "user",
                 "content": [
                     {
                         "type": "text",
-                        "text": f"Create a game story plot based on the following: \n"
-                                f"Environment: {environment}\n"
-                                f"Protagonist: {protagonist}\n"
-                                f"Antagonist: {antagonist}"
+                        "text": story_prompt
                     }
                 ]
             }
         ]
     )
-    return extract_relevant_info(response)
-
-# Sidebar for user inputs
-st.sidebar.header("Input Your Game Details")
-st.sidebar.text("Please provide detailed and creative descriptions.")
-
-game_environment_input = st.sidebar.text_input("🌍 Game Environment", 
-                                               "Describe the setting of your game. Include details like time period, location, and atmosphere.")
-protagonist_input = st.sidebar.text_input("🦸 Protagonist", 
-                                          "Describe the main character. Include their background, motivations, and abilities.")
-antagonist_input = st.sidebar.text_input("🦹 Antagonist", 
-                                         "Describe the main enemy. Include their backstory, goals, and relationship with the protagonist.")
-
-# Main app title and description
-st.title("Story Forger")
-st.write("Developed by Salman Maqbool")
-st.write("**Story Forger** is a tool for game developers to easily create detailed game design documents. Input your game environment, protagonist, and antagonist, and generate a solid foundation for your game's story.")
-
-# Generating the content based on user inputs
-if st.sidebar.button("Generate"):
-    if game_environment_input and protagonist_input and antagonist_input:
-        game_environment = generate_game_environment(game_environment_input)
-        protagonist = generate_protagonist(protagonist_input)
-        antagonist = generate_antagonist(antagonist_input)
+    return message.content[0].text
+
+# App Title
+st.title("Story Forger ✨")
+
+# App Description
+st.write("Story Forger helps game developers generate comprehensive Game Design Documents. Input details about your game environment, protagonist, and antagonist to create a structured design document.")
+
+# Sidebar Inputs
+with st.sidebar:
+    st.header("📝 Game Details")
+    game_environment = st.text_input("🌍 Game Environment", "Describe the setting of your game")
+    protagonist = st.text_input("🦸 Protagonist", "Describe the main character")
+    antagonist = st.text_input("👹 Antagonist", "Describe the main villain or opposing force")
+    if st.button("Generate Document 🖨️"):
+        # Generate content based on user input
+        env_description = generate_game_environment(game_environment)
+        protagonist_description = generate_protagonist(protagonist)
+        antagonist_description = generate_antagonist(antagonist)
         game_story = generate_game_story(game_environment, protagonist, antagonist)
+        
+        # Store results in session state
+        st.session_state.env_description = env_description
+        st.session_state.protagonist_description = protagonist_description
+        st.session_state.antagonist_description = antagonist_description
+        st.session_state.game_story = game_story
+
+# Layout with two columns
+col1, col2 = st.columns(2)
 
-        col1, col2 = st.columns(2)
+with col1:
+    st.header("🌍 Game Environment")
+    if 'env_description' in st.session_state:
+        st.write(st.session_state.env_description)
+    else:
+        st.write(game_environment)
 
-        with col1:
-            st.subheader("🌍 Game Environment")
-            st.write(game_environment)
-            
-            st.subheader("🦸 Protagonist")
-            st.write(protagonist)
+with col2:
+    st.header("📖 Game Story")
+    if 'game_story' in st.session_state:
+        st.write(st.session_state.game_story)
+    else:
+        st.write("Your game story will be generated based on the inputs provided.")
 
-        with col2:
-            st.subheader("📖 Game Story")
-            st.write(game_story)
+with col1:
+    st.header("🦸 Protagonist")
+    if 'protagonist_description' in st.session_state:
+        st.write(st.session_state.protagonist_description)
+    else:
+        st.write(protagonist)
 
-            st.subheader("🦹 Antagonist")
-            st.write(antagonist)
+with col2:
+    st.header("👹 Antagonist")
+    if 'antagonist_description' in st.session_state:
+        st.write(st.session_state.antagonist_description)
     else:
-        st.error("Please fill in all the details in the sidebar before generating.")
+        st.write(antagonist)
+
+# Footer
+st.write("**Developed by Salman Maqbool**")