lilbool/vuln-code-analysis

1import json
2import pandas as pd
3
4def load_json_to_dataframe(json_file, label):
5    """Carrega um JSON de códigos e adiciona a coluna de rótulo (label)."""
6    with open(json_file, 'r', encoding='utf-8') as f:
7        data = json.load(f)
8    df = pd.DataFrame(data)
9    df['label'] = label  # Adiciona o rótulo: 1 para vulnerável, 0 para seguro
10    return df
11
12if __name__ == "__main__":
13    # Caminhos dos arquivos JSON
14    safe_json = "safe-code-analyzer/compiled_safe_codes.json"
15    vuln_json = "exploit-analyzer/compiled_exploits.json"
16
17    # Carregar os dados
18    safe_df = load_json_to_dataframe(safe_json, label=0)
19    vuln_df = load_json_to_dataframe(vuln_json, label=1)
20
21    # Combinar os dados
22    dataset_df = pd.concat([safe_df, vuln_df], ignore_index=True)
23
24    # Salvar como CSV para treinamento
25    dataset_df.to_csv("code_analysis_dataset.csv", index=False)
26    print("[SUCCESS] Dataset saved to code_analysis_dataset.csv")
27