Spaces:
Sleeping
Sleeping
# from sparkai.llm.llm import ChatSparkLLM, ChunkPrintHandler | |
# from sparkai.core.messages import ChatMessage | |
import pandas as pd | |
nv_prompt_file = pd.read_excel('汉服-女词库.xlsx') | |
na_prompt_file = pd.read_excel('汉服-男词库.xlsx') | |
nv_prompt = nv_prompt_file.to_string(index=False) | |
na_prompt = na_prompt_file.to_string(index=False) | |
# | |
# #星火认知大模型Spark Max的URL值,其他版本大模型URL值请前往文档(https://www.xfyun.cn/doc/spark/Web.html)查看 | |
# SPARKAI_URL = 'wss://spark-api.xf-yun.com/v3.5/chat' | |
# #星火认知大模型调用秘钥信息,请前往讯飞开放平台控制台(https://console.xfyun.cn/services/bm35)查看 | |
# SPARKAI_APP_ID = '11ce2152' | |
# SPARKAI_API_SECRET = 'N2ExOTc3MDc1OWZjMTkyNzFlYjA3ZTAz' | |
# SPARKAI_API_KEY = '4f6313fa6c05dea06e4e18b46e63b20f' | |
# #星火认知大模型Spark Max的domain值,其他版本大模型domain值请前往文档(https://www.xfyun.cn/doc/spark/Web.html)查看 | |
# SPARKAI_DOMAIN = 'generalv3.5' | |
# | |
# def prompt_gen(advise): | |
# spark = ChatSparkLLM( | |
# spark_api_url=SPARKAI_URL, | |
# spark_app_id=SPARKAI_APP_ID, | |
# spark_api_key=SPARKAI_API_KEY, | |
# spark_api_secret=SPARKAI_API_SECRET, | |
# spark_llm_domain=SPARKAI_DOMAIN, | |
# streaming=False, | |
# ) | |
# messages = [ChatMessage( | |
# role="user", | |
# content=advise + "\n根据建议,从触发词、种类、上衣、裙子、领子、袖子、袖口、腰饰、裙子详述中每个挑选一个词,分点描述,把英文也附在后面的括" | |
# "号里,最后下面加一条prompt,总结所有英文描述,用逗号间隔\n" + nv_prompt, | |
# )] | |
# print(messages[0].content) | |
# handler = ChunkPrintHandler() | |
# a = spark.generate([messages], callbacks=[handler]) | |
# print(a.generations[0][0].text) | |
# return a.generations[0][0].text | |
import os | |
os.environ["OPENAI_API_KEY"] = "sk-vtyR3fdgk08jmJ5e3eF6F5Ef663c4a3bAd0166C3549a1a8e" #输入网站发给你的转发key | |
os.environ["OPENAI_BASE_URL"] = "http://15.204.101.64:4000/v1" | |
from openai import OpenAI | |
def prompt_gen(advise, gender): | |
if gender == "男": | |
prompt = na_prompt | |
else: | |
prompt = nv_prompt | |
client = OpenAI() | |
completion = client.chat.completions.create( | |
model="gpt-4o", | |
messages=[ | |
{"role": "system", | |
"content": "You are a helpful assistant.",}, | |
{"role": "user", | |
"content": advise + "根据建议,从以下的触发词、种类、上衣、裙子、领子、袖子、袖口、腰饰、裙子详述中每个挑选一个词,分点描述," | |
"把英文也附在后面的括号里,最后下面加一条prompt,总结所有英文描述,用逗号间隔" + prompt, | |
} | |
] | |
) | |
print(completion.choices[0].message.content) | |
return completion.choices[0].message.content | |