Function scan_secret

deeppass2.py:183–218

Main function to scan the document for secrets

(doc)

    return bert_secret

async def scan_secret(doc):
    """Main function to scan the document for secrets"""
    secrets = []
    try:
        np_secret, doc_np_cleaned = await get_nosey_parker_results(doc)
        secrets.append({'Nosey Parker': np_secret})
    except Exception as e:
        print(f"Nosey Parker error: {e}")
        doc_np_cleaned = doc  # Use original doc if Nosey Parker fails
        secrets.append({'Nosey Parker': 'Failed'})

    try:
        chunks = chunk_document(doc_np_cleaned, tokenizer)
    except Exception as e:
        print(f"Chunking error: {e}")
        chunks = None

    if chunks:  # Only proceed if chunking succeeded
        try:
            bert_secret = get_secrets_BERT(chunks)
            secrets.append({'BERT_secrets': bert_secret})
        except Exception as e:
            print(f"BERT error: {e}")
            bert_secret = None
            secrets.append({'BERT_secrets': 'Failed'})
        
        if bert_secret:  # Only run LLM if BERT succeeded
            try:
                LLM_secret = get_secrets_LLM(bert_secret)
                secrets.append({'LLM_scanning': LLM_secret})
            except Exception as e:
                print(f"LLM scanning error: {e}")
                secrets.append({'LLM_scanning': 'Failed'})
        

    return secrets


@app.route('/api/deeppass2', methods=['POST'])