refactoring

src/forms.py

@@ -16,6 +16,125 @@ from src.prompts import (
 from src.questions import get_questions
 from src.utils import get_embedding, load_env_vars
 
+# 変更タイプとサブタイプの定義
+CHANGE_TYPES = [
+    "",
+    "原薬",
+    "添加物",
+    "資材変更",
+    "デザイン変更",
+    "製造方法・製造設備",
+    "製造支援システム",
+    "構造設備・設備移設",
+    "品質管理",
+    "出荷・物流",
+]
+
+SUBTYPES = {
+    "原薬": [
+        "",
+        "グレード",
+        "製造設備",
+        "製造方法",
+        "製造所（A社→B社）",
+        "製造場所（製造メーカーに変更はない。A社の①工場→A社の②工場）",
+    ],
+    "添加物": [
+        "",
+        "グレード（型番）",
+        "製造設備",
+        "製造メーカー（A社→B社）",
+        "製造場所（製造メーカーに変更はない。A社の①工場→A社の②工場）",
+    ],
+    "資材変更": [
+        "",
+        "チューブ・容器キャップ（直接容器）",
+        "容器ラベル",
+        "添付文書",
+        "個箱",
+        "セット箱",
+        "シール（例：見本用シール、変更品シール）",
+        "パッキングケース",
+        "パッキングケースラベル",
+    ],
+    "製造方法・製造設備": [
+        "",
+        "仕込量変更（スケールアップ・ダウン）",
+        "標準的仕込量変更（100%仕込み→10●%仕込みに変更）",
+        "出荷規格の変更",
+        "調合設備の変更",
+        "充填設備の変更",
+        "包装設備の変更",
+        "調合工程のパラメーター変更",
+        "充填工程のパラメーター変更（例：充填速度、設定温度、規格範囲、トルク値）",
+        "包装工程のパラメーター変更",
+        "重量基準値の変更（ウエイトチェッカーでの計量）",
+        "包装工程 作業人数の変更",
+        "洗浄方法の変更",
+    ],
+    "製造支援システム": ["", "製造支援システム（製造用水、空調システム）"],
+    "構造設備・設備移設": ["", "構造設備", "設備の移設（製造設備・試験設備）"],
+    "品質管理": [
+        "",
+        "試験方法の変更",
+        "試験室の変更",
+        "外部試験機関の変更",
+        "試験設備の変更・更新",
+        "バルク保管期間の変更",
+    ],
+    "出荷・物流": ["", "製品納品先の変更"],
+}
+
+SUBTYPES2 = {
+    ("資材変更", "チューブ・容器キャップ（直接容器）"): [
+        "",
+        "製造設備の変更",
+        "製造所の変更",
+        "材質の変更",
+        "形状変更",
+        "版番号の変更",
+    ],
+    ("資材変更", "容器ラベル"): [
+        "",
+        "版番号の変更",
+        "セルフメディケーション税制対応マークの追加",
+    ],
+    ("資材変更", "添付文書"): [
+        "",
+        "製造所の変更",
+        "材質変更",
+        "厚さ変更",
+        "版番号の変更",
+    ],
+    ("資材変更", "個箱"): [
+        "",
+        "製造所の変更",
+        "捺印位置の変更",
+        "個箱形状変更",
+        "個箱の材質（表面加工）",
+        "版番号の変更",
+        "セルフメディケーション税制対応マークの追加",
+    ],
+    ("資材変更", "セット箱"): [
+        "",
+        "製造所の変更",
+        "捺印位置の変更",
+        "セット箱形状変更",
+        "版番号の変更",
+    ],
+    ("資材変更", "シール（例：見本用シール、変更品シール）"): [
+        "",
+        "サイズの変更",
+        "版番号の変更",
+    ],
+    ("資材変更", "パッキングケース"): ["", "版番号の変更"],
+    ("資材変更", "パッキングケースラベル"): ["", "版番号の変更"],
+    (
+        "製造方法・製造設備",
+        "充填工程のパラメーター変更（例：充填速度、設定温度、規格範囲、トルク値）",
+    ): ["", "液", "クリーム", "チューブ"],
+}
+
 
 def change_request_form(embed_output):
     st.title("変更申請書 文書生成")
@@ -47,8 +166,8 @@ def change_request_form(embed_output):
         )
         implementation_date = st.text_area("実施予定時期")
 
-        submit_button0 = st.form_submit_button("進む")
-        if submit_button0:
+        moveon_button = st.form_submit_button("進む")
+        if moveon_button:
             st.session_state["form_submitted"] = True
 
     change_num = int(change_num)
@@ -75,219 +194,30 @@ def change_request_form(embed_output):
                     st.markdown(f"**{change+1}つ目の変更**")
                 types[change] = st.selectbox(
                     "変更の種類を選択してください:",
-                    [
-                        "",
-                        "原薬",
-                        "添加物",
-                        "資材変更",
-                        "デザイン変更",
-                        "製造方法・製造設備",
-                        "製造支援システム",
-                        "構造設備・設備移設",
-                        "品質管理",
-                        "出荷・物流",
-                    ],
+                    CHANGE_TYPES,
                     key=f"type_{change}",
                 )
                 if types[change]:
                     st.session_state[f"type_chosen_{change}"] = True
 
             if st.session_state[f"type_chosen_{change}"]:
-                if types[change] == "原薬":
-                    subtypes2[change] = "x"
-                    subtypes[change] = st.selectbox(
-                        "変更内容を選択してください:",
-                        [
-                            "",
-                            "グレード",
-                            "製造設備",
-                            "製造方法",
-                            "製造所（A社→B社）",
-                            "製造場所（製造メーカーに変更はない。A社の①工場→A社の②工場）",
-                        ],
-                        key=f"subtype_1_{change}",
-                    )
-                elif types[change] == "添加物":
-                    subtypes2[change] = "x"
-                    subtypes[change] = st.selectbox(
-                        "変更内容を選択してください:",
-                        [
-                            "",
-                            "グレード（型番）",
-                            "製造設備",
-                            "製造メーカー（A社→B社）",
-                            "製造場所（製造メーカーに変更はない。A社の①工場→A社の②工場）",
-                        ],
-                        key=f"subtype_2_{change}",
-                    )
-                elif types[change] == "資材変更":
-                    subtypes[change] = st.selectbox(
-                        "変更内容を選択してください:",
-                        [
-                            "",
-                            "チューブ・容器キャップ（直接容器）",
-                            "容器ラベル",
-                            "添付文書",
-                            "個箱",
-                            "セット箱",
-                            "シール（例：見本用シール、変更品シール）",
-                            "パッキングケース",
-                            "パッキングケースラベル",
-                        ],
-                        key=f"subtype_3_{change}",
-                    )
-                elif types[change] == "デザイン変更":
-                    subtypes2[change] = "x"
-                    subtypes[change] = "x"
-                elif types[change] == "製造方法・製造設備":
-                    subtypes2[change] = "x"
-                    subtypes[change] = st.selectbox(
-                        "変更内容を選択してください:",
-                        [
-                            "",
-                            "仕込量変更（スケールアップ・ダウン）",
-                            "標準的仕込量変更（100%仕込み→10●%仕込みに変更）",
-                            "出荷規格の変更",
-                            "調合設備の変更",
-                            "充填設備の変更",
-                            "包装設備の変更",
-                            "調合工程のパラメーター変更",
-                            "充填工程のパラメーター変更（例：充填速度、設定温度、規格範囲、トルク値）",
-                            "包装工程のパラメーター変更",
-                            "重量基準値の変更（ウエイトチェッカーでの計量）",
-                            "包装工程 作業人数の変更",
-                            "洗浄方法の変更",
-                        ],
-                        key=f"subtype_4_{change}",
-                    )
-                elif types[change] == "製造支援システム":
-                    subtypes2[change] = "x"
-                    subtypes[change] = st.selectbox(
-                        "変更内容を選択してください:",
-                        ["", "製造支援システム（製造用水、空調システム）"],
-                        key=f"subtype_5_{change}",
-                    )
-                elif types[change] == "構造設備・設備移設":
-                    subtypes2[change] = "x"
-                    subtypes[change] = st.selectbox(
-                        "変更内容を選択してください:",
-                        ["", "構造設備", "設備の移設（製造設備・試験設備）"],
-                        key=f"subtype_6_{change}",
-                    )
-                elif types[change] == "品質管理":
-                    subtypes2[change] = "x"
-                    subtypes[change] = st.selectbox(
-                        "変更内容を選択してください:",
-                        [
-                            "",
-                            "試験方法の変更",
-                            "試験室の変更",
-                            "外部試験機関の変更",
-                            "試験設備の変更・更新",
-                            "バルク保管期間の変更",
-                        ],
-                        key=f"subtype_7_{change}",
-                    )
-                elif types[change] == "出荷・物流":
-                    subtypes2[change] = "x"
-                    subtypes[change] = st.selectbox(
-                        "変更内容を選択してください:",
-                        ["", "製品納品先の変更"],
-                        key=f"subtype_8_{change}",
-                    )
-
-                if (
-                    subtypes[change]
-                    == "充填工程のパラメーター変更（例：充填速度、設定温度、規格範囲、トルク値）"
+                subtypes[change] = st.selectbox(
+                    "変更内容を選択してください:",
+                    SUBTYPES.get(types[change], [""]),
+                    key=f"subtype_{change}",
+                )
+                if subtypes[change] in SUBTYPES2.get(
+                    (types[change], subtypes[change]), []
                 ):
                     subtypes2[change] = st.selectbox(
                         "詳しい内容を選択してください:",
-                        ["", "液", "クリーム", "チューブ"],
-                        key=f"subtype_9_{change}",
+                        SUBTYPES2[(types[change], subtypes[change])],
+                        key=f"subtype2_{change}",
                     )
 
-                match types[change], subtypes[change]:
-                    case "資材変更", "チューブ・容器キャップ（直接容器）":
-                        subtypes2[change] = st.selectbox(
-                            "詳しい内容を選択してください:",
-                            [
-                                "",
-                                "製造設備の変更",
-                                "製造所の変更",
-                                "材質の変更",
-                                "形状変更",
-                                "版番号の変更",
-                            ],
-                            key=f"subtype_3_1_{change}",
-                        )
-                    case "資材変更", "容器ラベル":
-                        subtypes2[change] = st.selectbox(
-                            "詳しい内容を選択してください:",
-                            [
-                                "",
-                                "版番号の変更",
-                                "セルフメディケーション税制対応マークの追加",
-                            ],
-                            key=f"subtype_3_2_{change}",
-                        )
-                    case "資材変更", "添付文書":
-                        subtypes2[change] = st.selectbox(
-                            "詳しい内容を選択してください:",
-                            [
-                                "",
-                                "製造所の変更",
-                                "材質変更",
-                                "厚さ変更",
-                                "版番号の変更",
-                            ],
-                            key=f"subtype_3_3_{change}",
-                        )
-                    case "資材変更", "個箱":
-                        subtypes2[change] = st.selectbox(
-                            "詳しい内容を選択してください:",
-                            [
-                                "",
-                                "製造所の変更",
-                                "捺印位置の変更",
-                                "個箱形状変更",
-                                "個箱の材質（表面加工）",
-                                "版番号の変更",
-                                "セルフメディケーション税制対応マークの追加",
-                            ],
-                            key=f"subtype_3_4_{change}",
-                        )
-                    case "資材変更", "セット箱":
-                        subtypes2[change] = st.selectbox(
-                            "詳しい内容を選択してください:",
-                            [
-                                "",
-                                "製造所の変更",
-                                "捺印位置の変更",
-                                "セット箱形状変更",
-                                "版番号の変更",
-                            ],
-                            key=f"subtype_3_5_{change}",
-                        )
-                    case "資材変更", "シール（例：見本用シール、変更品シール）":
-                        subtypes2[change] = st.selectbox(
-                            "詳しい内容を選択してください:",
-                            ["", "サイズの変更", "版番号の変更"],
-                            key=f"subtype_3_6_{change}",
-                        )
-                    case "資材変更", "パッキングケース":
-                        subtypes2[change] = st.selectbox(
-                            "詳しい内容を選択してください:",
-                            ["", "版番号の変更"],
-                            key=f"subtype_3_7_{change}",
-                        )
-                    case "資材変更", "パッキングケースラベル":
-                        subtypes2[change] = st.selectbox(
-                            "詳しい内容を選択してください:",
-                            ["", "版番号の変更"],
-                            key=f"subtype_3_8_{change}",
-                        )
-
-                if subtypes[change] and subtypes2[change]:
+                if subtypes[change] and (
+                    subtypes2[change] or subtypes2[change] is None
+                ):
                     st.session_state[f"subtype_chosen_{change}"] = True
 
     if st.session_state[f"subtype_chosen_{change_num-1}"]:
@@ -338,17 +268,18 @@ def change_request_form(embed_output):
                 if nos[k]:
                     input_values[k] = "特になし"
 
-            system_prompt = systemprompt_template_1
-            system_prompt += systemprompt_template_2
+            system_prompt = (
+                system_template_review
+                + systemprompt_template_1
+                + systemprompt_template_2
+            )
             for k in range(question_num):
-                system_prompt += q[k]
-                system_prompt += "\n"
+                system_prompt += f"  {q[k]} : {input_values[k]}\n"
             system_prompt += systemprompt_template_3
 
             user_prompt = ""
             for k in range(question_num):
-                user_prompt += f"  A{q[k][1:4]} :  {input_values[k]}"
-                user_prompt += "\n"
+                user_prompt += f"  A{q[k][1:4]} :  {input_values[k]}\n"
 
             client = OpenAI(api_key=open_api_key)
             response = client.chat.completions.create(
@@ -357,7 +288,7 @@ def change_request_form(embed_output):
                     {"role": "system", "content": system_prompt},
                     {"role": "user", "content": user_prompt},
                 ],
-                response_format={"type": "json_object"},
+                response_format="json_object",
                 temperature=0.0,
             )
             response_text = response.choices[0].message.content

src/generate_guery.py

@@ -42,7 +42,7 @@ This is very important to my career.
 """
 
 completion = client.chat.completions.create(
-    model="gpt-4-0125-preview",
+    model="gpt-4o",
     messages=[
         {
             "role": "user",

src/questions.py

@@ -2,6 +2,8 @@ import json
 
 
 def get_questions(type, subtype, subtype2):
+    if subtype2 == None:
+        subtype2 = "x"
     with open(
         f"data/questions/{type}_{subtype}_{subtype2}.json", "r", encoding="utf-8"
     ) as file: