def main()

in infra-as-code/modules/ingest-pipeline/cf-transcript-correction/main.py [0:0]

• 31 lines of code
• 1 McCabe index (conditional complexity)

def main(request):
    event = request.get_json()
    print("Cloud event: {}".format(event))

    #Getting variables
    formatted_audio_file_name = event.get("event_filename")
    formatted_audio_bucket_id = event.get("event_bucket")
    transcript_bucket_id = event.get("transcript_bucket")
    transcript_file_name = event.get("transcript_filename")
    project_id = os.environ.get('PROJECT_ID')
    location_id = os.environ.get('LOCATION_ID')
    model_name = os.environ.get('MODEL_NAME')
    ingest_record_bucket_id = os.environ.get('INGEST_RECORD_BUCKET_ID')
    original_file_name = event.get("original_file_name")
    client_specific_constraints = os.environ.get("CLIENT_SPECIFIC_CONSTRAINTS")
    client_specific_context = os.environ.get("CLIENT_SPECIFIC_CONTEXT")
    few_shot_examples = os.environ.get("FEW_SHOT_EXAMPLES")


    genai = GenAIFixer (
    project_id,
    location_id,
    model_name,
    transcript_bucket_id,
    transcript_file_name,
    formatted_audio_file_name, 
    formatted_audio_bucket_id,
    ingest_record_bucket_id,
    original_file_name,
    client_specific_constraints,
    client_specific_context,
    few_shot_examples
    )

    event_dict = genai.fix()
    return event_dict