Update app.py

app.py

@@ -1,34 +1,140 @@
 import streamlit as st
+import anthropic
+import os
+from dotenv import load_dotenv
+
+# Load API key from .env file
+load_dotenv()
+api_key = os.getenv("Claude_api_key")
+
+# Initialize the Anthrop client
+client = anthropic.Anthropic(api_key=api_key)
+
+def extract_relevant_info(response):
+    """Extract relevant information from the API response."""
+    return response['content'].strip()
+
+def generate_game_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.",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": f"Create a game environment based on the following description: {description}"
+                    }
+                ]
+            }
+        ]
+    )
+    return extract_relevant_info(message)
+
+def generate_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.",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": f"Create a game protagonist based on the following description: {description}"
+                    }
+                ]
+            }
+        ]
+    )
+    return extract_relevant_info(message)
+
+def generate_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.",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": f"Create a game antagonist based on the following description: {description}"
+                    }
+                ]
+            }
+        ]
+    )
+    return extract_relevant_info(message)
+
+def generate_game_story(environment, protagonist, antagonist):
+    message = client.messages.create(
+        model="claude-3-5-sonnet-20240620",
+        max_tokens=1000,
+        temperature=0.7,
+        system="You are a world-class game designer. Respond only with detailed and creative game story plots.",
+        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}"
+                    }
+                ]
+            }
+        ]
+    )
+    return extract_relevant_info(message)
 
 # Sidebar for user inputs
 st.sidebar.header("Input Your Game Details")
 st.sidebar.text("Please provide detailed and creative descriptions.")
 
-game_environment = st.sidebar.text_input("🌍 Game Environment", 
-                                         "Describe the setting of your game. Include details like time period, location, and atmosphere.")
-protagonist = st.sidebar.text_input("🦸 Protagonist", 
-                                    "Describe the main character. Include their background, motivations, and abilities.")
-antagonist = st.sidebar.text_input("🦹 Antagonist", 
-                                   "Describe the main enemy. Include their backstory, goals, and relationship with the protagonist.")
+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.")
 
-# Columns for displaying game details
-col1, col2 = st.columns(2)
+# 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)
+        game_story = generate_game_story(game_environment, protagonist, antagonist)
+
+        col1, col2 = st.columns(2)
 
-with col1:
-    st.subheader("🌍 Game Environment")
-    st.write(game_environment)
-    
-    st.subheader("🦸 Protagonist")
-    st.write(protagonist)
+        with col1:
+            st.subheader("🌍 Game Environment")
+            st.write(game_environment)
+            
+            st.subheader("🦸 Protagonist")
+            st.write(protagonist)
 
-with col2:
-    st.subheader("📖 Game Story")
-    st.write("Your generated game story will appear here.")
+        with col2:
+            st.subheader("📖 Game Story")
+            st.write(game_story)
 
-    st.subheader("🦹 Antagonist")
-    st.write(antagonist)
+            st.subheader("🦹 Antagonist")
+            st.write(antagonist)
+    else:
+        st.error("Please fill in all the details in the sidebar before generating.")