Update app.py

app.py

@@ -1,4 +1,3 @@
-
 # import os
 # import streamlit as st
 # import anthropic
@@ -135,9 +134,14 @@
 #     st.markdown("---")
 
 # with col2:
-#     st.header("🦸‍♂️ Protagonist")
+#     st.header("🦸‍♂️Protagonist")
 #     if 'protagonist_description' in st.session_state:
 #         st.write(st.session_state.protagonist_description)
+#         if st.button("Edit Protagonist Details", key="edit_protagonist"):
+#             new_protagonist_description = st.text_area("Edit Protagonist Description", st.session_state.protagonist_description)
+#             if st.button("Update Protagonist", key="update_protagonist"):
+#                 st.session_state.protagonist_description = generate_protagonist(new_protagonist_description)
+#                 st.experimental_rerun()
 #     else:
 #         st.write(protagonist)
 #     st.markdown("---")
@@ -146,6 +150,11 @@
 #     st.header("🦹‍♀️ Antagonist")
 #     if 'antagonist_description' in st.session_state:
 #         st.write(st.session_state.antagonist_description)
+#         if st.button("Edit Antagonist Details", key="edit_antagonist"):
+#             new_antagonist_description = st.text_area("Edit Antagonist Description", st.session_state.antagonist_description)
+#             if st.button("Update Antagonist", key="update_antagonist"):
+#                 st.session_state.antagonist_description = generate_antagonist(new_antagonist_description)
+#                 st.experimental_rerun()
 #     else:
 #         st.write(antagonist)
 #     st.markdown("---")
@@ -163,175 +172,59 @@
 # st.header("📜 Game Scenario & Script")
 # st.write(combined_content)
 
-
-import os
+#
+#0----------------
 import streamlit as st
-import anthropic
-from dotenv import load_dotenv
-
-# Load environment variables from .env file
-load_dotenv()
-
-# Retrieve the API key from environment variables
-api_key = os.getenv("Claude_api_key")
-
-# Initialize the Anthropic client with the API key
-client = anthropic.Anthropic(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.",
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": f"Create a detailed description of a game environment based on this input: {environment_description}"
-                    }
-                ]
-            }
-        ]
-    )
-    return message.content[0].text
-
-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.",
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": f"Create a detailed description of a game protagonist based on this input: {protagonist_description}"
-                    }
-                ]
-            }
-        ]
-    )
-    return message.content[0].text
-
-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.",
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": f"Create a detailed description of a game antagonist based on this input: {antagonist_description}"
-                    }
-                ]
-            }
-        ]
-    )
-    return message.content[0].text
-
-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.",
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": story_prompt
-                    }
-                ]
-            }
-        ]
-    )
-    return message.content[0].text
-
-# App Title
-st.title("🎮 Safwan's GameMaker Studio")
-
-# App Description
-st.write("Welcome to Safwan's GameMaker Studio, a popular game development platform known for its ease of use and powerful features. Enter your ideas according to your taste and this will generate a script with a real scenario. Generate more ideas to compete with the big communities.")
-
-# 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 three columns
 col1, col2, col3 = st.columns(3)
 
 with col1:
     st.header("🌍 Game Environment")
+    st.image("path_to_environment_image.jpg", caption="Game Environment", use_column_width=True)
     if 'env_description' in st.session_state:
         st.write(st.session_state.env_description)
     else:
-        st.write(game_environment)
+        st.write("Default game environment")
     st.markdown("---")
 
 with col2:
     st.header("🦸‍♂️ Protagonist")
+    st.image("path_to_protagonist_image.jpg", caption="Protagonist", use_column_width=True)
     if 'protagonist_description' in st.session_state:
         st.write(st.session_state.protagonist_description)
         if st.button("Edit Protagonist Details", key="edit_protagonist"):
             new_protagonist_description = st.text_area("Edit Protagonist Description", st.session_state.protagonist_description)
             if st.button("Update Protagonist", key="update_protagonist"):
-                st.session_state.protagonist_description = generate_protagonist(new_protagonist_description)
+                st.session_state.protagonist_description = new_protagonist_description
                 st.experimental_rerun()
     else:
-        st.write(protagonist)
+        st.write("Default protagonist description")
     st.markdown("---")
 
 with col3:
     st.header("🦹‍♀️ Antagonist")
+    st.image("path_to_antagonist_image.jpg", caption="Antagonist", use_column_width=True)
     if 'antagonist_description' in st.session_state:
         st.write(st.session_state.antagonist_description)
         if st.button("Edit Antagonist Details", key="edit_antagonist"):
             new_antagonist_description = st.text_area("Edit Antagonist Description", st.session_state.antagonist_description)
             if st.button("Update Antagonist", key="update_antagonist"):
-                st.session_state.antagonist_description = generate_antagonist(new_antagonist_description)
+                st.session_state.antagonist_description = new_antagonist_description
                 st.experimental_rerun()
     else:
-        st.write(antagonist)
+        st.write("Default antagonist description")
     st.markdown("---")
 
 # Combine and merge sections to generate a scenario and script
-if 'env_description' in st.session_state and 'protagonist_description' in st.session_state and 'antagonist_description' in st.session_state:
-    combined_content = (f"### Game Scenario\n\n"
-                        f"**Environment:** {st.session_state.env_description}\n\n"
-                        f"**Protagonist:** {st.session_state.protagonist_description}\n\n"
-                        f"**Antagonist:** {st.session_state.antagonist_description}\n\n"
-                        f"**Story:** {st.session_state.game_story}")
+if all(k in st.session_state for k in ['env_description', 'protagonist_description', 'antagonist_description']):
+    combined_content = (
+        f"### Game Scenario\n\n"
+        f"**Environment:** {st.session_state.env_description}\n\n"
+        f"**Protagonist:** {st.session_state.protagonist_description}\n\n"
+        f"**Antagonist:** {st.session_state.antagonist_description}\n\n"
+        f"**Story:** This is where the story will go..."
+    )
 else:
     combined_content = "Your complete game scenario and script will be generated based on the inputs provided."