从零开始-与大语言模型对话学技术-gradio篇（4）

前言

本文介绍「星火杯」认知大模型场景创新赛中的落选项目- AI命理分析系统，属于个人娱乐练手。总结提炼了往期文章精华并发掘出新的知识。
包括本地部署版本和Web在线版本，两种打包方式基于
半自动化使用.bat手动打包迁移python项目

如何把 Gradio 应用上传到 Hugging Face
往期回顾：
从零开始-与大语言模型对话学技术-gradio篇（1）

从零开始-与大语言模型对话学技术-gradio篇（2）

从零开始-与大语言模型对话学技术-gradio篇（3）

项目简介

本项目是一个集成化的AI命理分析系统,实现了星座解析、塔罗解牌、八字合婚等多种智能化命理服务。系统通过Python和Gradio实现了交互式的网页界面,用户只需要输入必要个人信息,即可获得智能的命理运势分析。
Github链接

使用说明

最新Web在线版本使用

Web公开版本已在Hugging Face开源，点击链接即可在线使用，不同于本地版本，你必须配置自己的星火APl，全平台可用
hugging face连接

本地部署版本

1. 打开AI命理分析系统V4.0进入虚拟python环境，自动检测依赖，安装环境并运行程序
   显示如下提示表示编译运行成功，打开这个连接即可进入系统

2. 你也可以打开GUI.py在末尾将demo.launch()修改为 demo.launch(share=True)
   那么你会额外获取一个随机的公开链接，你可以在任何设备上输入网址访问这个AI命理分析系统
   Running on local URL: http://127.0.0.1:7860
   Running on public URL: https://436fda53710f62fbbc.gradio.live
   This share link expires in 72 hours. For free permanent hosting and GPU upgrades,
   run gradio deploy from Terminal to deploy to Spaces (https://huggingface.co/spaces)

功能列表

• 接入星火认知大模型:你可以选择使用我的API或者自行配置 配置好API才能使用后续功能!
  

• AI星座解读:输入个人信息,获得当前月星座运势解析
  

• AI塔罗牌解读:用户提问,系统抽取塔罗牌进行占卜
  

• AI八字合婚分析:输入双方八字,智能匹配分析婚姻
  

• AI兔年运势预测:基于八字分析未来财运、事业等

• AI公司命理解析:根据个人信息以及公司名称和行业,分析公司运势。

• AI姓名配对:评估两人姓名匹配程度
  

• AI月老姻缘:分析最佳配对对象
  

• AI八字精批:输入八字信息获得对应运势的专业精批。
  

• AI姓名分析:分析输入的姓名对个人命运的影响。
  

• AI紫薇斗数解析:根据八字信息计算紫薇数值并进行解读
  

完整代码

app.py

# -*- coding = utf-8 -*-
"""
# @Time : 2023/7/31 19:33
# @Author : CSDN:FriKlogff
# @File : app.py
# @Software: PyCharm
# @Function: 请输入项目功能
"""
import os
os.system("""python -m pip install -i https://mirrors.aliyun.com/pypi/simple/ --upgrade pip setuptools
pip install -i https://mirrors.aliyun.com/pypi/simple/ websocket
pip install -i https://mirrors.aliyun.com/pypi/simple/ websocket-client
pip install -i https://mirrors.aliyun.com/pypi/simple/ gradio
pip install -i https://mirrors.aliyun.com/pypi/simple/ sxtwl
""")
from PublicFunctions import *
import gradio as gr

# 定义星座选项
signs = ["白羊座", "金牛座", "双子座", "巨蟹座", "狮子座", "处女座",
         "天秤座", "天蝎座", "射手座", "摩羯座", "水瓶座", "双鱼座"]
cards_num = [1, 2, 3, 4, 5]
months = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
days = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12,
        13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31]
hours = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23]
# 使用 Gradio 的模块化组件，构建包含五个选项卡的界面
with gr.Blocks() as demo:
    with gr.Tab("星火api配置"):
        xh_input = [
                    gr.components.Textbox(label="appid"),
                    gr.components.Textbox(label="api_secret"),
                    gr.components.Textbox(label="api_key"),
                    gr.components.Textbox(label="gpt_url")
                    ]
        xh_output = gr.components.Textbox(label="点击提交返回配置情况，请自行配置星火大模型API再使用后续功能")
        xh_button = gr.components.Button("提交")
        xh_button.click(xh_api, inputs=xh_input, outputs=xh_output
                        )
    with gr.Tab("AI星座解读"):
        horoscope_input = [gr.components.Radio(choices=["男", "女"], label="性别"),
                           gr.components.Textbox(label="姓名"),
                           gr.components.Number(label="出生年份"),
                           gr.components.Dropdown(months, label="出生月份"),
                           gr.components.Dropdown(days, label="出生日"),
                           gr.components.Dropdown(hours, label="出生时辰"),
                           gr.components.Dropdown(signs, label="选择您的星座")
                           ]
        horoscope_output = gr.components.Textbox(label="星座解读(由于我们的解析是由AI生成的，结果仅供娱乐，如果不成功请多试几次)")
        horoscope_button = gr.components.Button("提交")
        horoscope_button.click(horoscope_reading, inputs=horoscope_input, outputs=horoscope_output
                               )

    with gr.Tab("AI塔罗牌解读"):
        tarot_input = [gr.components.Textbox(label="你想问的问题"),
                       gr.components.Dropdown(cards_num, label="你想抽几张牌"),
                       ]
        tarot_output = gr.components.Textbox(label="塔罗牌解析(由于我们的解析是由AI生成的，结果仅供娱乐，如果不成功请多试几次)")
        upload_button = gr.components.Button("抽取")
        upload_button.click(tarot_reading, inputs=tarot_input, outputs=tarot_output)
    with gr.Tab("AI八字合婚分析"):
        marriage_input = [gr.components.Textbox(label="新郎姓名"),
                          gr.components.Number(label="出生年份"),
                          gr.components.Dropdown(months, label="出生月份"),
                          gr.components.Dropdown(days, label="出生日"),
                          gr.components.Dropdown(hours, label="出生时辰"),

                          gr.components.Textbox(label="新娘姓名"),
                          gr.components.Number(label="出生年份"),
                          gr.components.Dropdown(months, label="出生月份"),
                          gr.components.Dropdown(days, label="出生日"),
                          gr.components.Dropdown(hours, label="出生时辰"),
                          ]
        marriage_analysis_output = gr.components.Textbox(label="婚姻分析(由于我们的解析是由AI生成的，结果仅供娱乐，如果不成功请多试几次)")
        analyze_button = gr.components.Button("马上测算")
        analyze_button.click(marriage_bazi_analysis,
                             inputs=marriage_input,
                             outputs=marriage_analysis_output)
    with gr.Tab("AI兔年运程预测"):
        birth_year_input = [gr.components.Radio(choices=["男", "女"], label="性别"),
                            gr.components.Textbox(label="姓名"),
                            gr.components.Number(label="出生年份"),
                            gr.components.Dropdown(months, label="出生月份"),
                            gr.components.Dropdown(days, label="出生日"),
                            gr.components.Dropdown(hours, label="出生时辰"),
                            ]
        prediction_output = gr.components.Textbox(label="运程预测(由于我们的解析是由AI生成的，结果仅供娱乐，如果不成功请多试几次)")
        predict_button = gr.components.Button("预测运势")
        predict_button.click(rabbit_year_prediction,
                             inputs=birth_year_input,
                             outputs=prediction_output)
    with gr.Tab("AI公司命理解析"):
        company_name_input = [gr.components.Radio(choices=["男", "女"], label="性别"),
                              gr.components.Textbox(label="姓名"),
                              gr.components.Number(label="出生年份"),
                              gr.components.Dropdown(months, label="出生月份"),
                              gr.components.Dropdown(days, label="出生日"),
                              gr.components.Dropdown(hours, label="出生时辰"),
                              gr.components.Textbox(label="公司名称"),
                              gr.components.Textbox(label="所属行业")]
        name_analysis_output = gr.components.Textbox(label="命理分析(由于我们的解析是由AI生成的，结果仅供娱乐，如果不成功请多试几次)")
        analyze_button = gr.components.Button("分析")
        analyze_button.click(company_name_analysis,
                             inputs=company_name_input,
                             outputs=name_analysis_output)
    with gr.Tab("AI姓名配对"):
        name1_input = [gr.components.Textbox(label="姓名1"),
                       gr.components.Textbox(label="姓名2"),
                       ]
        matching_output = gr.components.Textbox(label="配对结果(由于我们的解析是由AI生成的，结果仅供娱乐，如果不成功请多试几次)")
        match_button = gr.components.Button("分析配对")
        match_button.click(name_compatibility,
                           inputs=name1_input,
                           outputs=matching_output)

    with gr.Tab("AI月老姻缘"):
        yue_lau_input = [gr.components.Radio(choices=["男", "女"], label="性别"),
                         gr.components.Textbox(label="姓名"),
                         gr.components.Number(label="出生年份"),
                         gr.components.Dropdown(months, label="出生月份"),
                         gr.components.Dropdown(days, label="出生日"),
                         gr.components.Dropdown(hours, label="出生时辰"),
                         ]
        affinity_output = gr.components.Textbox(label="姻缘分析(由于我们的解析是由AI生成的，结果仅供娱乐，如果不成功请多试几次)")
        analyze_button = gr.components.Button("分析姻缘")
        analyze_button.click(yue_lau_affinity,
                             inputs=yue_lau_input,
                             outputs=affinity_output)

    with gr.Tab("AI八字精批"):
        bazi_input = [gr.components.Radio(choices=["男", "女"], label="性别"),
                      gr.components.Textbox(label="姓名"),
                      gr.components.Number(label="出生年份"),
                      gr.components.Dropdown(months, label="出生月份"),
                      gr.components.Dropdown(days, label="出生日"),
                      gr.components.Dropdown(hours, label="出生时辰"),
                      ]
        analysis_output = gr.components.Textbox(label="精批结果(由于我们的解析是由AI生成的，结果仅供娱乐，如果不成功请多试几次)")
        batch_button = gr.components.Button("八字精批")
        batch_button.click(bazi_analysis,
                           inputs=bazi_input,
                           outputs=analysis_output)

    with gr.Tab("AI姓名分析"):
        name_input = [gr.components.Radio(choices=["男", "女"], label="性别"),
                      gr.components.Textbox(label="姓名")]
        name_output = gr.components.Textbox(label="命理分析(由于我们的解析是由AI生成的，结果仅供娱乐，如果不成功请多试几次)")
        analyze_button = gr.components.Button("分析姓名")
        analyze_button.click(name_analysis,
                             inputs=name_input,
                             outputs=name_output)
    with gr.Tab("AI紫薇斗数解析"):
        zhiwei_input = [gr.components.Radio(choices=["男", "女"], label="性别"),
                        gr.components.Textbox(label="姓名"),
                        gr.components.Number(label="出生年份"),
                        gr.components.Dropdown(months, label="出生月份"),
                        gr.components.Dropdown(days, label="出生日"),
                        gr.components.Dropdown(hours, label="出生时辰"),
                        ]
        zhiwei_output = gr.components.Textbox(label="紫薇解读(由于我们的解析是由AI生成的，结果仅供娱乐，如果不成功请多试几次)")
        zhiwei_button = gr.components.Button("解读运势")
        zhiwei_button.click(zhiwei_analysis,
                            inputs=zhiwei_input,
                            outputs=zhiwei_output)
demo.launch()
# demo.launch(share=True)

PublicFunctions.py

# -*- coding = utf-8 -*-
"""
# @Time : 2023/7/31 19:35
# @Author : CSDN:FriKlogff
# @File : PublicFunctions.py
# @Software: PyCharm
# @Function: 请输入项目功能
"""

import os
os.system("""python -m pip install -i https://mirrors.aliyun.com/pypi/simple/ --upgrade pip setuptools
pip install -i https://mirrors.aliyun.com/pypi/simple/ websocket
pip install -i https://mirrors.aliyun.com/pypi/simple/ websocket-client
pip install -i https://mirrors.aliyun.com/pypi/simple/ gradio
pip install -i https://mirrors.aliyun.com/pypi/simple/ sxtwl
""")
import sxtwl
from XhApi import *
import XhApi

# print(XhApi.response_content)
Gan = ["甲", "乙", "丙", "丁", "戊", "己", "庚", "辛", "壬", "癸"]
Zhi = ["子", "丑", "寅", "卯", "辰", "巳", "午", "未", "申", "酉", "戌", "亥"]
appid = ''
api_secret = ''
api_key = ''
gpt_url = ''


def generate_bazi(year, month, day, hour):
    year = int(year)
    month = int(month)
    day = int(day)
    hour = int(hour)

    date = sxtwl.fromSolar(year, month, day)
    # 获取年柱
    yTG = date.getYearGZ()
    ganzhi_year = Gan[yTG.tg] + Zhi[yTG.dz]

    # 获取月柱
    mTG = date.getMonthGZ()
    ganzhi_month = Gan[mTG.tg] + Zhi[mTG.dz]

    # 获取日柱
    dTG = date.getDayGZ()
    ganzhi_day = Gan[dTG.tg] + Zhi[dTG.dz]

    # 获取时柱
    dayGan = dTG.tg
    hTG = sxtwl.getShiGz(dayGan, hour)
    ganzhi_hour = Gan[hTG.tg] + Zhi[hTG.dz]

    return f"{
      
      ganzhi_year}年{
      
      ganzhi_month}月{
      
      ganzhi_day}日{
      
      ganzhi_hour}时"


def xh_api( user_appid, user_api_secret, user_api_key, user_gpt_url):
    global appid, api_secret, api_key, gpt_url
    if user_appid == '' or user_api_secret == '' or user_api_key == '' or user_gpt_url == '':
        return "any api cannot be empty!"
    appid = str(user_appid)
    api_secret = str(user_api_secret)
    api_key = str(user_api_key)
    gpt_url = str(user_gpt_url)
    # print(type(appid), type(api_secret), type(api_key), type(gpt_url))
    # print(appid, api_secret, api_key, gpt_url)
    return "appid = "+appid+"\napi_secret = "+api_secret+"\napi_key = "+api_key+"\ngpt_url = "+gpt_url


def horoscope_reading(sex, name, birth_year, birth_month, birth_day, birth_hour, star):
    XhApi.response_content = ''
    global appid, api_secret, api_key, gpt_url
    birth_year = str(int(birth_year))
    if sex is None:
        return "sex cannot be empty!"
    if name == '' or birth_year == '' or birth_month == '' or birth_day == '' or birth_hour == '' or star == '':
        return "Name or birth_year or birth_month or birth_day or star cannot be empty!"
    if birth_year == "0":
        return "0 is not a suitable value of birth_year!"
    template = "假设你是一位专业的星座运势分析师,根据客户提供的出生日期和时间,你需要进行以下几方面的详细分析工作:\n" \
               "1. 分析用户的星座及性格特点\n" \
               "2. 根据月球周期判断事业和学业趋势\n" \
               "3. 提供维持提升感情的建议\n" \
               "4. 预测财务收入和投资趋势\n" \
               "5. 提出健康保健建议\n" \
               "作为专业分析师,你需要用通俗语言解释理论,并提供专业建议\n"
    template += "客户信息:\n"
    template += "性别:{sex}\n"
    template += "姓名:{name}\n"
    template += "星座:{star}\n"
    template += "出生日期:{birth_year}-{birth_month}-{birth_day}-{birth_hour}"
    question = template.format(sex=sex, name=name, birth_year=birth_year, birth_month=birth_month, birth_day=birth_day,
                               birth_hour=birth_hour,
                               star=star)
    # print(appid, api_secret, api_key, gpt_url)
    return main(
        appid=appid,
        api_secret=api_secret,
        api_key=api_key,
        gpt_url=gpt_url,
        question=question)


def tarot_reading(question, num_cards):
    XhApi.response_content = ''
    global appid, api_secret, api_key, gpt_url
    if question == '' or num_cards is None or num_cards == 0:
        return "question ornum_cards cannot be empty!"
    template = "假设你是一位专业的塔罗牌占卜师。用户提出的问题是:{question}。"
    template += "根据用户的问题,你需要为TA抽取{num_cards}张塔罗牌,"
    template += "解读每张塔罗牌的含义,"
    template += "综合牌面分析用户所问的问题,"
    template += "并根据占卜结果给予专业的建议。"
    template += "具体来说,你需要:\n"
    template += "1. 为用户抽取指定数量的塔罗牌\n"
    template += "2. 逐一解析每张塔罗牌的符号和含义\n"
    template += "3. 综合各牌面意义,对用户提问进行占卜分析\n"
    template += "4. 根据占卜结果,给出专业建议或预言"
    question = template.format(question=question, num_cards=int(num_cards))
    return main(
        appid=appid,
        api_secret=api_secret,
        api_key=api_key,
        gpt_url=gpt_url,
        question=question)


def marriage_bazi_analysis(name_husband, birth_year_husband, birth_month_husband, birth_day_husband, birth_hour_husband,
                           name_wife, birth_year_wife, birth_month_wife, birth_day_wife, birth_hour_wife):
    XhApi.response_content = ''
    global appid, api_secret, api_key, gpt_url
    birth_year_husband = int(birth_year_husband)
    birth_year_wife = int(birth_year_wife)
    if name_husband == '' or birth_year_husband == '' or birth_month_husband == '' or birth_day_husband == '' or birth_hour_husband == '' \
            or name_wife == '' or birth_year_wife == '' or birth_month_wife == '' or birth_day_wife == '' or birth_hour_wife == '':
        return "Name or birth_year or birth_month or birth_day  cannot be empty!"
    if birth_year_husband == 0 or birth_year_wife == 0:
        return "0 is not a suitable value of birth_year!"
    bazi_husband = generate_bazi(birth_year_husband, birth_month_husband, birth_day_husband, birth_hour_husband)
    bazi_wife = generate_bazi(birth_year_wife, birth_month_wife, birth_day_wife, birth_hour_wife)
    # print(bazi_wife, bazi_husband)
    template = "假设你是一位专业的八字合婚分析师,你正在为一对新人进行八字合婚分析。" \
               "分析基于八字五行、十神、四柱的原理判断两人姻缘。重点看天格、年格五行相生相克。" \
               "以下是他们的基本信息\n"
    template += "新郎信息:\n"
    template += "姓名:{name_husband}\n"
    template += "出生日期:{birth_year_husband}-{birth_month_husband}-{birth_day_husband}\n"  # 根据用户的选择生成问题
    template += "八字:{bazi_husband}\n"
    template += "新娘信息:\n"
    template += "姓名:{name_wife}\n"
    template += "出生日期:{birth_year_wife}-{birth_month_wife}-{birth_day_wife}\n"  # 根据用户的选择生成问题
    template += "八字:{bazi_wife}\n"
    template += "作为资深的合婚分析师,你需要:\n"
    template += "1. 分析两人八字五行相生相克关系\n"
    template += "2. 比较两人十神是否匹配\n"
    template += "3. 检查四柱运势是否协调\n"
    template += "4. 给出姻缘匹配度及建议\n"
    question = template.format(name_husband=name_husband, name_wife=name_wife, bazi_husband=bazi_husband,
                               bazi_wife=bazi_wife, birth_year_husband=birth_year_husband,
                               birth_month_husband=birth_month_husband, birth_day_husband=birth_day_husband,
                               birth_year_wife=birth_year_wife, birth_month_wife=birth_month_wife,
                               birth_day_wife=birth_day_wife)
    return main(
        appid=appid,
        api_secret=api_secret,
        api_key=api_key,
        gpt_url=gpt_url,
        question=question)



# 兔年运程
def rabbit_year_prediction(sex, name, birth_year, birth_month, birth_day, birth_hour):
    XhApi.response_content = ''
    global appid, api_secret, api_key, gpt_url

    birth_year = int(birth_year)
    bazi = generate_bazi(int(birth_year), int(birth_month), int(birth_day), int(birth_hour))
    if sex is None:
        return "sex cannot be empty!"
    if name == '' or birth_year == '' or birth_month == '' or birth_day == '' or birth_hour == '':
        return "Name or birth_year or birth_month or birth_day cannot be empty!"
    if birth_year == 0:
        return "0 is not a suitable value of birth_year!"
    template = "假设你是一位专业的命理师,仔细分析客户信息,结合通胜原理,考量客户的五行八字、天干合化等,对在兔年客户的事业、财富、姻缘等命局进行预测,并给出建议。"
    template += "\n客户信息:\n"
    template += "性别:{sex}\n"
    template += "姓名:{name}\n"
    template += "出生日期:{birth_year}-{birth_month}-{birth_day}-{birth_hour}\n"
    template += "八字:{bazi}\n"
    template += "具体来说,你需要:\n"
    template += "1. 检查客户八字和五行属性\n"
    template += "2. 分析天干合化对命局的影响\n"
    template += "3. 考量通胜原理对运势的作用\n"
    template += "4. 对事业、财富、姻缘等命局给出预测\n"
    template += "5. 提供专业建议"
    question = template.format(sex=sex, name=name, birth_year=birth_year, birth_month=birth_month, birth_day=birth_day,
                               birth_hour=birth_hour, bazi=bazi)
    return main(
        appid=appid,
        api_secret=api_secret,
        api_key=api_key,
        gpt_url=gpt_url,
        question=question)



# 公司测名
def company_name_analysis(sex, name, birth_year, birth_month, birth_day, birth_hour, company_name, industry):
    XhApi.response_content = ''
    global appid, api_secret, api_key, gpt_url
    birth_year = int(birth_year)
    if sex is None:
        return "sex cannot be empty!"
    if name == '' or birth_year == '' or birth_month == '' or birth_day == '' or birth_hour == '' or company_name == '' or industry == '':
        return "Name or birth_year or birth_month or birth_day  or company_name  or industry cannot be empty!"
    if birth_year == 0:
        return "0 is not a suitable value of birth_year!"
    template = "假设你是一位公司命理专家,根据立命八字学说,姓名、公司名与行业之间存在相生相克的关系," \
               "需要综合考量五行、八卦、吉凶等理论,分析它们之间的互动对企业发展的影响," \
               "发掘其中蕴含的福禄文星,提出建议以改善财运。"
    template += "\n客户信息:\n"
    template += "性别:{sex}\n"
    template += "姓名:{name}\n"
    template += "出生日期:{birth_year}-{birth_month}-{birth_day}-{birth_hour}\n"
    template += "公司名:{company_name}\n"
    template += "行业:{industry}\n"
    template += "具体来说,你需要:\n"
    template += "1. 分析客户姓名五行属性\n"
    template += "2. 考量公司名五行与行业五行关系\n"
    template += "3. 判断相生相克对企业运势的影响\n"
    template += "4. 发掘姓名、公司名蕴含的福星\n"
    template += "5. 提出改善企业财运的专业建议"

    question = template.format(sex=sex, name=name, birth_year=birth_year, birth_month=birth_month, birth_day=birth_day,
                               birth_hour=birth_hour,
                               company_name=company_name, industry=industry)
    return main(
        appid=appid,
        api_secret=api_secret,
        api_key=api_key,
        gpt_url=gpt_url,
        question=question)



# 姓名配对
def name_compatibility(name1, name2):
    XhApi.response_content = ''
    global appid, api_secret, api_key, gpt_url
    if name1 == '' or name2 == '':
        return "name1 or name2 cannot be empty!"
    template = "假设你是一位姓名学专家。用户提供了两人的姓名:{name1}和{name2}。"
    template += "作为专家,你需要分析他们两人姓名的五行、笔画等特征,"
    template += "判断姓名间的五行关系是否协调、笔画关系是否匹配,"
    template += "从姓名学角度出发,分析这两人的姓名是否配对。"
    template += "具体来说,你需要:\n"
    template += "1. 分析{name1}的五行属性和笔画数\n"
    template += "2. 分析{name2}的五行属性和笔画数\n"
    template += "3. 判断两人姓名的五行相生相克关系\n"
    template += "4. 判断两人姓名笔画数差是否合适\n"
    template += "5. 从姓名学角度给出配对建议\n"
    template += "最后要给出专业建议,说明这对姓名的搭配优劣势。"
    question = template.format(name1=name1, name2=name2)
    return main(
        appid=appid,
        api_secret=api_secret,
        api_key=api_key,
        gpt_url=gpt_url,
        question=question)



# 月老姻缘
def yue_lau_affinity(sex, name, birth_year, birth_month, birth_day, birth_hour):
    XhApi.response_content = ''
    global appid, api_secret, api_key, gpt_url
    birth_year = int(birth_year)
    bazi = generate_bazi(int(birth_year), int(birth_month), int(birth_day), int(birth_hour))

    if sex is None:
        return "sex cannot be empty!"
    if name == '' or birth_year == '' or birth_month == '' or birth_day == '' or birth_hour == '':
        return "Name or birth_year or birth_month or birth_day cannot be empty!"
    if birth_year == 0:
        return "0 is not a suitable value of birth_year!"
    template = "假设你是一位月老姻缘专家。有客户需要你的帮助,其信息如下:\n"
    template += "性别:{sex}\n"
    template += "姓名:{name}\n"
    template += "出生日期:{birth_year}-{birth_month}-{birth_day}-{birth_hour}\n"
    template += "八字:{bazi}\n"
    template += "作为月老专家,你需要基于客户的姓名、性别、出生日期等信息,"
    template += "来分析其感情运势、最佳配对对象,"
    template += "给出专业的建议,帮助客户找到适合的另一半。"
    template += "具体来说,你需要:\n"
    template += "1. 分析客户八字姻缘格局\n"
    template += "2. 考量姓名数字对婚姻的影响\n"
    template += "3. 判断最佳配对对象的特征\n"
    template += "4. 提出改善感情运势的建议"
    question = template.format(sex=sex, name=name, birth_year=birth_year, birth_month=birth_month, birth_day=birth_day,
                               birth_hour=birth_hour, bazi=bazi)
    return main(
        appid=appid,
        api_secret=api_secret,
        api_key=api_key,
        gpt_url=gpt_url,
        question=question)



# 八字精批
def bazi_analysis(sex, name, birth_year, birth_month, birth_day, birth_hour):
    XhApi.response_content = ''
    global appid, api_secret, api_key, gpt_url
    birth_year = int(birth_year)
    if sex is None:
        return "sex cannot be empty!"
    if name == '' or birth_year == '' or birth_month == '' or birth_day == '':
        return "Name or birth_year or birth_month or birth_day cannot be empty!"
    if birth_year == 0:
        return "0 is not a suitable value of birth_year!"
    bazi = generate_bazi(int(birth_year), int(birth_month), int(birth_day), int(birth_hour))
    template = "假设你是一位资深的八字命理师。有客户需要你对其八字进行专业精批,其信息如下:\n"

    template += "性别:{sex}\n"
    template += "姓名:{name}\n"
    template += "八字:{bazi}\n"

    template += "作为八字命理专家,你需要根据客户的八字,"
    template += "分析事业财运、健康等方面的运势趋势,"
    template += "具体来说,你需要:\n"
    template += "1. 检查天干五行对事业财运的影响\n"
    template += "2. 分析八字各宫协调性和局部格局\n"
    template += "3. 指出八字优势和劣势\n"
    template += "4. 提出合理的改善建议"

    question = template.format(sex=sex, name=name, bazi=bazi)
    return main(
        appid=appid,
        api_secret=api_secret,
        api_key=api_key,
        gpt_url=gpt_url,
        question=question)


def zhiwei_analysis(sex, name, birth_year, birth_month, birth_day, birth_hour):
    XhApi.response_content = ''
    global appid, api_secret, api_key, gpt_url
    birth_year = int(birth_year)
    if sex is None:
        return "sex cannot be empty!"
    if name == '' or birth_year == '' or birth_month == '' or birth_day == '':
        return "Name or birth_year or birth_month or birth_day cannot be empty!"
    if birth_year == 0:
        return "0 is not a suitable value of birth_year!"
    # print(sex, name, birth_year, birth_month, birth_day)
    bazi = generate_bazi(int(birth_year), int(birth_month), int(birth_day), int(birth_hour))
    template = "假设你是一位紫薇斗数专家,接收到客户的出生八字后,你会依次完成以下步骤:\n" \
               "1. 计算该八字的紫微星位置,代表其总体运势\n" \
               "2. 分析年柱运程,判断事业财运\n" \
               "3. 分析月柱运程,判断感情运\n" \
               "4. 分析日柱运程,判断健康运\n" \
               "5. 综合四柱运势对该客户的综合运势做出详细的预言分析\n" \
               "客户信息：\n" \
               "性别：{sex}\n" \
               "姓名：{name}\n" \
               "八字为：{bazi}"

    question = template.format(sex=sex, name=name, bazi=bazi)
    return main(
        appid=appid,
        api_secret=api_secret,
        api_key=api_key,
        gpt_url=gpt_url,
        question=question)



# 姓名分析

def name_analysis(sex, name):
    XhApi.response_content = ''
    global appid, api_secret, api_key, gpt_url
    if sex is None:
        return "sex cannot be empty!"
    if name == '':
        return "Name cannot be empty!"
    template = "假设你是一位姓名学专家,请根据客户的姓名,分析其一生运势。\n"
    template += "要点包括:\n"
    template += "- 姓名的谐音是否吉利\n"
    template += "- 姓名笔画多寡对品性的影响\n"
    template += "- 单名双名优劣\n"
    template += "客户的姓名为 {name},性别为{sex},分析对其事业、婚姻、健康等方面的影响,并提出建议。"
    template += "具体来说,你需要\n"
    template += "1. 分析客户姓名谐音\n"
    template += "2. 判断姓名笔画数命理含义\n"
    template += "3. 讨论单名双名特点\n"
    template += "4. 分析姓名对运势各方面的影响\n"
    template += "5. 提出改善命运的专业建议"
    question = template.format(name=name, sex=sex)
    return main(
        appid=appid,
        api_secret=api_secret,
        api_key=api_key,
        gpt_url=gpt_url,
        question=question)

XhApi.py

# -*- coding = utf-8 -*-
"""
# @Time : 2023/7/20 12:37
# @Author : CSDN:FriKlogff
# @File : XhApi.py
# @Software: PyCharm
# @Function: 星火大模型API
"""
import os
os.system("""python -m pip install -i https://mirrors.aliyun.com/pypi/simple/ --upgrade pip setuptools
pip install -i https://mirrors.aliyun.com/pypi/simple/ websocket
pip install -i https://mirrors.aliyun.com/pypi/simple/ websocket-client
pip install -i https://mirrors.aliyun.com/pypi/simple/ gradio
pip install -i https://mirrors.aliyun.com/pypi/simple/ sxtwl
""")
import _thread as thread  # 导入线程模块
import base64  # 导入base64编码模块
import datetime  # 导入datetime模块
import hashlib  # 导入hashlib模块
import hmac  # 导入hmac模块
import json  # 导入json模块
from urllib.parse import urlparse  # 从urllib.parse导入urlparse用于url解析
import ssl  # 导入ssl模块
from datetime import datetime  # 从datetime导入datetime类
from time import mktime  # 从time导入mktime用于生成时间戳
from urllib.parse import urlencode  # 从urllib.parse导入urlencode用于编码请求参数
from wsgiref.handlers import format_date_time  # 从wsgiref.handlers导入format_date_time用于格式化时间

import websocket  # 导入websocket模块

response_content = ""


# 请求参数类
class Ws_Param:
    # 初始化
    def __init__(self, APPID, APIKey, APISecret, gpt_url):
        self.APPID = APPID  # 应用ID
        self.APIKey = APIKey  # API Key
        self.APISecret = APISecret  # API Secret
        self.host = urlparse(gpt_url).netloc  # 从url解析出host
        self.path = urlparse(gpt_url).path  # 从url解析出path
        self.gpt_url = gpt_url  # 完整的url

    # 生成签名和url的方法
    def create_url(self):
        now = datetime.now()  # 当前时间
        date = format_date_time(mktime(now.timetuple()))  # 格式化的时间戳
        # 拼接签名原文
        signature_origin = "host: " + self.host + "\n"
        signature_origin += "date: " + date + "\n"
        signature_origin += "GET " + self.path + " HTTP/1.1"
        # 生成签名
        signature_sha = hmac.new(self.APISecret.encode('utf-8'), signature_origin.encode('utf-8'),
                                 digestmod=hashlib.sha256).digest()
        signature_sha_base64 = base64.b64encode(signature_sha).decode(encoding='utf-8')
        # 生成授权header
        authorization_origin = f'api_key="{
      
      self.APIKey}", algorithm="hmac-sha256", headers="host date request-line", signature="{
      
      signature_sha_base64}"'
        authorization = base64.b64encode(authorization_origin.encode('utf-8')).decode(encoding='utf-8')
        # 生成url参数字典
        v = {
    
    
            "authorization": authorization,
            "date": date,
            "host": self.host
        }
        # 构造最终url
        url = self.gpt_url + '?' + urlencode(v)
        return url


# 收到websocket错误的处理
def on_error(ws, error):
    print("### error:", error)


# 收到websocket关闭的处理
def on_close(ws):
    print("### closed ###")


# 收到websocket连接建立的处理
def on_open(ws):
    thread.start_new_thread(run, (ws,))


# 发送请求的方法
def run(ws, *args):
    data = json.dumps(gen_params(appid=ws.appid, question=ws.question))
    ws.send(data)


# 收到websocket消息的处理
def on_message(ws, message):
    print(message)
    data = json.loads(message)
    code = data['header']['code']
    if code != 0:
        print(f'请求错误: {
      
      code}, {
      
      data}')
        ws.close()
    else:
        choices = data["payload"]["choices"]
        status = choices["status"]
        content = choices["text"][0]["content"]
        print(content, end='')
        global response_content
        response_content += content
        if status == 2:
            ws.close()


# 生成请求参数
def gen_params(appid, question):
    """
    通过appid和用户的提问来生成请参数
    """
    data = {
    
    
        "header": {
    
    
            "app_id": appid,
            "uid": "1234"
        },
        "parameter": {
    
    
            "chat": {
    
    
                "domain": "general",
                "random_threshold": 0.5,
                "max_tokens": 2048,
                "auditing": "default"
            }
        },
        "payload": {
    
    
            "message": {
    
    
                "text": [
                    {
    
    "role": "user", "content": question}
                ]
            }
        }
    }
    return data




def main(appid, api_key, api_secret, gpt_url, question):
    wsParam = Ws_Param(appid, api_key, api_secret, gpt_url)
    websocket.enableTrace(False)
    wsUrl = wsParam.create_url()
    ws = websocket.WebSocketApp(wsUrl, on_message=on_message, on_error=on_error, on_close=on_close, on_open=on_open)
    ws.appid = appid
    ws.question = question
    ws.run_forever(sslopt={
    
    "cert_reqs": ssl.CERT_NONE})
    return response_content