Upload 2 files

prompt_gen.py

@@ -1,39 +1,39 @@
-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):
-    prompt = nv_prompt
-    trigger = "a Hanfu"
-    if gender == "男":
-        prompt = na_prompt
-        trigger = "A Hanfu"
-    elif gender == "女":
-        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 + "根据上述建议，从以下prompt库中的触发词、种类、上衣、裙子、领子、袖子、袖口、腰饰、裙子详述中每个挑选一个词，分点描述，触发"
-                                 "词固定选择为" + trigger + "，然后在最下面列出所有Prompt，以‘Begin’为开头后换行, 输出所有英文描述，用逗"
-                                 "号间隔，再加上‘, white background’, 再然后换行后以'End'结尾。prompt库如下：" + prompt,
-             }
-        ]
-
-    )
-    print(completion.choices[0].message.content)
-    return completion.choices[0].message.content
+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):
+    prompt = nv_prompt
+    trigger = "a Hanfu"
+    if gender == "男":
+        prompt = na_prompt
+        trigger = "A Hanfu"
+    elif gender == "女":
+        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 + "根据上述建议，从以下prompt库中的触发词、种类、上衣、裙子、领子、袖子、袖口、腰饰、裙子详述中每个挑选一个词，分点描述，触发"
+                                 "词固定选择为" + trigger + "，然后在最下面列出所有Prompt，以‘Begin’为开头后换行, 输出所有英文描述，用逗"
+                                 "号间隔，再加上‘, white background’, 再然后换行后以'End'结尾。prompt库如下：" + prompt,
+             }
+        ]
+
+    )
+    print(completion.choices[0].message.content)
+    return completion.choices[0].message.content

sugg_gene.py

@@ -1,34 +1,34 @@
-from suggestion import generate_outfit_advice
-from clothGen import pro_gen
-import os
-from openai import OpenAI
-
-os.environ["OPENAI_API_KEY"] = "sk-vtyR3fdgk08jmJ5e3eF6F5Ef663c4a3bAd0166C3549a1a8e"
-os.environ["OPENAI_BASE_URL"] = "http://15.204.101.64:4000/v1"
-
-
-def suggest_gene(user_name, height, weight, waist, chest, hip, shoulder_width, leg_length, arm_length, gender,
-                 body_type, skin_color, style_preference, lifestyle_requirements, special_requirements,
-                 feedback, user_pic):
-    analyse = generate_outfit_advice(user_name, height, weight, waist, chest, hip, shoulder_width, leg_length,
-                                     arm_length, gender, body_type, skin_color, style_preference,
-                                     lifestyle_requirements, special_requirements, feedback, user_pic)
-    prompts = ""
-    for i in range(1, 4):
-        prompts += pro_gen(analyse, gender, i)
-
-    client = OpenAI()
-    completion = client.chat.completions.create(
-        model="gpt-4o",
-        messages=[
-            {"role": "system",
-             "content": "You are a helpful assistant.", },
-            {"role": "user",
-             "content": "你是一位专业的民族服饰搭配大师，你需要充分了解中华民族的所有民族服饰的相关知识，包括不同民族服饰适合什么样的人群等。"
-                        "以下是用户分析与三套服饰描述，请据此给出穿搭建议，要求以三段提示词为主要建议参考" + analyse + prompts,
-             }
-        ]
-
-    )
-    print(completion.choices[0].message.content)
-    return completion.choices[0].message.content
+from suggestion import generate_outfit_advice
+from clothGen import pro_gen
+import os
+from openai import OpenAI
+
+# os.environ["OPENAI_API_KEY"] = "sk-vtyR3fdgk08jmJ5e3eF6F5Ef663c4a3bAd0166C3549a1a8e"
+# os.environ["OPENAI_BASE_URL"] = "http://15.204.101.64:4000/v1"
+
+
+def suggest_gene(user_name, height, weight, waist, chest, hip, shoulder_width, leg_length, arm_length, gender,
+                 body_type, skin_color, style_preference, lifestyle_requirements, special_requirements,
+                 feedback, user_pic):
+    analyse = generate_outfit_advice(user_name, height, weight, waist, chest, hip, shoulder_width, leg_length,
+                                     arm_length, gender, body_type, skin_color, style_preference,
+                                     lifestyle_requirements, special_requirements, feedback, user_pic)
+    prompts = ""
+    for i in range(1, 4):
+        prompts += pro_gen(analyse, gender, i)
+
+    client = OpenAI()
+    completion = client.chat.completions.create(
+        model="gpt-4o",
+        messages=[
+            {"role": "system",
+             "content": "You are a helpful assistant.", },
+            {"role": "user",
+             "content": "你是一位专业的民族服饰搭配大师，你需要充分了解中华民族的所有民族服饰的相关知识，包括不同民族服饰适合什么样的人群等。"
+                        "以下是用户分析与三套服饰描述，请据此给出穿搭建议，要求以三段提示词为主要建议参考" + analyse + prompts,
+             }
+        ]
+
+    )
+    print(completion.choices[0].message.content)
+    return completion.choices[0].message.content