DEV Community

Akshay Keerthi
Akshay Keerthi

Posted on

Building a Cybersecurity Assistant using Lyzr SDK

In today’s digital age, cybersecurity is paramount. To help individuals safeguard their online activities and devices, I developed a Cybersecurity Assistant app using Lyzr Automata SDK and OpenAI’s GPT-4 Turbo. This blog post walks you through the creation of this app, which provides personalized cybersecurity tips and a custom security checklist based on user input.

Image description

Why use Lyzr SDK’s?

With Lyzr SDKs, crafting your own GenAI application is a breeze, requiring only a few lines of code to get up and running swiftly.

Lets get Started!

Setting Up the Environment

To start, we need to import the necessary libraries and set up the environment, including the OpenAI API key.

import streamlit as st
from lyzr_automata.ai_models.openai import OpenAIModel
from lyzr_automata import Agent, Task
from PIL import Image
from lyzr_automata.tasks.task_literals import InputType, OutputType
import os
Enter fullscreen mode Exit fullscreen mode

Set the OpenAI API key

os.environ["OPENAI_API_KEY"] = st.secrets["apikey"]
Enter fullscreen mode Exit fullscreen mode

Creating the App Title and Introduction

We then set the title and provide a brief introduction to guide the users on what information they need to input.

st.title("Cybersecurity Assistant")
st.markdown("Welcome to Cybersecurity Assistant, your personalized cybersecurity advisor. Simply input your online activities and your device specification, and receive tailored tips to keep your digital life secure and protected.")
st.markdown("1) Mention your online activities (websites visited, download habits, device and network usage etc).")
st.markdown("2) Mention your device specifications.")
input = st.text_input("Please enter the above details:", placeholder="Type here")
Enter fullscreen mode Exit fullscreen mode

Initializing the OpenAI Model

We initialize the OpenAI model with specific parameters for text completion. This model will generate the personalized cybersecurity advice.

open_ai_text_completion_model = OpenAIModel(
    api_key=st.secrets["apikey"],
    parameters={
        "model": "gpt-4-turbo-preview",
        "temperature": 0.2,
        "max_tokens": 1500,
    },
)
Enter fullscreen mode Exit fullscreen mode

Defining the Generation Function

The generation function uses the OpenAI model to generate personalized cybersecurity tips and a custom security checklist based on user input. The function defines the agent's role and prompt for the task.

def generation(input):
    generator_agent = Agent(
        role="Expert CYBERSECURITY CONSULTANT",
        prompt_persona="Your task is to DEVELOP Personalized Security Tips and CREATE a Custom Security Checklist tailored to an individual's online activities and device specifications.")
    prompt = """
[Prompts here]
"""
    generator_agent_task = Task(
        name="Generation",
        model=open_ai_text_completion_model,
        agent=generator_agent,
        instructions=prompt,
        default_input=input,
        output_type=OutputType.TEXT,
        input_type=InputType.TEXT,
    ).execute()
    return generator_agent_task
Enter fullscreen mode Exit fullscreen mode

Adding the Assist Button

if st.button("Assist!"):
    solution = generation(input)
    st.markdown(solution)
Enter fullscreen mode Exit fullscreen mode

The Cybersecurity Assistant app helps users receive personalized cybersecurity advice by analyzing their online activities and device specifications. By leveraging the power of Lyzr Automata SDK and OpenAI’s GPT-4 Turbo, this app provides practical and actionable security tips to keep users’ digital lives secure.

App link: https://cybersecurityassistant-lyzr.streamlit.app/

Source Code: https://github.com/isakshay007/cybersecurity_assistant

Feel free to try building your own version of the Cybersecurity Assistant app and explore the potential of AI-driven cybersecurity solutions! If you have any questions or need further assistance, don’t hesitate to contact Lyzr.

Website: Lyzr.ai
Book a Demo: Book a Demo
Discord: Join our Discord community
Slack: Join our Slack channel

Top comments (0)