prompt_util.py

import re
import random
import config

import jieba
import jieba.posseg as pseg


original_post_prompt = [
"最近过得怎么样？",
"你这段时间都在忙什么？",
"离我们上次见面以来，有什么新鲜事吗？",
"最近有什么有趣的经历吗？",
"你近期的生活如何？",
"距离我们上次见面已经过了一段时间了，你过得好吗？",
"你最近都在做些什么？",
"近期有什么重要的变化吗？",
"有什么新的发展吗？",
"能和我分享一下你近期的生活吗？",
"你最近有什么特别的事情发生吗？",
"你最近的生活有什么变化吗？",
"最近有没有什么大事发生？",
"过去的几个月里，你都经历了些什么？",
"你过去一段时间都在忙些什么？",
"有什么新鲜事要和我分享吗？",
"这段时间有没有什么让你难忘的事情发生？",
"最近在工作或生活方面有什么挑战吗？",
"从我们上次见面到现在，你都遇到了哪些有意思的事情？",
"过去几个月你都忙些什么？",
"最近你都有哪些新发现？",
"你近期有哪些值得分享的经历？",
"跟我聊聊你最近的日常吧？",
"近期有什么值得一提的事吗？",
"这段时间有没有什么特别的回忆？",
"最近有没有什么让你感到骄傲的事？",
"你最近有什么惊喜发生吗？",
"过去一段时间，你的生活中有哪些亮点？",
"近来你都有哪些新的尝试？",
"你最近的日子过得怎样？有什么精彩的时刻？",
"这段时间有什么特别的见闻吗？",
"从上次见面到现在，你有什么有趣的故事吗？",
"最近有没有什么特殊的事情让你感到开心？",
"你最近有什么新的梦想或目标吗？",
"过去的一段时间里，你有什么难忘的经历吗？",
"你最近有没有什么特别的发现或灵感？",
"最近有没有什么突破性的成就？",
"这段时间你都学到了哪些新知识？",
"最近有没有什么值得一提的事情发生在你身边？",
"最近你有什么值得庆祝的事情吗？",
"从我们上次见面以来，你都有哪些新的探险？",
"最近有没有什么让你感到充实的事情？",
"过去一段时间里，你有什么特别的体验？",
"你近期有什么令人惊讶的收获？",
"跟我分享一下你最近的喜怒哀乐吧？"
]

related_topic_prompt = [
    "关于[AAA]，你有什么想法？",
    "你觉得[AAA]在日常生活中扮演着什么角色？",
    "你曾经与[AAA]有过什么相关经历吗？",
    "你觉得[AAA]有哪些相关的话题值得探讨？",
    "关于[AAA]，你有什么好的经验可以分享吗？",
    "你如何看待[AAA]与我们生活中的其他方面的关系？",
    "你觉得[AAA]有哪些值得关注的趋势？",
    "你认为[AAA]对于我们的日常生活有多大意义？",
    "你觉得[AAA]有哪些令人好奇的方面？",
    "你有没有关于[AAA]的有趣故事？",
    "你觉得[AAA]有哪些令人欣赏的方面？",
    "关于[AAA]，你觉得有哪些有趣的讨论点？",
    "关于[AAA]，你有什么新的见解吗？",
    "你觉得[AAA]是如何影响你生活的？",
    "你曾经有过关于[AAA]的有趣经历吗？",
    "关于[AAA]，有哪些热门话题值得关注？",
    "你有没有关于[AAA]的特别记忆？",
    "你觉得[AAA]与你生活中的其他元素有何联系？",
    "你觉得[AAA]有哪些令人惊讶的方面？",
    "你有没有听过关于[AAA]的趣闻轶事？",
    "关于[AAA]，你觉得有哪些引人入胜的话题？",
    "你对[AAA]有哪些深刻的印象？",
    "你有没有关于[AAA]的奇闻异事？",
    "关于[AAA]，你有什么特别的见解吗？",
    "你有没有关于[AAA]的特别喜好？",
    "你对[AAA]有哪些独到的见解？",
    "你觉得[AAA]如何影响我们的思考方式？",
    "你有没有关于[AAA]的趣味事例？",
    "你觉得[AAA]有哪些令人叹为观止的特点？",
    "关于[AAA]，你有什么富有启发性的想法？",
    "你有没有关于[AAA]的奇特经历？",
    "你觉得[AAA]有哪些吸引人的特质？",
    "关于[AAA]，你有什么令人兴奋的见解？"
]

def cut_sent(text):
    sub_sentences = re.split(r'([\。|\！|\？|\；|\，|\n])', text)
    sub_sentences = [s1 + s2 for s1, s2 in zip(sub_sentences[0::2], sub_sentences[1::2])] + ([sub_sentences[-1]] if len(sub_sentences) % 2 == 1 else [])
    return [s.replace("\n", "") for s in sub_sentences if (s.strip() != "" and s.strip() != "(media)" and s.strip() != "(link)")]


def findTopic(md):
    substring = cut_sent(md)
    if len(substring) > 1:
        rr = random.randint(0, len(substring)-1)
        # choose a random substring, maximum length is 5
        topic = substring[rr]
        # cut off the last word, if it's a chinese punctuation
        if topic[-1] in ["，", "。", "！", "？", "；"]:
            topic = topic[:-1]

        # use jieba to do the word segmentation and pos tagging
        tokens = [word for word,flag in pseg.cut(topic) if 'n' in flag] 
        if len(tokens) > 0:
            topic = random.choice(tokens)
        else:
            # it's just it.
            topic = topic

        instruction = random.choice(related_topic_prompt).replace("[AAA]", topic)
        return instruction
    else:
        return None


def checkResponse(response):
    # check if the user's response is too short. Filter it out. 
    if len(response.replace("\n","").replace(" ", "").replace("(media)","").replace("(link)", "")) < config.RESPONSE_THRESH:
        return False
    return True