added AI analysis

backend/core/views.py

@@ -2,7 +2,8 @@ import logging
 
 from django.shortcuts import render, redirect
 from .forms import OrganizationForm
-
+from .models import Organization,Document,Risk
+from backend.core.utils import get_top_risk
 # @login_required
 # def index(request):
 #     return HttpResponse('<h1>Django</h1><p>Página simples.</p>')
@@ -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})