Update app.py

app.py

@@ -1,3 +1,4 @@
+
 # import os
 # import streamlit as st
 # import anthropic
@@ -80,7 +81,7 @@
 #                     f"Antagonist: {antagonist}")
 #     message = client.messages.create(
 #         model="claude-3-5-sonnet-20240620",
-#         max_tokens= 150,
+#         max_tokens=150,
 #         temperature=0.7,
 #         system="You are a master storyteller. Generate a detailed game story based on the inputs provided.",
 #         messages=[
@@ -101,7 +102,7 @@
 # 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 real scenario. Generate more ideas to compete with the big communities.")
+# 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:
@@ -122,8 +123,8 @@
 #         st.session_state.antagonist_description = antagonist_description
 #         st.session_state.game_story = game_story
 
-# # Layout with two columns
-# col1, col2 = st.columns(2)
+# # Layout with three columns
+# col1, col2, col3 = st.columns(3)
 
 # with col1:
 #     st.header("🌍 Game Environment")
@@ -131,216 +132,38 @@
 #         st.write(st.session_state.env_description)
 #     else:
 #         st.write(game_environment)
+#     st.markdown("---")
 
 # 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 col1:
 #     st.header("🦸‍♂️ Protagonist")
 #     if 'protagonist_description' in st.session_state:
 #         st.write(st.session_state.protagonist_description)
 #     else:
 #         st.write(protagonist)
+#     st.markdown("---")
 
-# with col2:
+# with col3:
 #     st.header("🦹‍♀️ Antagonist")
 #     if 'antagonist_description' in st.session_state:
 #         st.write(st.session_state.antagonist_description)
 #     else:
 #         st.write(antagonist)
-#---------------------------------------------------------------------
-# import os
-# import streamlit as st
-# import anthropic
-# from dotenv import load_dotenv
+#     st.markdown("---")
 
-# # Load environment variables from .env file
-# load_dotenv()
+# # 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}")
+# else:
+#     combined_content = "Your complete game scenario and script will be generated based on the inputs provided."
 
-# # 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
+# st.header("📜 Game Scenario & Script")
+# st.write(combined_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.",
-#         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
-
-# def generate_scenario_and_script(environment, protagonist, antagonist):
-#     scenario_prompt = (f"Create a scenario and script based on the following details:\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=200,
-#         temperature=0.7,
-#         system="You are an expert in game scenario and script writing. Generate a scenario and script based on the inputs provided.",
-#         messages=[
-#             {
-#                 "role": "user",
-#                 "content": [
-#                     {
-#                         "type": "text",
-#                         "text": scenario_prompt
-#                     }
-#                 ]
-#             }
-#         ]
-#     )
-#     return message.content[0].text
-
-# # App Title
-# st.title("🎮 StoryForge")
-
-# # App Description
-# st.write("StoryForge 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)
-#         scenario_and_script = generate_scenario_and_script(env_description, protagonist_description, antagonist_description)
-        
-#         # 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
-#         st.session_state.scenario_and_script = scenario_and_script
-
-# # Layout with three columns
-# col1, col2, col3 = st.columns(3)
-
-# 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 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 col1:
-#     st.header("🦸‍♂️ Protagonist")
-#     if 'protagonist_description' in st.session_state:
-#         st.write(st.session_state.protagonist_description)
-#     else:
-#         st.write(protagonist)
-
-# with col2:
-#     st.header("🦹‍♀️ Antagonist")
-#     if 'antagonist_description' in st.session_state:
-#         st.write(st.session_state.antagonist_description)
-#     else:
-#         st.write(antagonist)
-
-# with col3:
-#     st.header("🎬 Scenario and Script")
-#     if 'scenario_and_script' in st.session_state:
-#         st.write(st.session_state.scenario_and_script)
-#     else:
-#         st.write("Your scenario and script will be generated based on the inputs provided.")
-#=-----------------------------------------------------
 import os
 import streamlit as st
 import anthropic
@@ -480,6 +303,11 @@ with col2:
     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("---")
@@ -488,6 +316,11 @@ with col3:
     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("---")