diff --git a/backend/core/templates/thankyou.html b/backend/core/templates/thankyou.html index 0dbbd16..25d161f 100644 --- a/backend/core/templates/thankyou.html +++ b/backend/core/templates/thankyou.html @@ -7,6 +7,37 @@ We will send the document to {{ email }} when it is ready. +
| Risk ID | +Risk Name | +Category | +Primary Impact | +Secondary Impact | +Tertiary Impact | +Detection Difficulty | +Recovery Complexity | +Business Impact Severity | +
|---|---|---|---|---|---|---|---|---|
| {{ risk.risk_id }} | +{{ risk.risk_name }} | +{{ risk.category }} | +{{ risk.primary_impact }} | +{{ risk.secondary_impact }} | +{{ risk.tretiary_impact }} | +{{ risk.detection_difficulty }} | +{{ risk.recovery_complexity }} | +{{ risk.businnes_impact_severity }} | +
Página simples.
') @@ -19,8 +20,25 @@ def signup(request): if request.method == 'POST': form = OrganizationForm(request.POST) if form.is_valid(): - form.save() - return render(request, 'thankyou.html', {'email': form.data['email']}) + organization = form.save() + print(f"Organization :{organization}") + top_risk_ids = get_top_risk(organization) + print(f"Top risks: {top_risk_ids}") + top_risks = Risk.objects.filter(risk_id__in = top_risk_ids) + print(f"Final: {top_risks}") + + document = Document.objects.create(organization=organization) + document.add_segment('h1', "Top 10 Risk Identified") + + for risk in top_risks: + document.add_segment('h2',f"Risk: {risk.risk_id}:{risk.risk_name}") + document.add_segment('body',f"Category: {risk.category} \n Primary Impact: {risk.primary_impact} \n Secondary Impact: {risk.secondary_impact} \n Tertiary Impact: {risk.tretiary_impact} \n Detection Difficulty: {risk.detection_difficulty} \n Recovery Complexity: {risk.recovery_complexity} \n Business Impact Severity: {risk.businnes_impact_severity} ") + + return render(request, 'thankyou.html', { + 'email': form.data['email'], + 'top_risks':top_risks, + 'document':document + }) else: logging.error(form.errors) return render(request, 'signup.html', {'form': form}) diff --git a/backend/settings.py b/backend/settings.py index 33856be..885a7cd 100644 --- a/backend/settings.py +++ b/backend/settings.py @@ -14,6 +14,12 @@ from pathlib import Path from decouple import Csv, config from dj_database_url import parse as dburl +import os +from dotenv import load_dotenv +load_dotenv() + +#API key +OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") # Build paths inside the project like this: BASE_DIR / 'subdir'. BASE_DIR = Path(__file__).resolve().parent.parent