prompt_generate.py

import random
import json
from difflib import SequenceMatcher
import pandas as pd

def generate_prompt_with_examples(text, label, n, start_index=500, end_index=1000):

    text_len = len(text)
    label_len = len(label)
    end_index = min(end_index, text_len, label_len)

    if start_index >= end_index:
        return "none"

    available_examples = end_index - start_index
    n = min(n, available_examples)
    
    prompt = ""

    random_indices = random.sample(range(start_index, end_index), n)

    for i in random_indices:
        text_prompt = text.iloc[i]
        triple_prompt = label.iloc[i]
        prompt += text_prompt + '\n' + str(triple_prompt) + '\n'

    return prompt