Update pages/2_Auto_Generate_Plan_and_Personas.py

pages/2_Auto_Generate_Plan_and_Personas.py

@@ -18,8 +18,19 @@ if OpenAI_Key:
     
 
     ex = ExampleGenerator() #declare the instance
-    e = ex.auto_generate()
+    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')
@@ -36,9 +47,7 @@ if OpenAI_Key:
 
     if result:
         st.subheader("Auto generate specific plan")
-        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")
-        personas = ex.auto_generate(planSize=5, keywords=keywords,nkeywords=nkeywords, contextPrompt= ex.description)  
+
         st.subheader("Persona based on the existing description provided in the deception plan")
         _ = ex.generate_personas() 
         st.write( ex.persona_descriptions )