Upload prompt_gen.py

prompt_gen.py

@@ -0,0 +1,38 @@
+import pandas as pd
+import os
+from openai import OpenAI
+
+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)
+
+os.environ["OPENAI_API_KEY"] = "sk-vtyR3fdgk08jmJ5e3eF6F5Ef663c4a3bAd0166C3549a1a8e"
+os.environ["OPENAI_BASE_URL"] = "http://15.204.101.64:4000/v1"
+
+
+def prompt_gen(advise, gender):
+    if gender == "男":
+        prompt = na_prompt
+        trigger = "A Hanfu"
+    else:
+        prompt = nv_prompt
+        trigger = "a Hanfu"
+
+    client = OpenAI()
+    completion = client.chat.completions.create(
+        model="gpt-4o",
+        messages=[
+            {"role": "system",
+             "content": "You are a helpful assistant.",},
+            {"role": "user",
+             "content": advise + "根据建议，从以下的触发词、种类、上衣、裙子、领子、袖子、袖口、腰饰、裙子详述中每个挑选一个词，分点描述，触发词固定选择为" + trigger +
+                                 ", 把英文也附在后面的括号里，然后在下面先加一条prompt，标题为‘# prompt’,并在此标题下总结所有英文描述，用逗"
+                                 "号间隔，再加上‘, (white background)’, 然后换行后以'promptEnd'结尾；再然后加一条服饰风格介绍，标题为"
+                                 "‘# 服饰风格介绍’，并在此标题下以介绍的口吻总结所有的中文描述。" + prompt,
+             }
+        ]
+
+    )
+    print(completion.choices[0].message.content)
+    return completion.choices[0].message.content