pages/2_Auto_Generate_Plan_and_Personas.py

from aeo_ex_generator.aeo_example_generator import ExampleGenerator

import openai
import os
import streamlit as st

import streamlit as st 

st.subheader("This space will autognerate the plans and the personas")

OpenAI_Key = st.text_input(" Please enter your OpenAI key here to continue")
# only continue if the key is given
if OpenAI_Key:
    st.write("Please wait, it will take a while to generate. ")
    os.environ['OPENAI_API_KEY'] = OpenAI_Key
    openai.api_key =os.environ['OPENAI_API_KEY'] 

    

    ex = ExampleGenerator() #declare the instance
    example= """ You are part of a small business called MineralRUs Inc., which is a small, yet critical mining company in New Hampshire specializing in rare earth elements.
    Your main clients are from particular sectors of technology and defense industries, and are considered to be critical infrastructure. Therefore, your business and its operation is essential to the integrity of supply chains enabling those critical infrastructure.
    MineralRUs Inc. maintains paper-based record of their key intellectual property and operational procedures, which are stored in a physical safe. This fact is known by only a select few in the company.
    Your business has a small, unorganized computer network with only generic security infrastructure, and the company does not have the resources to employ an in-house information security team. Instead, a very small team is tasked with maintaining Operational Technology (OT) systems on the network.
    Your company has recently received an advisory from federal authorities, warning businesses in your industry about the rapidly growing threat of cyber espionage by state-sponsored Advanced Persistent Threats (APTs). In response, your company decides to engage an external cybersecurity provider to assist with timely incident response and prevention. The external provider recommends the employment of active adversary engagement plans, and asks you to develop a high-level deception and denial strategy.
    """
    context_prompt_input = st.text_area("Enter your example here:", example,height=500)
    keywords = st.text_area("Enter keywords to keep in the plan","breadcrumbtrail")
    nkeywords= st.text_area("You can pass keywords to exclude from the example generation", "honeypot")
    e = ex.auto_generate(planSize=5, keywords=keywords,nkeywords=nkeywords, contextPrompt= context_prompt_input)  
    # e = ex.auto_generate(contextPrompt=context_prompt_input)
    
    st.subheader("Plan")
    st.write(ex.description) #the description is cached in this variable 
 
    st.subheader('Prefix Namespaces')
    st.write(e['@context']) #here are the prefix namespaces mentioned in the example



    result = [i for i in e['@graph']]
    st.subheader('Full example:')
    try:
        st.json(result)
    except:
        st.text_area("", result, height=400)

    if result:
        st.subheader("Auto generate specific plan")

        st.subheader("Persona based on the existing description provided in the deception plan")
        _ = ex.generate_personas() 
        st.write( ex.persona_descriptions )