Create app.py

app.py

@@ -0,0 +1,236 @@
+import requests
+import json
+import gradio as gr
+from datetime import datetime
+
+def ai_record_analyze(score_prompt, standard_prompt, example_prompt, msg, chat_history):
+    sys_prompt = "你是一个销售培训专家, 我会给你完整的销售和客户沟通的对话资料, 其中每一行的会以[时间戳]和[角色]为开头，表示该对话发生的时间和说话的角色,你需要对我给你的会话进行打分和评价, 请你参照以下得分点和例子, 进行打分和评价："
+    sys_prompt3 = """以上得分点中, 你需要给出打分，并给出打分的理由以及支撑你打分的对话语句;如果未得分或者得了0分，也必须要给出理由并给出优化建议，支撑语句输出空字符。请严格按照以下json格式返回,且仅返回json格式内的内容：\n[{'得分': 'xxx',\n'理由':'xxx',\n'优化建议': 'xxx',\n'支撑语句': '0:1:00-0:2:00\n0:2:01-0:3:01'}]\n其中得分你只返回所获得的分数，例如，例如1代表1分，2代表2分。优化建议中返回你认为可以改进的点或者优化的部分。支撑语句返回具体支撑语句的时间戳（每一行最前面的时间戳），你只返回时间戳，不返回该行的具体内容，如果对应多个时间戳则用换行符分隔。注意：我给你的打分标准可能会有多个，但是你返回的时候要把所有得分合并，json列表中只返回一个元素，且不以json代码块形式返回。注意：你返回的时间戳必须是一段时间范围，不要只返回单个时间点，且确保你返回的支撑语句可以支持你的理由，如果未得分则不用返回时间戳"""
+
+    usr_prompt = msg
+
+    messages = []
+
+    if example_prompt == "":
+        messages.append({
+            "role": "system",
+            "content": f"{sys_prompt}\n\n得分点: {score_prompt}\n标准: {standard_prompt}\n\n{sys_prompt3}"
+        })
+    else:
+        messages.append({
+            "role": "system",
+            "content": f"{sys_prompt}\n\n得分点: {score_prompt}\n标准: {standard_prompt}\n例如: {example_prompt}\n\n{sys_prompt3}"
+        })
+        
+    messages.append({
+        "role": "user",
+        "content": usr_prompt
+    })      
+
+    url = "http://117.50.195.92:8080/qwen72-not-quantize/v1/chat/completions"
+
+    payload_qs = {
+        "model": "HiPilot-V1.5-Chat",
+        "messages": messages,
+        "temperature": 0.9,
+        "max_tokens": 2048,
+        "stream": True
+    }
+    headers = {
+        'Content-Type': 'application/json'
+    }
+    
+    response = requests.post(url, headers=headers, json=payload_qs, stream=True)
+
+    bot_message = ""
+
+    if example_prompt == "":
+        chat_history.append([f"得分点: {score_prompt}\n标准: {standard_prompt}", bot_message])
+    else:
+        chat_history.append([f"得分点: {score_prompt}\n标准: {standard_prompt}\n例如: {example_prompt}", bot_message])
+
+
+    for chunk in response.iter_lines():
+        if chunk:  # 忽略空行
+            try:
+                chunk = chunk.decode('utf-8')
+                chunk = chunk.strip().split("data:")[-1]
+                chunk = json.loads(chunk)
+                if "content" in chunk["choices"][0]["delta"]:
+                    bot_message += chunk["choices"][0]["delta"]["content"]
+                    chat_history[-1][1] = bot_message
+                    yield msg, chat_history
+
+            except json.JSONDecodeError as e:
+                print(f"Error decoding JSON: {e}")
+
+    return msg, chat_history
+
+def update_prompt(template):
+    standard_dict = {
+        "一通电话同花顺开场白,总分1分": "关键字包括同花顺、云软件、回访、回馈、来电、软件使用情况等，只要满足其中任何一个或者其他相关开场白，即为完成开场白，得1分。未完成同花顺开场白, 则得0分",
+        "一通电话挖需求,总分5分":  """总分5分，包括：
+1.询问到了客户持仓股票（你在做什么股票, 帮你看一下，持股、有什么股票、持仓、做什么方向)
+2.询问到了客户购买该股票的理由(为什么去买这个股票, 出于什么原因)
+3.询问到了客户的买入价格(什么时候买的, 多少钱的买的，买了多久)
+4.询问到了客户在该点买入的原因(为什么会在这个位置买，买入的原因)
+5.询问到了客户如果看到主力资金流入流出是否还会购买（如果你看到机构资金在出货你还会买吗, 还会卖吗）
+以上每点,如果销售只是发出询问，但客户未回答，则仍算0分，只有客户回答了问题，该点算完成并获得1分, 满分5分，且请给具体的股票、股票代码、时间、原因、理由等信息""",
+        "一通电话对比算账,总分1分": """总分1分，包括：
+是否和客户对比过使用软件前后差异, 完成这一模块则得1分, 未完成则得0分。""",
+        "一通电话总结需求,总分1分": """总分1分，包括：
+总结需求:回顾客户选股或者买卖点的需求, 完���得1分, 未完成得0分。""",
+        "一通电话灌输理念,总分1分": """总分1分，包括：
+灌输理念:做股票要跟着国家政策主力资金, 完成得1分, 未完成得0分。""",
+        "一通电话引出产品,总分1分": """总分1分，包括：
+引出产品:敢于说出产品的名字（产品包括超级level2/超级L2/手机短线宝/同花顺手机VIP/主力密码/荐股师/涨停助手/手机云参数/黄金十分钟/手机诊股/神奇电波选股/AI分时顶底/AI波段雷达/金牛情报/AI机构探测器/决策先锋/金融大师/大研究/财富先锋等）, 完成得1分, 未完成得0分。""",
+        "一通电话产品介绍1,总分1分": """总分2分，包括： 
+是否论述: GS点(买入卖出信号)/主力资金(主力资金有明盘和暗盘)/AI机构活跃度(股票资金参与活跃情况, 活跃度高, 涨跌幅度大), 如均完成得2分, 完成部分得1分, 均未完成得0分。
+""",
+        "一通电话产品介绍2,总分1分": """总分1分，包括： 
+是否案例补充, 存在案例补充得1分, 未完成则得0分。""",
+        "一通电话产品介绍3,总分1分": """总分1分，包括： 
+图片发送, 完成得1分, 未完成得0分。""",
+        "一通电话尝试成交1,总分1分":  """总分1分，包括：
+场景:在案例补充后(办理/合F/支付宝/付钱/微信), 完成得1分, 未完成得0分。""",
+        "一通电话尝试成交2,总分1分": """总分1分，包括：
+答疑后:在答疑补充后(办理/合作/支付宝/付钱/微信), 完成得1分, 未完成得0分。""",
+        "一通电话尝试成交3,总分1分":  """总分1分，包括：
+优惠方案:定金，优惠券，限时活动(送个月), 完成得1分, 未完成得0分。""",
+        "一通电话收尾": """总分1分，包括：
+以股票或者群服务, 或者是约下次打电话时间. 完成得1分，未完成得0分""",
+
+        "二通电话暖场,总分1分":  """自我介绍、询问客户近期的操作，完成暖场, 得1分。未完成暖场, 则得0分。""",
+        "二通电话回顾需求,总分1分": """总分1分，包括：
+回顾客户选股或买卖点的问题。完成得1分。未完成则得0分。""",
+        "二通电话引出产品,总分1分":  """总分1分，包括：
+再次说出产品, 完成则得1分, 未完成则得0分。""",
+        "二通电话产品介绍，功能介绍1,总分1分":  """总分1分，包括：
+强化介绍产品的几个核心功能, 完成得1分, 未完成得0分。""",
+        "二通电话产品介绍，功能介绍2,总分1分": """总分1分，包括：
+论证：案例补充, 完成得1分, 未完成得0分。""",
+        "二通电话产品介绍，功能介绍3,总分1分": """总分1分，包括：
+截图发送, 完成得1分, 未完成得0分。""",
+        "二通电话答疑,总分1分": """总分1分，包括： 
+解决客户的疑虑点, 完成得1分, 未完成得0分。""",
+        "二通电话尝试成交,总分1分": """总分1分，包括：
+引导尝试客户成交例如（办理/合作/支付宝/付钱/微信）, 完成得1分, 未完成得0分。""",
+        "二通电话结束语,总分1分": """总分1分，包括：
+引发客户思考并预约, 完成得1分, 未完成得0分。""",
+        "空白模板":""
+    }
+    score_dict = {
+        "一通电话同花顺开场白,总分1分": "同花顺开场白",
+        "一通电话挖需求,总分5分":  "挖需求",
+        "一通电话对比算账,总分1分": "对比算账",
+        "一通电话总结需求,总分1分": "总结需求, 灌输理念, 引出产品",
+        "一通电话灌输理念,总分1分": "总结需求, 灌输理念, 引出产品",
+        "一通电话引出产品,总分1分": "总结需求, 灌输理念, 引出产品",
+        "一通电话产品介绍1,总分1分": "产品介绍",
+        "一通电话产品介绍2,总分1分": "产品介绍",
+        "一通电话产品介绍3,总分1分": "产品介绍",
+        "一通电话尝试成交1,总分1分": "尝试成交",
+        "一通电话尝试成交2,总分1分": "尝试成交",
+        "一通电话尝试成交3,总分1分": "尝试成交",
+        "一通电话收尾": "收尾",
+
+        "二通电话暖场,总分1分":  "暖场",
+        "二通电话回顾需求,总分1分": "回顾需求",
+        "二通电话引出产品,总分1分":  "引出产品",
+        "二通电话产品介绍，功能介绍1,总分1分": "产品介绍，功能介绍",
+        "二通电话产品介绍，功能介绍2,总分1分": "产品介绍，功能介绍",
+        "二通电话产品介绍，功能介绍3,总分1分": "产品介绍，功能介绍",
+        "二通电话答疑,总分1分": "答疑",
+        "二通电话尝试成交,总分1分": "尝试成交",
+        "二通电话结束语,总分1分": "结束语",
+        "空白模板":""
+    }
+    example_dict = {
+        "一通电话同花顺开场白,总分1分": "",
+        "一通电话挖需求,总分5分":  "",
+        "一通电话对��算账,总分1分": """如果当时跟我们做, 从什么时候开始做到什么时候结束, 你大概就可以赚, 少亏多少, 就可以覆盖这个软件的成本了, 或者是除去软件成本聂可以赚多少。""",
+        "一通电话总结需求,总分1分": """唐先生像你这样买卖点把握不好""",
+        "一通电话灌输理念,总分1分": """唐先生你还是要跟着资金和国家政策""",
+        "一通电话引出产品,总分1分": """像你的买卖点把握不好, 可以跟着我们决策先锋的主力资金来走, 决策的GS点, AI机构活跃度""",
+        "一通电话产品介绍1,总分1分": "",
+        "一通电话产品介绍2,总分1分": """先生300033, 主力资金连续进XX, GS又发出来G点买入信号, 从买入到现在位置, 多少个工作日涨幅多少XX。""",
+        "一通电话产品介绍3,总分1分": """唐先生, 我这个图发给你, 你看到没""",
+        "一通电话尝试成交1,总分1分":  """想你买卖点把握不好，可以和我们合作起来，我们这个功能很好。""",
+        "一通电话尝试成交2,总分1分": """想你买卖点把握不好，可以和我们合作起来，我们这个功能很好。""",
+        "一通电话尝试成交3,总分1分":  "",
+        "一通电话收尾": "",
+
+        "二通电话暖场,总分1分":  """最近股票有操作吗？之前持有的、还在持有吗""",
+        "二通电话回顾需求,总分1分": """像你这样、买卖点把握不好、买高/卖早、很难获利、之前介绍的决策。""",
+        "二通电话引出产品,总分1分":  """上次提到的直播间优惠活动、上次说的决策先锋、直播听完后、觉得怎么样？""",
+        "二通电话产品介绍，功能介绍1,总分1分":  """G点买入、S点卖出、趋势启动、趋势终结、明盘资金、暗盘资金""",
+        "二通电话产品介绍，功能介绍2,总分1分": """XXX股票、XX个交易日、XX收益""",
+        "二通电话产品介绍，功能介绍3,总分1分": """截图发你了，你看一下；微信打开一下，图""",
+        "二通电话答疑,总分1分": "",
+        "二通电话尝试成交,总分1分": """"我带你办理一下、你是微信还是支付宝、你打开同花顺app、我给你发了个链接有看到吗、付好后发我一下截图. 优惠、活动、折扣、办理、合作、支付宝、微信、付钱，定金、赠送、价值1164、服务包.""",
+        "二通电话结束语,总分1分": """考虑清楚、能不能帮到你、想不想改变、下次再联系你、你可以关注一下""",
+        "空白模板":""
+    }
+
+    return score_dict[template], standard_dict[template], example_dict[template]
+
+
+
+def download_prompt(score_prompt, standard_prompt, example_prompt):
+    current_time = str(datetime.now())
+    address = f"/mnt/data/code_generation/cuizhe/yunruanjian/prompt_{current_time}.txt"
+    with open(address,'w') as f:
+        if example_prompt == "":
+            f.write(f"得分点: {score_prompt}\n标准: {standard_prompt}")
+        else:
+            f.write(f"得分点: {score_prompt}\n标准: {standard_prompt}\n例如: {example_prompt}")
+    
+    return address
+
+
+
+
+if __name__ == "__main__":
+    with gr.Blocks(analytics_enabled = False) as demo:
+        chatbot = gr.Chatbot(height=800)
+        with gr.Row():
+            msg = gr.Textbox(placeholder="复制黏贴对话内容", scale=4, max_lines=20, label = "录音内容")
+        with gr.Row():
+            template_choice = gr.Radio(
+                choices=[
+                    "一通电话同花顺开场白,总分1分", "一通电话挖需求,总分5分", "一通电话对比算账,总分1分",
+                    "一通电话总结需求,总分1分", "一通电话灌输理念,总分1分", "一通电话引出产品,总分1分",
+                    "一通电话产品介绍1,总分1分", "一通电话产品介绍2,总分1分", "一通电话产品介绍3,总分1分",
+                    "一通电话尝试成交1,总分1分", "一通电话尝试成交2,总分1分", "一通电话尝试成交3,总分1分",
+                    "一通电话收尾", "二通电话暖场,总分1分", "二通电话回顾需求,总分1分",
+                    "二通电话引出产品,总分1分", "二通电话产品介绍，功能介绍1,总分1分",
+                    "二通电话产品介绍，功能介绍2,总分1分", "二通电话产品介绍，功能介绍3,总分1分",
+                    "二通电话答疑,总分1分", "二通电话尝试成交,总分1分", "二通电话结束语,总分1分",
+                    "空白模板"
+                ], 
+                label="选择模板"
+            )
+            
+        with gr.Row():
+            score_prompt = gr.Textbox(placeholder="得分点", scale=4, max_lines=20, label = "得分点prompt")
+            standard_prompt = gr.Textbox(placeholder="标准", scale=4, max_lines=20, label = "得分标准prompt")
+            example_prompt = gr.Textbox(placeholder="举例", scale=4, max_lines=20, label = "得分举例prompt")
+
+        with gr.Row():
+            submit = gr.Button("提交", variant="primary")
+        with gr.Row():
+            clear = gr.Button("清除")
+            stop = gr.Button("停止", variant="stop")
+            
+        with gr.Row():
+            download_btn = gr.Button("生成文件")
+            prompt_file = gr.File(label="文件")
+
+        template_choice.change(fn=update_prompt, inputs=template_choice, outputs=[score_prompt, standard_prompt, example_prompt])
+        submit_event = submit.click(ai_record_analyze, inputs=[score_prompt, standard_prompt, example_prompt, msg, chatbot], outputs=[msg, chatbot])
+        download_btn.click(download_prompt, inputs=[score_prompt, standard_prompt, example_prompt], outputs=prompt_file)
+        clear.click(lambda: None, None, chatbot, queue=False)
+        stop.click(fn=None, inputs=None, outputs=None, cancels=[submit_event])
+
+    demo.queue()
+    demo.launch(share=False, server_port=8888, server_name='0.0.0.0')