demo init

.gitignore

@@ -0,0 +1,151 @@
+# Azure az webapp deployment details
+.azure
+*_env
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# NPM
+npm-debug.log*
+node_modules
+static/
+
+.DS_Store
+wandb/
+codebase/

app.py

@@ -0,0 +1,275 @@
+import gradio as gr
+from langchain_openai import AzureChatOpenAI
+from langchain_core.prompts import (
+    ChatPromptTemplate,
+    SystemMessagePromptTemplate,
+    HumanMessagePromptTemplate,
+)
+from langchain.schema import StrOutputParser
+import json
+import os
+from dotenv import load_dotenv
+
+
+example1 = """
+业务员:您好女士，欢迎光临徐家汇店! 请问您是第一次来我们店吗?
+顾客:是的，我是通过广告来的。
+业务员: 太好了，欢迎新朋友! 请问您今天是一个人来的吗?
+顾客:对，我一个人。
+业务员: 很好。您在店里计划逗留多久呢? 是想快速选购，还是想慢慢挑选看看?
+顾客:我想大概要逗留30分钟到1小时。
+业务员:好的，您有什么特别感兴趣的商品吗? 比如我们的潜水表或者万年历表?
+顾客: 我对潜水表很感兴趣，我看到过一些广告，星级的话，我给它3颗星吧
+业务员:潜水表确实是我们的明星产品。您想试戴看看吗?
+顾客:好的，我想试戴潜水表。
+业务员:没问题。试戴之后如果您喜欢，我们有哪些支付方式可以为您服务? 我们支持信用卡、支付宝、微信、会员预充值卡，当然也接受现金
+顾客: 我再看看吧。
+业务员:明白了。在您挑选商品的过程中，有什么我可以帮助的，或者有什么特别的反馈吗?
+顾客: 要是有更多的款式就好了，我过一阵再来看看
+业务员:好的，如果有任何问题或需要帮助，请随时告诉我
+""".strip()
+
+
+example2 = """
+业务员: 您好，欢迎光临徐家汇店! 请问我可以称呼您为先生吗?
+顾客: 是的，你可以这么叫我。
+业务员: 很高兴认识您，先生。今天您是第一次来我们店吗?
+顾客:不是，我之前来过。
+业务员: 好的，欢迎再次光临!您今天想要逛逛还是看到了我们的广告有特别的需求呢?
+顾客:我是看到了广告，对你们的潜水表和万年历表都挺感兴趣的。
+业务员: 太好了!这两款表都是我们的热销产品。您想先试戴哪一款?
+顾客:我想先试戴一下潜水表。
+(顾客试戴后)
+业务员: 感觉如何，先生? 我们还有万年历表，您也想试试看吗?
+顾客:潜水表感觉挺不错的，万年历表我也试试。
+(顾客试戴万年历表后)
+业务员: 两款表您都试戴了，有没有哪一款特别喜欢的?
+顾客: 我还是比较喜欢潜水表，就买这个吧。
+业务员:好的，非常精明的选择!请问您打算如何支付呢?
+顾客:我用信用卡支付。
+业务员:没问题，先生。在完成购买之后，您还有其他需要我帮助的吗?
+顾客: 没有了，谢谢!
+业务员: 不客气，如果您对我们的产品或服务有任何建议，欢迎随时告诉我们。祝您今天过得愉快!
+""".strip()
+
+example3 = """
+今天在静安寺店接待了一位新顾客，是位女士，和朋友两人一起来店。\
+他是通过我们的广告被吸引来的，对潜水表表现出了浓厚的兴趣，并给了高达3颗星的评价。\
+他计划在店里停留大约30分钟到1小时。我为他展示了潜水表，并且他决定试戴。\
+最终用户并没有购买。目前他没有提出特别的需求或反馈，但我会保持关注，确保他的购物体验顺利愉快
+""".strip()
+
+example4 = """
+新天地店今天接待了一位之前来过的男士顾客，他是被我们的广告吸引来的。\
+他对潜水表和万年历表都表示了兴趣，最后试戴了这两款表。他特别喜欢潜水表，并决定购买。\
+整个过程中，他的停留时间大约是30分钟。顾客最终选择使用信用卡支付。整个沟通过程很顺利，顾客没有提出额外的要求或反馈。
+""".strip()
+
+load_dotenv()
+
+demploy_model = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME_GPT4T")
+api_version = os.getenv("OPENAI_API_VERSION")
+api_key = os.getenv("OPENAI_API_KEY")
+endpoint = os.getenv("OPENAI_API_BASE")
+
+print(demploy_model, api_version, api_key, endpoint)
+
+
+def extract_json(s):
+    i = s.index("{")
+    count = 1  # 当前所在嵌套深度，即还没闭合的'{'个数
+    for j, c in enumerate(s[i + 1 :], start=i + 1):
+        if c == "}":
+            count -= 1
+        elif c == "{":
+            count += 1
+        if count == 0:
+            break
+    assert count == 0  # 检查是否找到最后一个'}'
+    return s[i : j + 1]
+
+
+def process_text_LLM(text):
+    llm = AzureChatOpenAI(
+        temperature=0,
+        deployment_name="testgpt4turbo",
+        openai_api_version="2023-05-15",
+        azure_endpoint="https://testbotgpt4.openai.azure.com/",
+        api_key="89bda4e620a348858c24dd985598fa99",
+    )
+
+    prompt = ChatPromptTemplate.from_messages(
+        [
+            SystemMessagePromptTemplate.from_template(
+                """你是一个专业客服人员，需要根据和用户的对话记录，提取出相关的信息，填入对应的表单。
+表单填写规则描述: 表单1 为必填，如果顾客有试戴或购物行为，则���要填写表单2。如果没有提取到对应的信息，不要编造，而是填入 null。
+
+====================表单1描述====================
+| 表单项     | 类型         | 子选项           |
+|------------|--------------|------------------|
+| 顾客性别（gender）   | 单选🔘       | 男               |
+|            |              | 女               |
+| 人数 （number）      | 输入框       | -                |
+| 门店名称 (store_name)  | 下拉菜单，单选| 徐家汇店         |
+|            |              | 静安寺店         |
+|            |              | 新天地店         |
+| 客流类型 (type)  | 下拉菜单，单选| 新用户 - 闲逛    |
+|            |              | 新用户 - 广告引流 |
+|            |              | 老用户 - 闲逛    |
+|            |              | 老用户 - 广告引流 |
+| 停留时长 (duration)  | 下拉菜单，单选| 10 分钟以内      |
+|            |              | 10 - 30 分钟     |
+|            |              | 30 - 60 分钟     |
+|            |              | 60 分钟以上      |
+| 是否购物 (shopping)  | 单选🔘       | 是               |
+|            |              | 否               |
+============================================================
+
+====================表单2描述：====================
+| 表单项                 | 类型         | 子选项    | 备注                                                     |
+|----------------------|--------------|----------|--------------------------------------------------------|
+| 顾客感兴趣的商品 (interested)       | 多选         | 潜水表    |             |
+|                      |              | 万年历表  |                                                        |
+| 顾客感兴趣的程度 (interest_levels)       | 单选(数字)         | 1    |            |
+|                      |              | 2    |               |
+|                      |             |  3    |             |
+| 用户试戴的商品 (tried)         | 多选         | 潜水表    |                                                        |
+|                      |              | 万年历表  |                                                        |
+| 最终成单商品 (final)          | 多选         | 潜水表    | 不选择 = 没有成单。每个选项可以选择多次（比如，一次买了两个潜水表） |
+|                      |              | 万年历表  |                                                        |
+| 支付方式 (payment)             | 单选         | 信用卡    |                                                        |
+|                      |              | 支付宝    |                                                        |
+|                      |              | 微信      |                                                        |
+|                      |              | 会员预充值卡 |                                                        |
+|                      |              | 现金      |                                                        |
+|                      |              | 其它      |                                                        |
+| 其它沟通内容和顾客反馈 (feedback)  | 输入框       | -        |                                                        |
+============================================================
+
+====================输出例子====================
+{{
+    "gender": "女",
+    "number": 1,
+    "store_name": "徐家汇店",
+    "type": "新用户 - 闲逛",
+    "duration": "10 分钟以内",
+    "shopping": "否",
+    "interested": "潜水表",
+    "interest_levels": 2,
+    "tried": null,
+    "final": "万年历表",
+    "payment": "微信",
+    "feedback": "哈哈哈哈",
+}}
+============================================================
+
+必须按照以上输出例子里的 JSON 格式输出。没有提取到的信息填入 null。只有 JSON 本身，不要有其他字符输出。
+""".strip()
+            ),
+            HumanMessagePromptTemplate.from_template("对话记录: ```{input}```"),
+        ]
+    )
+
+    chain = prompt | llm | StrOutputParser()
+    return chain.invoke({"input": text})
+
+
+with gr.Blocks() as demo:
+    with gr.Row():
+
+        with gr.Column():
+            text = gr.Text()
+            submit_btn = gr.Button(value="Submit")
+            examples = gr.Examples(
+                examples=[example1, example2, example3, example4], inputs=[text]
+            )
+
+        with gr.Column():
+            gender = gr.Radio(label="顾客性别", choices=["男", "女"])
+            number = gr.Number(label="人数")
+            store_name = gr.Dropdown(
+                label="门店名称", choices=["徐家汇店", "静安寺店", "新天地店"]
+            )
+            type = gr.Dropdown(
+                label="客流类型",
+                choices=[
+                    "新用户 - 闲逛",
+                    "新用户 - 广告引流",
+                    "老用户 - 闲���",
+                    "老用户 - 广告引流",
+                ],
+            )
+            duration = gr.Dropdown(
+                label="停留时长",
+                choices=[
+                    "10 分钟以内",
+                    "10 - 30 分钟",
+                    "30 - 60 分钟",
+                    "60 分钟以上",
+                ],
+            )
+            shopping = gr.Radio(label="是否购物", choices=["是", "否"])
+
+        with gr.Column():
+            interested = gr.CheckboxGroup(
+                label="顾客感兴趣的商品", choices=["潜水表", "万年历表"]
+            )
+            interest_levels = gr.Radio(
+                choices=[("🌟", 1), ("🌟🌟", 2), ("🌟🌟🌟", 3)],
+                label="感兴趣程度 (为最近选中的商品评级)",
+            )
+            tried = gr.CheckboxGroup(
+                label="用户试戴的商品", choices=["潜水表", "万年历表"]
+            )
+            final = gr.CheckboxGroup(
+                label="最终成单商品", choices=["潜水表", "万年历表"]
+            )
+            payment = gr.Radio(
+                label="支付方式",
+                choices=["信用卡", "支付宝", "微信", "会员预充值卡", "现金", "其它"],
+            )
+            feedback = gr.Textbox(label="其它沟通内容和顾客反馈")
+
+        def process_form(text):
+
+            res = process_text_LLM(text)
+            print(res)
+            strRes = extract_json(res)
+            print(strRes)
+            jsonRes = json.loads(strRes, strict=False)
+
+            return {
+                gender: jsonRes["gender"],
+                number: jsonRes["number"],
+                store_name: jsonRes["store_name"],
+                type: jsonRes["type"],
+                duration: jsonRes["duration"],
+                shopping: jsonRes["shopping"],
+                interested: jsonRes["interested"],
+                interest_levels: jsonRes["interest_levels"],
+                tried: jsonRes["tried"],
+                final: jsonRes["final"],
+                payment: jsonRes["payment"],
+                feedback: jsonRes["feedback"],
+            }
+
+        submit_btn.click(
+            fn=process_form,
+            inputs=text,
+            outputs=[
+                gender,
+                number,
+                store_name,
+                type,
+                duration,
+                shopping,
+                interested,
+                interest_levels,
+                tried,
+                final,
+                payment,
+                feedback,
+            ],
+        )
+
+demo.launch()

requirements.txt

@@ -0,0 +1,4 @@
+gradio==4.14.0
+langchain==0.1.10
+langchain-openai==0.0.8
+python-dotenv==1.0.1