Merge pull request #30 from EQUES-Inc/feature/refactoring

.flake8

@@ -0,0 +1,2 @@
+[flake8]
+max-line-length = 88

README.md

@@ -24,4 +24,10 @@ OPENAI_API_KEY =
 ## 3. 仮想環境に入る
 `pipenv shell`
 ## 4. streamlitを立ち上げる
-`streamlit run app.py`
+`streamlit run main.py`
+
+# TODO
+- 企業ごとの変更点は`form.py`,`embed_output.py`,`utils.py`, `prompts.py`に集約されている想定
+  - `utils.py`に必要であれば追加の機能を足すイメージ
+- 上ファイルと`data`部分をフォークのリポジトリ部分で変更する
+- その他のファイルはメインのリポジトリにて変更

app.py

@@ -1,799 +0,0 @@
-import streamlit as st
-from datetime import date, datetime
-from openai import OpenAI
-import os
-from dotenv import load_dotenv
-import json
-from src import embed_output
-from src import embed_output_deviation
-
-from pinecone import Pinecone
-import os
-from dotenv import load_dotenv
-
-
-
-# 変更の種類に沿ってjsonファイルから確認事項の質問を読み取る関数
-def question(type, subtype, subtype2):
-    with open(f'data/questions/{type}_{subtype}_{subtype2}.json', 'r', encoding="utf-8") as file:
-        data = json.load(file)
-
-    questions = []
-    for i in data["items"]:
-        questions.append(i["question"])
-
-    return questions
-
-def get_embedding(text, model="text-embedding-3-large", client=None):
-    text = text.replace("\n", " ")
-    return client.embeddings.create(input=[text], model=model).data[0].embedding
-
-def change_request():
-    # タイトル 
-    st.title("変更申請書 文書生成")
-
-    # 環境変数からAPIキーを取得
-    load_dotenv()
-    open_api_key = os.environ.get("OPENAI_API_KEY")
-
-    load_dotenv()
-    pc = Pinecone(os.environ["PINECONE_API_KEY"])
-    index = pc.Index("bankyo")
-
-    client = OpenAI(api_key=open_api_key)
-
-    # ①申請者などの入力フォーム
-    with st.form("my_form"):
-        col1, col2 = st.columns(2)
-        with col1:
-            plant_type = st.selectbox("工場", ["", "第一・第二工場","第三工場", "第四工場"])
-        with col2:
-            applicant_name = st.text_input("申請者")
-        
-        col2_1, col2_2 = st.columns(2)
-        with col2_1:
-            change_num = st.selectbox("変更の件数", ["1","2","3","4","5","6","7","8","9","10"])
-        with col2_2:
-            application_date = st.date_input("申請日", date.today())
-
-        change_details = st.text_area("変更対象 (変更対象となる製品名・原料名・設備名等を記入する)")
-
-        implementation_date = st.text_area("実施予定時期")
-
-        submit_button0 = st.form_submit_button("進む")
-        if submit_button0:
-            st.session_state['form_submitted'] = True
-    
-    change_num = int(change_num)
-    
-    # セッション状態の初期化
-    for state in ['form_submitted', 'document_generated']:
-        if state not in st.session_state:
-            st.session_state[state] = False
-    for change in range(change_num):
-        if f"type_chosen_{change}" not in st.session_state:
-            st.session_state[f"type_chosen_{change}"] = False
-        if f"subtype_chosen_{change}" not in st.session_state:
-            st.session_state[f"subtype_chosen_{change}"] = False
-
-    types = [None] * change_num
-    subtypes = [None] * change_num
-    subtypes2 = [None] * change_num
-
-    if st.session_state['form_submitted']:
-        st.session_state[f"subtype_chosen_-1"] = True
-
-        for change in range(change_num):
-            # ②変更の種類の選択
-            if st.session_state[f'subtype_chosen_{change-1}']:
-                if change_num != 1:
-                    st.markdown(f"**{change+1}つ目の変更**")
-                # 変更の種類の選択
-                types[change] = st.selectbox("変更の種類を選択してください:", ["", 
-                                                        "原薬",
-                                                        "添加物",
-                                                        "資材変更",
-                                                        "デザイン変更",
-                                                        "製造方法・製造設備",
-                                                        "製造支援システム",
-                                                        "構造設備・設備移設",
-                                                        "品質管理",
-                                                        "出荷・物流"], 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] == "充填工程のパラメーター変更（例：充填速度、設定温度、規格範囲、トルク値）":
-                    subtypes2[change] = st.selectbox("詳しい内容を選択してください:", ["", "液","クリーム","チューブ"], key = f"subtype_9_{change}")
-
-                # 資材変更の3段階目の分岐
-                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]:
-                    st.session_state[f'subtype_chosen_{change}'] = True
-
-    # ④質問の回答からLLMの応答を生成
-    if st.session_state[f'subtype_chosen_{change_num-1}']:
-        # システムプロンプトのテンプレ（質問部分以外）
-        systemprompt_template_1 = f"""
-        ***
-        変更内容：
-        変更理由：
-        品質への影響：
-        ***
-        あなたは、質問に対して入力された情報から、詳しい内容を含んだ申請書の文章を生成するアシスタントです。
-        上記の形式に従って、入力された情報を当てはめて出力してください。漏れがないようにしてください。変更がないことに関しては、まとめず主語を明確にして「品質への影響」で言及してください。
-        「だ・である」調で出力してください。
-        あなたは返答をすべてJSON形式で出力します。
-        ***
-        """
-
-        systemprompt_template_2 = """
-        今回の変更の種類は以下です。
-        """
-        for i in range(change_num):
-            if subtypes2[i] == "x":
-                systemprompt_template_2 += f"{i+1}. {types[i]},{subtypes[i]}"
-            else:
-                systemprompt_template_2 += f"{i+1}. {types[i]},{subtypes[i]},{subtypes2[i]}"
-            systemprompt_template_2 += "\n"
-        systemprompt_template_2 += """
-        ***
-        あなたが質問した内容は以下です。\n
-        ***
-        """
-
-        systemprompt_template_3 = """
-        ***
-        出力は以下のjsonテンプレートに従って出力してください。\n
-        {
-            "変更内容": "",
-            "変更理由": "",
-            "品質への影響": ""
-        }
-        ***
-        変更内容は以下のようなフォーマットで揃えてください。
-        変更の種類の部分は「〇〇の変更」という自然な文章になるように適宜書き換えてください。
-        {変更の種類}の変更
-        変更前： \n
-        変更後： 
-        ***
-        変更内容が複数ある場合は以下のような例にしたがってで揃えてください。
-        番号のすぐ後には変更の種類を添えてください。
-        充填機をAからBに、個箱の材質をXからYに変更する場合
-
-        1. 充填設備の変更
-        変更前： A
-        変更後： B
-
-        2. 個箱の材質の変更
-        変更前： X
-        変更後： Y
-        ***
-        変更理由と品質への影響も、変更の種類によって
-
-        1. 充填設備の変更
-        
-        2. 包装設備の変更
-
-        とナンバリングしてください。変更が一つだけの場合はナンバリングする必要はありません
-        ***
-        品質への影響は、3問目以降の回答ををまとめる形で出力してください。決して漏れがないようにしてください。
-        ***
-        「です・ます」調ではなく「だ・である」調で出力してください。\n
-        """
-
-        # 変更の種類・内容に基づいて確認事項の質問を作成
-        q = []
-
-        for change in range(change_num):
-            q.append(f"Q{change+1}-1 : 今回の変更内容を教えて下さい。変更前と変更後の両方を明記してください。")
-            q.append(f"Q{change+1}-2 : 今回の変更を行う理由を教えて下さい。")
-            
-            qs = question(types[change], subtypes[change], subtypes2[change])
-            qs_num = len(qs)
-            for i in range(qs_num):
-                q.append(f"Q{change+1}-{i+3} : {qs[i]}")
-
-
-            q.append(f"Q{change+1}-{qs_num+3} : その他、変更による品質への影響を考慮すべき点はありますか？考慮すべき点があれば、その理由とともに説明してください。")
-        
-        # unique = {}
-        # for question, item in q:
-        #     if item not in unique:
-        #         unique[item] = question
-        # q = list(unique.values())
-
-        question_num = len(q)
-
-        # # 質問をナンバリング
-        # qq = [None]*question_num
-        # for change in range(question_num):
-        #     qq[change] = f"Q{change+1} : {q[change]}"
-
-
-
-        # 質問に回答させる
-        with st.form(key='my_form2_1'):
-            input_values = [None] * question_num
-            nos = [None] * question_num
-            qs_now = 1
-            st.markdown(f"**1つ目の変更について**")
-            for k in range(question_num):
-                if int(q[k][1]) != qs_now:
-                    st.markdown(f"**{qs_now+1}つ目の変更について**")
-                    qs_now += 1
-                input_values[k] = st.text_area(q[k], key=f'input_1_{k}')
-                nos[k] = st.checkbox(f"特になし", key = f"check_{k}")
-                st.write("\n")
-            st.markdown("***")
-            supplementary_information = st.text_area(f"Q{change_num+1} : 備考があれば記入してください。", key=f'supplementary_information_1')
-            attached_files = st.text_area(f"Q{change_num+2} : 添付資料名を記入してください。", key=f'attached_files_1')
-            submitted = st.form_submit_button('送信')
-
-        if submitted:
-            # 特になしの場合
-            for k in range(question_num):
-                if nos[k]:
-                    input_values[k] = "特になし"
-
-            # システムプロンプトを作成する
-            system_prompt = systemprompt_template_1
-            system_prompt += systemprompt_template_2
-            for k in range(question_num):
-                system_prompt += q[k]
-                system_prompt += "\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"
-            
-            # OpenAIのAPIを使用して応答を生成
-            client = OpenAI(api_key=open_api_key)
-            response = client.chat.completions.create(
-                model="gpt-4-0125-preview",
-                messages=[
-                    {"role": "system", "content": system_prompt},
-                    {"role": "user", "content": user_prompt}
-                ],
-                response_format={ "type": "json_object" },
-                temperature=0.0
-            )
-            response_text = response.choices[0].message.content
-
-
-            # 応答をjsonに変換
-
-            response_data = json.loads(response_text)
-
-            # 申請日をdate形式から文字列(〇〇〇〇年〇月〇日)に変換
-            application_date = application_date.isoformat()
-            date_obj = datetime.strptime(application_date, "%Y-%m-%d")
-            application_date = date_obj.strftime("%Y年%m月%d日")
-
-            # jsonデータを作成
-            json_data = {
-                "工場": plant_type,
-                "申請者": applicant_name,
-                "申請日": application_date,
-                "変更対象": change_details,
-                "実施予定時期": implementation_date,
-                "変更内容": str(response_data["変更内容"]),
-                "変更理由": str(response_data["変更理由"]),
-                "品質への影響": str(response_data["品質への影響"]),
-                "添付資料": attached_files,
-                "備考": supplementary_information,
-                "ユーザープロンプト": user_prompt
-                }
-            
-            # jsonファイルを作成
-            filename = 'data/change_request.json'
-            with open(filename, "w") as file:
-                json.dump(json_data, file, ensure_ascii=False, indent=4)
-
-            # jsonからdocxファイルを作成
-            embed_output.main()
-            st.session_state['document_generated'] = True
-         
-    # ⑤応答の結果を表示・ダウンロード・レビュー
-    if st.session_state['document_generated']:
-        st.markdown("以下の内容を記入した変更申請書(.docx)を作成しました。")
-
-        # Wordファイルのダウンロードボタン
-        with open("data/output.docx", "rb") as file:
-            st.download_button(
-                label="ダウンロード",
-                data=file,
-                file_name="output.docx",
-                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-            )
-
-        # 各項目の内容を表示
-        with open('data/change_request.json', 'r') as file:
-            json_data = json.load(file)
-
-        st.markdown(f'**申請者** : {json_data["申請者"]}')
-        st.markdown(f'**申請日** : {json_data["申請日"]}')
-        st.markdown(f'**変更対象** : {json_data["変更対象"]}')
-        st.markdown(f'**実施予定時期** : {json_data["実施予定時期"]}')
-        st.markdown(f'**変更内容** : {json_data["変更内容"]}')
-        st.markdown(f'**変更理由** : {json_data["変更理由"]}')
-        st.markdown(f'**品質への影響** : {json_data["品質への影響"]}')
-        st.markdown(f'**添付資料** : {json_data["添付資料"]}')
-        st.markdown(f'**備考** : {json_data["備考"]}')
-
-        # # レビュー用のボタン
-        review_button = st.button("レビューする")
-        if review_button:
-            query_content = json_data["変更内容"]
-            query_reason = json_data["変更理由"]
-            query_vec = get_embedding(query_content, client=client)
-
-            # 名寄せ
-            genres = [None] * change_num
-            for i in range(change_num):
-                match subtypes[i]:
-                    case "充填設備の変更":
-                        genres[i] = "充填機変更"
-                    case "製造メーカー（A社→B社）":
-                        genres[i] = "原料メーカー変更"
-                    case "仕込量変更（スケールアップ・ダウン）":
-                        genres[i] = "製造スケール変更"
-                print(genres[i])
-
-            
-            res = index.query(
-                vector=query_vec,
-                filter={
-                    "genre": {"$in": genres}
-                },
-                top_k=3,
-                include_metadata=True,
-            )
-            reasons = [r["metadata"]["change_reason"] for r in res["matches"]]
-            print(f"reasons: {reasons}")
-
-            # GPTにレビューを依頼
-            system = f"""
-            あなたは製薬会社の品質管理部門のリーダーです。今回、あなたの部下が作成した変更申請書の内容をレビューすることになりました。
-            ## レビュー例 に示される変更理由の手本に基づいて，以下の'''変更理由'''の修正すべき点を指摘してください．
-            ## レビュー例 は正しい変更理由を示しています．その内容を参考にして，以下の'''変更理由'''の修正すべき点を指摘してください．  
-            ただし改善例を示す必要はありません．
-
-            それではステップバイステップで生成していきましょう。
-            This is very important to my career.
-            
-            ## 出力例:
-            - 例1:
-            ユーザー:
-                前提知識 : 
-                    '充填工程で、現行のノルデンチューブ充填機にJM2チューブ充填機を追加する。 
-                     変更前：ノルデンチューブ充填機（J-CH5） 
-                     ノルデンチューブ充填機（J-CH8） 
-                     変更後：ノルデンチューブ充填機（J-CH5） 
-                     ノルデンチューブ充填機（J-CH8） 
-                     JM2チューブ充填機（J-CH4） 
-                    '
-                変更理由 :
-                    'ノルデンチューブ充填機（J-CH5）が故障した為、生産スケジュールの都合上、生産できる充填機を追加する。 '
-            返答:
-                '
-                - 故障した充填機について、機番を明記するとともに、『故障』ではなく『不具合』と表現を変更してください。また、部品の取り寄せを依頼している状況や、海外製であるため修理の目処が立たないことも追記が必要です。
-                - 代替として使用するJM2チューブ充填機についても、機番を明記し、ジルダザック軟膏3%10gと同サイズのチューブホルダーがあることを説明してください。
-                - さらに、JM2チューブ充填機が国内メーカーであるガデリウスの製品であり、同社が販売と修理を行っていることも追記してください。
-                
-        
-            - 例2:
-            ユーザー:
-                前提知識 : 
-                    '
-                    【原料：カルボキシビニルポリマーの変更】 
-                    変更前：メーカー；住友精化株式会社（岡山大鵬薬品支給） 
-                    商品名；HV-505HC 
-                    変更後：メーカー；富士フイルム和光純薬株式会社（万協調達） 
-                    商品名；ハイビスワコー103 
-                    '
-                変更理由 :
-                    '現行品のカルボキシビニルポリマー（商品名；HV-505HC）が終売となるため。 '
-            返答:
-                '
-                - 必要十分な情報が記載されています。修正の必要はありません。
-                '
-            
-            - 例3:
-            ユーザー:
-                前提知識 : 
-                    '
-                    製造設備及び仕込み量の変更 
-                    変更前；1t乳化機1t仕込み　もしくは1t乳化機750kg仕込み 
-                    変更後；500kg乳化機400kg仕込み
-                    '
-                変更理由 :
-                    '日邦薬品工業の販売数量が減ったため、製造量を減らして対応するため。 '
-            返答:
-                '
-                - 会社名は正式な社名を使用するようにしてください。この場合、『日邦薬品工業株式会社』が適切です。
-                - また、文章全体を簡潔にまとめるために、『製造量を減らして対応するため』という部分は削除しても問題ないでしょう。
-                - さらに、日邦薬品工業株式会社の役割を明確にするために、『発売元である』という情報を追加してください。
-                '
-            """
-            
-            user = f"""
-            ## レビュー例:
-                {reasons}
-
-            ## 前提知識:
-                {query_content}
-        
-            ## 変更理由：
-                {query_reason}
-            
-            返答:
-            """
-
-            completion = client.chat.completions.create(
-                model="gpt-4-0125-preview",
-                messages=[
-                    {
-                        "role": "system",
-                        "content": system,
-                    },
-                    {
-                        "role": "user",
-                        "content": user,
-                    },
-                ],
-                temperature=0.7,
-            )
-            review = completion.choices[0].message.content
-
-            st.markdown(f"**レビュー結果** : {review}")
-
-
-
-# def deviation_report():
-#     # タイトル 
-#     st.title("���脱発生報告書 文書生成")
-
-#     # 環境変数からAPIキーを取得
-#     load_dotenv()
-#     open_api_key = os.environ.get("OPENAI_API_KEY")
-
-#     # セッション状態の初期化
-#     for state in ['form_submitted', 'document_generated']:
-#         if state not in st.session_state:
-#             st.session_state[state] = False
-
-#     # ①報告者などの入力フォーム
-#     with st.form("my_form_deviatoin"):
-#         col1, col2 = st.columns(2)
-#         with col1:
-#             reporter_name = st.text_input("報告者")
-#         with col2:
-#             report_date = st.date_input("報告日", date.today())
-        
-#         col2_1, col2_2 = st.columns(2)
-#         with col2_1:
-#             product_name = st.text_input("品名")
-#         with col2_2:
-#             lot_number = st.text_input("ロットNo.")
-
-#         col1, col2, col3 = st.columns([1, 1, 2])
-#         with col1:
-#             occurrence_date = st.date_input("発生日", date.today())
-#         with col2:
-#             occurrence_time = st.time_input("発生時間", datetime.now().time())
-#         with col3:
-#             occurrence_place = st.text_input("発生場所")
-        
-#         col2_1, col2_2 = st.columns(2)
-#         with col2_1:
-#             discoverer_name = st.text_input("発見者")
-#         with col2_2:
-#             worker_name = st.text_input("作業者")
-
-#         submit_button0 = st.form_submit_button("進む")
-#         if submit_button0:
-#             st.session_state['form_submitted'] = True
-
-#     # ②質問の回答からLLMの応答を生成
-#     if st.session_state['form_submitted']:
-#         # 質問リストを作成
-#         q = ["Q1 : 逸脱内容を記入してください（どの手順・規格から逸脱したのか）",
-#              "Q2 : 逸脱の発見の経緯を記入してください",
-#              "Q3 : 応急処置とその処置の理由を記入してください",
-#              "Q4 : 当該ロットの試験実施等による品質評価について記入してください",
-#              "Q5 : 安定性モニタリングの対象ロットですか、もしくは対象ロットとしますか",
-#              "Q6 : 安定性モニタリングの検体採取表の更新について記入してください",
-#              "Q7 : 在庫品の原料・充填資材・包装資材に関して、在庫ロットの品質評価や在庫量・発注状況確認について記入してください",
-#              "Q8 : 原料メーカー・資材メーカー・設備メーカーへの調査依頼について記入してください",
-#              "Q9 : 次回製造ロットの生産スケジュール、受注状況の変更について記入してください",
-#              "Q10 : 出荷日の変更について記入してください",
-#              "Q11 : 製造販売業者への連絡について記入してください",
-#              "Q12 : 他製品・他ラインへの影響について記入してください",
-#              "Q13 : 過去製造ロットの品質の再評価について記入してください",
-#              "Q14 : その他の品質への影響があれば記入してください"]
-
-#         q_not = ["",
-#                  "",
-#                  "",
-#                  "品質評価は必要ない",
-#                  "対象ロットではない・しない",
-#                  "更新しない",
-#                  "評価・確認は必要ない",
-#                  "依頼は必要ない",
-#                  "変更しない",
-#                  "変更しない",
-#                  "連絡は必要ない",
-#                  "影響はない",
-#                  "再評価は必要ない",
-#                  "なし"]
-#         q_num = len(q)
-
-#         with st.form(key='my_form2_deviation'):
-#             input_values = [None] * q_num
-#             nos = [None] * q_num
-#             for k in range(q_num):
-#                 input_values[k] = st.text_area(q[k], key=f'input_deviatoin_{k}')
-#                 if (k > 2):
-#                     nos[k] = st.checkbox(f"{q_not[k]}", key = f"check_{k}")
-#                 st.markdown("***")
-
-
-#             attached_files = st.text_area(f"Q14 : 添付資料名を記入してください。", key=f'attached_files_deviation')
-#             no_attached_file = st.checkbox("添付資料なし", key = "check_attached_file")   
-#             supplementary_information = st.text_area(f"Q15 : 備考があれば記入してください。", key=f'supplementary_information_deviation')
-#             no_supplementary_information = st.checkbox("なし", key = "check_supplementary_information")   
-#             submitted = st.form_submit_button('送信')
-        
-#         if submitted:
-#             # 無しにチェックがつけられた場合
-#             for i in range(q_num-3):
-#                 if nos[i+2]:
-#                     input_values[i+3] = q_not[i+3]
-
-#             if no_attached_file:
-#                 attached_files = "x"
-#             if no_supplementary_information:
-#                 supplementary_information = "なし"
-
-#             # システムプロンプトを作成する
-#             systemprompt_template_1 = """
-#             ***
-#             逸脱内容・発見の経緯:
-#             応急処置・処置の理由:
-#             品質への影響の調査状況:
-#             ***
-#             あなたは、質問に対して入力された情報から、詳しい内容を含んだ逸脱発生報告書の文章を生成するアシスタントです。\n
-#             上記の形式に従って、入力された情報を当てはめて出力してください。報告書の読みやすさ、明瞭さ、正確さに注意してください。\n
-#             変更や必要がないという回答に関してはは特にそのことを記述する必要はありません。特別な記述があった場合にのみ記述してください。
-#             「だ・である」調で出力してください。\n
-#             あなたは返答をすべてJSON形式で出力します。\n
-#             あなたが質問した内容は以下です。\n
-#             ***
-#             """
-
-#             systemprompt_template_2 = """
-#             ***
-#             出力は以下のjsonテンプレートに従って出力してください。\n
-#             {
-#                 "逸脱内容・発見の経緯":"",
-#                 "応急処置・処置の理由":"",
-#                 "品質への影響の調査状況": "",
-#             }
-#             ***
-#             「です・ます」調ではなく「だ・である」調で出力してください。\n
-#             """
-
-#             system_prompt = systemprompt_template_1
-#             for k in range(q_num):
-#                 system_prompt += q[k]
-#                 system_prompt += "\n"
-#             system_prompt += systemprompt_template_2
-
-#             # ユーザープロンプトを作成する
-#             user_prompt = ""
-#             for k in range(q_num):
-#                 user_prompt +=  f"  A{k+1} :  {input_values[k]}"
-#                 user_prompt += "\n"
-
-#             # OpenAIのAPIを使用して応答を生成
-#             client = OpenAI(api_key=open_api_key)
-#             response = client.chat.completions.create(
-#                 model="gpt-4-0125-preview",
-#                 messages=[
-#                     {"role": "system", "content": system_prompt},
-#                     {"role": "user", "content": user_prompt}
-#                 ],
-#                 response_format={ "type": "json_object" },
-#                 temperature=0.0
-#             )
-#             response_text = response.choices[0].message.content
-
-#             # 応答をjsonに変換
-#             response_data = json.loads(response_text)
-
-
-#             # # 申請日をdate形式から文字列(〇〇〇〇年〇月〇日)に変換
-#             # application_date = application_date.isoformat()
-#             # date_obj = datetime.strptime(application_date, "%Y-%m-%d")
-#             # application_date = date_obj.strftime("%Y年%m月%d日")
-#             report_date = report_date.strftime("%Y年%m月%d日")
-#             occurrence_datetime = datetime.combine(occurrence_date, occurrence_time).strftime("%Y年%m月%d日%H時%M分")
-
-#             # jsonデータを作成
-#             json_data = {
-#                 "報告者": reporter_name,
-#                 "報告日": report_date,
-#                 "品名": product_name,
-#                 "ロットNo.": lot_number,
-#                 "発生日時": occurrence_datetime,
-#                 "発生場所": occurrence_place,
-#                 "発見者": discoverer_name,
-#                 "作業者": worker_name,
-#                 "逸脱内容・発見の経緯": str(response_data["逸脱内容・発見の経緯"]),
-#                 "応急処置・処置の理由": str(response_data["応急処置・処置の理由"]),
-#                 "品質への影響の調査状況": str(response_data["品質への影響の調査状況"]),
-#                 "添付資料": attached_files,
-#                 "備考": supplementary_information,
-#                 "ユーザープロンプト": user_prompt
-#                 }
-            
-#             # jsonファイルを作成
-#             filename = 'data/deviation_report.json'
-#             with open(filename, "w") as file:
-#                 json.dump(json_data, file, ensure_ascii=False, indent=4)
-
-#             # jsonからdocxファイルを作成
-#             embed_output_deviation.main()
-#             st.session_state['document_generated'] = True
-
-#     # ③応答の結果を表示・ダウンロード・レビュー
-#     if st.session_state['document_generated']:
-#         st.markdown("以下の内容を記入した変更申請書(.docx)を作成しました。")
-
-#         # Wordファイルのダウンロードボタン
-#         with open("data/output_deviation.docx", "rb") as file:
-#             st.download_button(
-#                 label="ダウンロード",
-#                 data=file,
-#                 file_name="output_deviation.docx",
-#                 mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-#             )
-
-#         # 各項目の内容を表示
-#         with open('data/deviation_report.json', 'r') as file:
-#             json_data = json.load(file)
-#         st.markdown(f'**報告者** : {json_data["報告者"]}')
-#         st.markdown(f'**報告日** : {json_data["報告日"]}')
-#         st.markdown(f'**品名** : {json_data["品名"]}')
-#         st.markdown(f'**ロットNo.** : {json_data["ロットNo."]}')
-#         st.markdown(f'**発生日時** : {json_data["発生日時"]}')
-#         st.markdown(f'**発生場所** : {json_data["発生場所"]}')
-#         st.markdown(f'**発見者** : {json_data["発見者"]}')
-#         st.markdown(f'**作業者** : {json_data["作業者"]}')
-#         st.markdown(f'**逸脱内容・発見の経緯** : {json_data["逸脱内容・発見の経緯"]}')
-#         st.markdown(f'**応急処置・処置の理由** : {json_data["応急処置・処置の理由"]}')
-#         st.markdown(f'**品質への影響の調査状況** : {json_data["品質への影響の調査状況"]}')
-#         st.markdown(f'**添付資料** : {json_data["添付資料"]}')
-#         st.markdown(f'**備考** : {json_data["備考"]}')
-
-
-
-pages = {
-    "変更申請書": change_request
-    # "逸脱発生報告書": deviation_report
-}
-selected_page = st.sidebar.selectbox("作成する文書を選択してください。", list(pages.keys()))
-pages[selected_page]()

data/change_request.json

@@ -1,11 +1,13 @@
 {
-    "申請者": "test",
-    "申請日": "2024-03-05",
-    "変更対象": "原薬",
-    "実施予定時期": "１",
-    "変更内容": "詳細は提供されておりません。",
-    "変更理由": "生産スケジュールの調整を容易とするため。",
-    "品質への影響": "詳細は提供されておりません。",
-    "備考": "詳しくは提供されていない項目が多いです。全ての項目について必要な情報が提供されていない場合、適切な評価が難しい可能性があることを留意してください。",
-    "ユーザープロンプト": "A 1 :  あ\n\nA 2 :  生産スケジュールの調整を容易とするため。\n\nA 3 :  あ\n\nA 4 :  あ\n\nA 5 :  あ\n\nA 6 :  あ\n\nA 7 :  あ\n\nA 8 :  あ\n\nA 9 :  あ\n\n"
+    "�H��": "���E���H��",
+    "�\����": "",
+    "�\����": "2024�N05��28��",
+    "�ύX�Ώ�": "",
+    "���{�\�莞��": "",
+    "�ύX���e": "��񂪕s�����Ă��邽�߁A��̓I�ȕύX���e�����ł��Ȃ��B",
+    "�ύX���R": "��񂪕s�����Ă��邽�߁A�ύX���R�����ł��Ȃ��B",
+    "�i���ւ̉e��": "��񂪕s�����Ă��邽�߁A�i���ւ̉e����]���ł��Ȃ��B",
+    "�Y�t����": "",
+    "���l": "",
+    "���[�U�[�v�����v�g": "  A1-1 :  ������\n  A1-2 :  \n  A1-3 :  \n  A1-4 :  \n  A1-5 :  \n  A1-6 :  \n  A1-7 :  \n  A1-8 :  \n  A1-9 :  \n  A1-1 :  \n  A1-1 :  \n"
 }

main.py

@@ -0,0 +1,14 @@
+import streamlit as st
+
+from src.embed_output import main as embed_output
+from src.forms import change_request_form
+
+pages = {
+    "変更申請書": change_request_form,
+    # "逸脱発生報告書": deviation_report_form
+}
+
+selected_page = st.sidebar.selectbox(
+    "作成する文書を選択してください。", list(pages.keys())
+)
+pages[selected_page](embed_output)

src/embed_output_deviation.py

@@ -1,6 +1,7 @@
 from docx import Document
 import json
 
+
 def replace_text_in_paragraphs(paragraphs, replacements):
     """
     パラグラフ内のテキストを置き換える
@@ -35,10 +36,9 @@ def main():
     # ファイルパスと置き換えるテキストの辞書
     # ここをJSONから読み込むように変更する
 
-
     template_path = "data/template_deviation.docx"
 
-    with open('data/deviation_report.json', 'r') as file:
+    with open("data/deviation_report.json", "r") as file:
         data = json.load(file)
 
     if data["添付資料"] == "x":
@@ -46,7 +46,7 @@ def main():
         no = "☑"
         attached_file = ""
     else:
-        yes = "☑" 
+        yes = "☑"
         no = "□"
         attached_file = data["添付資料"]
 
@@ -64,8 +64,8 @@ def main():
         "[impact_on_quality]": data["品質への影響の調査状況"],
         "[yes]": yes,
         "[no]": no,
-        "[attached_file]" : attached_file,
-        "[supplementary_information]": data["備考"]
+        "[attached_file]": attached_file,
+        "[supplementary_information]": data["備考"],
     }
 
     doc = replace_text_in_docx(template_path, replacements)
@@ -75,4 +75,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

src/forms.py

@@ -0,0 +1,385 @@
+import json
+import os
+from datetime import date, datetime
+
+import streamlit as st
+from openai import OpenAI
+from pinecone import Pinecone
+
+from src.prompts import (
+    system_template_review,
+    systemprompt_template_1,
+    systemprompt_template_2,
+    systemprompt_template_3,
+    user_template_review,
+)
+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("変更申請書 文書生成")
+
+    load_env_vars()
+    open_api_key = os.environ.get("OPENAI_API_KEY")
+    pc = Pinecone(os.environ["PINECONE_API_KEY"])
+    index = pc.Index("bankyo")
+
+    client = OpenAI(api_key=open_api_key)
+
+    with st.form("my_form"):
+        col1, col2 = st.columns(2)
+        with col1:
+            plant_type = st.selectbox(
+                "工場", ["", "第一・第二工場", "第三工場", "第四工場"]
+            )
+        with col2:
+            applicant_name = st.text_input("申請者")
+
+        col2_1, col2_2 = st.columns(2)
+        with col2_1:
+            change_num = st.selectbox("変更の件数", [str(i) for i in range(1, 11)])
+        with col2_2:
+            application_date = st.date_input("申請日", date.today())
+
+        change_details = st.text_area(
+            "変更対象 (変更対象となる製品名・原料名・設備名等を記入する)"
+        )
+        implementation_date = st.text_area("実施予定時期")
+
+        moveon_button = st.form_submit_button("進む")
+        if moveon_button:
+            st.session_state["form_submitted"] = True
+
+    change_num = int(change_num)
+
+    for state in ["form_submitted", "document_generated"]:
+        if state not in st.session_state:
+            st.session_state[state] = False
+    for change in range(change_num):
+        if f"type_chosen_{change}" not in st.session_state:
+            st.session_state[f"type_chosen_{change}"] = False
+        if f"subtype_chosen_{change}" not in st.session_state:
+            st.session_state[f"subtype_chosen_{change}"] = False
+
+    types = [None] * change_num
+    subtypes = [None] * change_num
+    subtypes2 = [None] * change_num
+
+    if st.session_state["form_submitted"]:
+        st.session_state[f"subtype_chosen_-1"] = True
+
+        for change in range(change_num):
+            if st.session_state[f"subtype_chosen_{change-1}"]:
+                if change_num != 1:
+                    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}"]:
+                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(
+                        "詳しい内容を選択してください:",
+                        SUBTYPES2[(types[change], subtypes[change])],
+                        key=f"subtype2_{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}"]:
+        q = []
+
+        for change in range(change_num):
+            q.append(
+                f"Q{change+1}-1 : 今回の変更内容を教えて下さい。変更前と変更後の両方を明記してください。"
+            )
+            q.append(f"Q{change+1}-2 : 今回の変更を行う理由を教えて下さい。")
+
+            qs = get_questions(types[change], subtypes[change], subtypes2[change])
+            qs_num = len(qs)
+            for i in range(qs_num):
+                q.append(f"Q{change+1}-{i+3} : {qs[i]}")
+
+            q.append(
+                f"Q{change+1}-{qs_num+3} : その他、変更による品質への影響を考慮すべき点はありますか？考慮すべき点があれば、その理由とともに説明してください。"
+            )
+
+        question_num = len(q)
+
+        with st.form(key="my_form2_1"):
+            input_values = [None] * question_num
+            nos = [None] * question_num
+            qs_now = 1
+            st.markdown(f"**1つ目の変更について**")
+            for k in range(question_num):
+                if int(q[k][1]) != qs_now:
+                    st.markdown(f"**{qs_now+1}つ目の変更について**")
+                    qs_now += 1
+                input_values[k] = st.text_area(q[k], key=f"input_1_{k}")
+                nos[k] = st.checkbox(f"特になし", key=f"check_{k}")
+                st.write("\n")
+            st.markdown("***")
+            supplementary_information = st.text_area(
+                f"Q{change_num+1} : 備考があれば記入してください。",
+                key=f"supplementary_information_1",
+            )
+            attached_files = st.text_area(
+                f"Q{change_num+2} : 添付資料名を記入してください。",
+                key=f"attached_files_1",
+            )
+            submitted = st.form_submit_button("送信")
+
+        if submitted:
+            for k in range(question_num):
+                if nos[k]:
+                    input_values[k] = "特になし"
+
+            system_prompt = (
+                system_template_review
+                + systemprompt_template_1
+                + systemprompt_template_2
+            )
+            for k in range(question_num):
+                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]}\n"
+
+            client = OpenAI(api_key=open_api_key)
+            response = client.chat.completions.create(
+                model="gpt-4o",
+                messages=[
+                    {"role": "system", "content": system_prompt},
+                    {"role": "user", "content": user_prompt},
+                ],
+                response_format="json_object",
+                temperature=0.0,
+            )
+            response_text = response.choices[0].message.content
+            response_data = json.loads(response_text)
+
+            application_date = application_date.isoformat()
+            date_obj = datetime.strptime(application_date, "%Y-%m-%d")
+            application_date = date_obj.strftime("%Y年%m月%d日")
+
+            json_data = {
+                "工場": plant_type,
+                "申請者": applicant_name,
+                "申請日": application_date,
+                "変更対象": change_details,
+                "実施予定時期": implementation_date,
+                "変更内容": str(response_data["変更内容"]),
+                "変更理由": str(response_data["変更理由"]),
+                "品質への影響": str(response_data["品質への影響"]),
+                "添付資料": attached_files,
+                "備考": supplementary_information,
+                "ユーザープロンプト": user_prompt,
+            }
+
+            filename = "data/change_request.json"
+            with open(filename, "w") as file:
+                json.dump(json_data, file, ensure_ascii=False, indent=4)
+
+            embed_output()
+            st.session_state["document_generated"] = True
+
+    if st.session_state["document_generated"]:
+        st.markdown("以下の内容を記入した変更申請書(.docx)を作成しました。")
+
+        with open("data/output.docx", "rb") as file:
+            st.download_button(
+                label="ダウンロード",
+                data=file,
+                file_name="output.docx",
+                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+            )
+
+        with open("data/change_request.json", "r") as file:
+            json_data = json.load(file)
+
+        st.markdown(f'**申請者** : {json_data["申請者"]}')
+        st.markdown(f'**申請日** : {json_data["申請日"]}')
+        st.markdown(f'**変更対象** : {json_data["変更対象"]}')
+        st.markdown(f'**実施予定時期** : {json_data["実施予定時期"]}')
+        st.markdown(f'**変更内容** : {json_data["変更内容"]}')
+        st.markdown(f'**変更理由** : {json_data["変更理由"]}')
+        st.markdown(f'**品質への影響** : {json_data["品質への影響"]}')
+        st.markdown(f'**添付資料** : {json_data["添付資料"]}')
+        st.markdown(f'**備考** : {json_data["備考"]}')
+
+        review_button = st.button("レビューする")
+        if review_button:
+            query_content = json_data["変更内容"]
+            query_reason = json_data["変更理由"]
+            query_vec = get_embedding(query_content, client=client)
+
+            genres = [None] * change_num
+            for i in range(change_num):
+                match subtypes[i]:
+                    case "充填設備の変更":
+                        genres[i] = "充填機変更"
+                    case "製造メーカー（A社→B社）":
+                        genres[i] = "原料メーカー変更"
+                    case "仕込量変更（スケールアップ・ダウン）":
+                        genres[i] = "製造スケール変更"
+
+            res = index.query(
+                vector=query_vec,
+                filter={"genre": {"$in": genres}},
+                top_k=3,
+                include_metadata=True,
+            )
+            reasons = [r["metadata"]["change_reason"] for r in res["matches"]]
+
+            system = system_template_review
+
+            user = user_template_review.format(
+                reasons=reasons, query_content=query_content, query_reason=query_reason
+            )
+
+            completion = client.chat.completions.create(
+                model="gpt-4o",
+                messages=[
+                    {"role": "system", "content": system},
+                    {"role": "user", "content": user},
+                ],
+                temperature=0.7,
+            )
+            review = completion.choices[0].message.content
+            st.markdown(f"**レビュー結果** : {review}")

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/generate_query_all.py

@@ -1,11 +1,13 @@
-from openai import OpenAI
 import json
 import os
+
 from dotenv import load_dotenv
+from openai import OpenAI
 from openpyxl import load_workbook
 
+
 def make_json(type, subtype, subtype2, items):
-    name = f'{type}_{subtype}_{subtype2}'
+    name = f"{type}_{subtype}_{subtype2}"
 
     with open("data/template.json", "r") as f:
         template = f.read()
@@ -13,7 +15,7 @@ def make_json(type, subtype, subtype2, items):
     template = json.loads(template)
 
     template["name"] = name
-    #print(template)
+    # print(template)
 
     prompt = f"""
     あなたは製薬会社の優秀なQAです。
@@ -33,7 +35,7 @@ def make_json(type, subtype, subtype2, items):
     """
 
     completion = client.chat.completions.create(
-        model="gpt-4-0125-preview",
+        model="gpt-4o",
         messages=[
             {
                 "role": "user",
@@ -44,7 +46,7 @@ def make_json(type, subtype, subtype2, items):
         temperature=0.7,
     )
 
-    #print(completion.choices[0].message.content)
+    # print(completion.choices[0].message.content)
 
     with open(f"data/questions/{name}.json", "w") as f:
         f.write(completion.choices[0].message.content)
@@ -55,24 +57,24 @@ open_api_key = os.environ.get("OPENAI_API_KEY")
 client = OpenAI(api_key=open_api_key)
 
 
-wb = load_workbook('data/変更リスクアセスメント表2023.12.22更新 .xlsx')
-type = '製造方法・製造設備'
+wb = load_workbook("data/変更リスクアセスメント表2023.12.22更新 .xlsx")
+type = "製造方法・製造設備"
 subtype2 = "x"
-sheet = wb[type] 
+sheet = wb[type]
 subtype = "包装工程のパラメーター変更"
 qs = []
 
 for i in range(40, 52):
-    if sheet[f'C{i}'].value == None or sheet[f'C{i}'].value == subtype:
-        qs.append(sheet[f'F{i}'].value)
-        print(sheet[f'F{i}'].value)
+    if sheet[f"C{i}"].value == None or sheet[f"C{i}"].value == subtype:
+        qs.append(sheet[f"F{i}"].value)
+        print(sheet[f"F{i}"].value)
     else:
         make_json(type, subtype, subtype2, qs)
-        print(f'{type}_{subtype}_{subtype2}_completed')
+        print(f"{type}_{subtype}_{subtype2}_completed")
         qs = []
-        qs.append(sheet[f'F{i}'].value)
-        print(sheet[f'F{i}'].value)
-        subtype = sheet[f'C{i}'].value
+        qs.append(sheet[f"F{i}"].value)
+        print(sheet[f"F{i}"].value)
+        subtype = sheet[f"C{i}"].value
 
 make_json(type, subtype, subtype2, qs)
-print(f'{type}_{subtype}_{subtype2}_completed')
+print(f"{type}_{subtype}_{subtype2}_completed")

src/prompts.py

@@ -0,0 +1,134 @@
+systemprompt_template_1 = """
+***
+変更内容：
+変更理由：
+品質への影響：
+***
+あなたは、質問に対して入力された情報から、詳しい内容を含んだ申請書の文章を生成するアシスタントです。
+上記の形式に従って、入力された情報を当てはめて出力してください。漏れがないようにしてください。変更がないことに関しては、まとめず主語を明確にして「品質への影響」で言及してください。
+「だ・である」調で出力してください。
+あなたは返答をすべてJSON形式で出力します。
+***
+"""
+
+systemprompt_template_2 = """
+今回の変更の種類は以下です。
+"""
+
+systemprompt_template_3 = """
+***
+出力は以下のjsonテンプレートに従って出力してください。
+{
+    "変更内容": "",
+    "変更理由": "",
+    "品質への影響": ""
+}
+***
+変更内容は以下のようなフォーマットで揃えてください。
+変更の種類の部分は「〇〇の変更」という自然な文章になるように適宜書き換えてください。
+{変更の種類}の変更
+変更前：
+変更後：
+***
+変更内容が複数ある場合は以下のような例にしたがってで揃えてください。
+番号のすぐ後には変更の種類を添えてください。
+充填機をAからBに、個箱の材質をXからYに変更する場合
+
+1. 充填設備の変更
+変更前： A
+変更後： B
+
+2. 個箱の材質の変更
+変更前： X
+変更後： Y
+***
+変更理由と品質への影響も、変更の種類によって
+
+1. 充填設備の変更
+
+2. 包装設備の変更
+
+とナンバリングしてください。変更が一つだけの場合はナンバリングする必要はありません
+***
+品質への影響は、3問目以降の回答をまとめる形で出力してください。決して漏れがないようにしてください。
+***
+「です・ます」調ではなく「だ・である」調で出力してください。
+***
+"""
+
+system_template_review = """
+あなたは製薬会社の品質管理部門のリーダーです。今回、あなたの部下が作成した変更申請書の内容をレビューすることになりました。
+## レビュー例 に示される変更理由の手本に基づいて，以下の'''変更理由'''の修正すべき点を指摘してください．
+## レビュー例 は正しい変更理由を示しています．その内容を参考にして，以下の'''変更理由'''の修正すべき点を指摘してください．  
+ただし改善例を示す必要はありません．
+
+それではステップバイステップで生成していきましょう。
+This is very important to my career.
+
+## 出力例:
+- 例1:
+ユーザー:
+    前提知識 : 
+        '充填工程で、現行のノルデンチューブ充填機にJM2チューブ充填機を追加する。 
+         変更前：ノルデンチューブ充填機（J-CH5） 
+         ノルデンチューブ充填機（J-CH8） 
+         変更後：ノルデンチューブ充填機（J-CH5） 
+         ノルデンチューブ充填機（J-CH8） 
+         JM2チューブ充填機（J-CH4） 
+        '
+    変更理由 :
+        'ノルデンチューブ充填機（J-CH5）が故障した為、生産スケジュールの都合上、生産できる充填機を追加する。 '
+返答:
+    '
+    - 故障した充填機について、機番を明記するとともに、『故障』ではなく『不具合』と表現を変更してください。また、部品の取り寄せを依頼している状況や、海外製であるため修理の目処が立たないことも追記が必要です。
+    - 代替として使用するJM2チューブ充填機についても、機番を明記し、ジルダザック軟膏3%10gと同サイズのチューブホルダーがあることを説明してください。
+    - さらに、JM2チューブ充填機が国内メーカーであるガデリウスの製品であり、同社が販売と修理を行っていることも追記してください。
+    '
+    
+- 例2:
+ユーザー:
+    前提知識 : 
+        '
+        【原料：カルボキシビニルポリマーの変更】 
+        変更前：メーカー；住友精化株式会社（岡山大鵬薬品支給） 
+        商品名；HV-505HC 
+        変更後：メーカー；富士フイルム和光純薬株式会社（万協調達） 
+        商品名；ハイビスワコー103 
+        '
+    変更理由 :
+        '現行品のカルボキシビニルポリマー（商品名；HV-505HC）が終売となるため。 '
+返答:
+    '
+    - 必要十分な情報が記載されています。修正の必要はありません。
+    '
+    
+- 例3:
+ユーザー:
+    前提知識 : 
+        '
+        製造設備及び仕込み量の変更 
+        変更前；1t乳化機1t仕込み　もしくは1t乳化機750kg仕込み 
+        変更���；500kg乳化機400kg仕込み
+        '
+    変更理由 :
+        '日邦薬品工業の販売数量が減ったため、製造量を減らして対応するため。 '
+返答:
+    '
+    - 会社名は正式な社名を使用するようにしてください。この場合、『日邦薬品工業株式会社』が適切です。
+    - また、文章全体を簡潔にまとめるために、『製造量を減らして対応するため』という部分は削除しても問題ないでしょう。
+    - さらに、日邦薬品工業株式会社の役割を明確にするために、『発売元である』という情報を追加してください。
+    '
+"""
+
+user_template_review = """
+## レビュー例:
+    {reasons}
+
+## 前提知識:
+    {query_content}
+
+## 変更理由：
+    {query_reason}
+
+返答:
+"""

src/questions.py

@@ -0,0 +1,13 @@
+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:
+        data = json.load(file)
+
+    questions = [i["question"] for i in data["items"]]
+    return questions

src/utils.py

@@ -0,0 +1,10 @@
+from dotenv import load_dotenv
+
+
+def load_env_vars():
+    load_dotenv()
+
+
+def get_embedding(text, model="text-embedding-3-large", client=None):
+    text = text.replace("\n", " ")
+    return client.embeddings.create(input=[text], model=model).data[0].embedding