Project_Risk_Management/main.py at main · SudeshnaPathak/Project_Risk_Management · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
from utils.s3_utils import list_subfolders, download_file, upload_file
from final_crew import MarketAnalysisCrew, MitigationStrategiesAnalysisCrew, OverallRiskAnalysisCrew, ProjectRiskAnalysisCrew, ResourceRiskAnalysisCrew, TransactionRiskAnalysisCrew
from datetime import datetime
import os
import json
from utils.rag_utils import get_answer_from_chain_local, get_vector_store, get_qdrant_vector_store
from utils.qdrant_utils import delete_qdrant_collection
from utils.mail_utils import send_mail


os.makedirs('output', exist_ok=True)
os.makedirs('project_details', exist_ok=True)

subfolders = list_subfolders()

for folder in subfolders:
    print(f"Processing folder: {folder}")
    download_file(f"{folder}resource.json", "project_details/resource.json")
    download_file(f"{folder}project.json", "project_details/project.json")
    download_file(f"{folder}transaction.json", "project_details/transaction.json")
    download_file(f"{folder}project_description.txt", "project_details/project_description.txt")

    with open("project_details/project_description.txt", "r") as f:
        project_description = f.read()

    with open("project_details/transaction.json", "r") as f:
        transaction_records = json.load(f)

    with open("project_details/static_rules.json", "r") as f:
        static_records = json.load(f)

    with open("project_details/resource.json", "r") as f:
        resource_records = json.load(f)

    with open("project_details/project.json", "r") as f:
        project_records = json.load(f)

    MarketAnalysisCrew().crew().kickoff({"project_docs": project_description, "year": datetime.now().year})

    with open("project_details/market_analysis_summary.json", "r") as f:
        market_records = json.load(f)

    TransactionRiskAnalysisCrew().crew().kickoff({"transaction_records": transaction_records, "static_records": ["transaction_rules"], "market_records": market_records})
    ResourceRiskAnalysisCrew().crew().kickoff({"resource_records": resource_records, "static_records": static_records["resource_management_rules"], "market_records": market_records})
    ProjectRiskAnalysisCrew().crew().kickoff({"project_records": project_records, "static_records": static_records["project_risk_rules"]})

    with open("output/project_risk_analysis.txt", "r") as f:
        project_risks = f.read()
    with open("output/resource_risk_analysis.txt", "r") as f:
        resources_risks = f.read()
    with open("output/transaction_risk_analysis.txt", "r") as f:
        transaction_risks = f.read()

    OverallRiskAnalysisCrew().crew().kickoff({"project_risks": project_risks, "static_records": static_records["overall_risk_rules"], "resources_risks": resources_risks, "transaction_risks": transaction_risks})

    with open("output/overall_project_risk_analysis.txt", "r") as f:
        risk_analysis = f.read()
    MitigationStrategiesAnalysisCrew().crew().kickoff({"risk_analysis": risk_analysis})

    with open("output/report.txt", 'w') as f:
        f.write(f"Project: {folder}\n")

    with open("output/report.txt", 'a') as f:
        f.write(f"Transaction Analysis: {transaction_risks}\n")
        f.write(f"Resource Analysis: {resources_risks}\n")
        f.write(f"Project Analysis: {project_risks}\n")
        f.write(f"Overall Risk Analysis: {risk_analysis}\n")
        f.write("\n\n")

    upload_file(f"{folder}report.txt", "output/report.txt")
    upload_file(f"{folder}mitigation_strategies.txt", "output/mitigation_strategies.txt")

    get_vector_store(["output/report.txt"])
    answer = get_answer_from_chain_local("Provide the details of all the risks identified as critical and High severity risks in the project report.")
    send_mail(answer["output_text"] , folder[:-1])

    paths = [
        "output/report.txt",
        "output/mitigation_strategies.txt",
    ]

    delete_qdrant_collection(folder[:-1])
    get_qdrant_vector_store(paths, folder[:-1])