This repository implements a Cybersecurity Threat agentic workflow that generates a report for cybersecurity professionals using the CrewAI framework and Streamlit for the user interface. The agents helps analyze detected cybersecurity threats by providing detailed analysis, forensic investigation, mitigation strategies, and generating comprehensive reports.
- Threat analysis using multiple AI agents
- CSV data upload for threat information
- Integration with Groq API for language model processing
- Customizable model selection
- Comprehensive report generation
- Python 3.7+
- Groq API key
-
Clone the repository:
git clone /~https://github.com/Smartshield-INSAT/Report-Generator.git cd Report-Generator
-
Install the required dependencies:
pip install -r requirements.txt
-
Create a
.env
file in the project root and add your Groq API key:GROQ_API_KEY = your_groq_api_key_here
-
Run the Streamlit app:
streamlit run app.py
-
Open your web browser and navigate to the URL provided by Streamlit (usually
http://localhost:8501
). -
Use the interface to:
- Select a language model
- Input a description of the detected threat
- Upload a CSV file containing threat data
- View the analysis and generated report
app.py
: Main Streamlit applicationagents.py
: Defines the AI agents used in the analysistasks.py
: Defines the tasks performed by the agentsrequirements.txt
: Lists all Python dependencies
The project uses the following AI agents:
- Threat Analyzer Agent
- Mitigation Strategist Agent
- AI Explainer Agent
- Security Researcher Agent
- Report Generator Agent
Each agent has a specific role in analyzing and responding to the detected threat.
This project is licensed under the MIT License - see the LICENSE file for details.
- CrewAI framework
- Streamlit
- Groq API