risk_list = """
"自然災害":"地震、洪水、台風、異常気象による被害など",
"製造物責任":"製品の瑕疵（かし）、リコールなど",
"サイバーセキュリティ":"サイバー攻撃やデータ漏洩など",
"事故":"交通事故や設備交渉など",
"感染症":"新型ウイルスのパンデミックなど",
"環境問題":"工場排水の不適切な処理など"
"不正会計":"粉飾決算、横領など",
"調達価格":"原材料、原油価格の高騰など",
"投資活動":"投資活動の失敗による損失など",    
"""

company_risk_list = risk_list

# risk list ex
risk_ex_t = {
    "ex1": {
        "news": "台風7号はきょう11日午前3時、「強い」台風から、「非常に強い」勢力に変わりました。きょう11日の夕方にかけて小笠原諸島に最接近し、小笠原諸島は大荒れの天気でしょう。14日(月)には勢力は一つランクを下げるものの、「強い」勢力で暴風域を伴って本州に接近し、15日(火)頃から16日(水)にかけて東日本や西日本にかなり接近、上陸する恐れがあります。",
        "company_info": "株式会社ブルーオーシャンフーズ\n\n事業内容:\nブルーオーシャンフーズは、持続可能な海洋資源を利用した健康食品の開発・販売を行っています。特に、高品質な海藻類を使用した商品が高い評価を受けています。\n\n市場情報:\n健康志向と環境保護意識の高まりにより、持続可能な食品市場が拡大しています。ブルーオーシャンフーズは、このトレンドに乗じて市場シェアを拡大しています。\n\nビジネス展開:\n国内販売に加え、アジア圏への輸出を強化しています。2024年度には、新たな健康食品のラインナップを市場に投入する予定です。",
        "risk_list": '{"自然災害":"持続可能な海洋資源を利用した健康食品の開発・販売を行っているため、地震、洪水、台風、異常気象による海洋資源への被害がリスクとなる。","感染症":"国内販売及びアジア圏への輸出を強化しているため、新型ウイルスのパンデミックが業務やサプライチェーンに影響を与えるリスクがある。","環境問題":"持続可能な海洋資源を利用している企業であり、環境保護意識の高まりの中で工場排水の不適切な処理などが発覚すると企業イメージに損害を与える可能性がある。","調達価格":"高品質な海藻類などの原材料を使用しているため、原材料価格の高騰がコスト増加につながり、収益性に影響を与えるリスクがある。"}',
        "risk": '{"ja": "yes", "risk": "yes", "risk_key":"自然災害","reason":"台風7号はきょう11日午前3時、「強い」台風から、「非常に強い」勢力に変わりました。"}',
        "title": '{"title":"[自然災害]台風7号接近、持続可能な海洋資源企業への影響、ご確認お願い致します。"}',
        "summary": '{"summary":"台風7号はきょう11日午前3時、「強い」台風から、「非常に強い」勢力に変わりました。15日(火)頃から16日(水)にかけて東日本や西日本にかなり接近、上陸する恐れがあります。"}',
        "alert": '{"alert":"!!台風7号接近、非常に悪天候、潜在的な被害、および混乱の高いリスク。海洋資源の確保と物流に影響が出る恐れがあります。企業は、暴風域の影響を受ける前に、対策を講じる必要があります。!!"}',
    },
    "ex2": {
        "news": "JetBrainsが提供するCI/CDツール「TeamCity」に深刻な脆弱性「CVE-2024-23917」が明らかとなった。悪用されるとリモートよりコードを実行されるおそれがある。同製品に認証回避の脆弱性「CVE-2024-23917」が明らかとなったもの。HTTP経由でアクセスできる場合、認証をバイパスして「TeamCityサーバ」の管理を行うことが可能となり、リモートよりコードを実行されるおそれがある。",
        "company_info": "株式会社サクラテクノロジー\n\n事業内容:\nサクラテクノロジーは、最先端のAI技術を活用して、製造業向けの自動化ソリューションを提供しています。特に、ロボティクスとデータ分析を組み合わせた製品で市場をリードしています。\n\n市場情報:\n製造業の自動化ニーズの増加に伴い、サクラテクノロジーの市場は年平均15%の成長が見込まれています。特に、自動車および電子機器製造業界からの強い需要があります。\n\nビジネス展開:\n国内市場に加えて、北米とヨーロッパ市場への進出を計画しており、2025年までに全世界での事業展開を目指しています。また、新しいAI技術の研究開発にも力を入れており、次世代製品の開発に注力しています。",
        "risk_list": '{"サイバーセキュリティ": "最先端のAI技術を活用しているため、サイバー攻撃やデータ漏洩のリスクがある。","不正会計": "拡大する事業規模と複雑化する経営状況における粉飾決算や横領などの財務不正リスク。","投資活動": "新しいAI技術の研究開発に力を入れており、次世代製品の開発に注力していることから、投資活動の失敗による損失が発生するリスクがある。"}',
        "risk": '{"ja": "yes", "risk": "yes", "risk_key":"サイバーセキュリティ","reason":"JetBrainsが提供するCI/CDツール「TeamCity」に深刻な脆弱性「CVE-2024-23917」が明らかとなった。"}',
        "title": '{"title":"[サイバーセキュリティ] JetBrainsのCI/CDツール「TeamCity」に深刻な脆弱性発覚、サクラテクノロジーの製造自動化への影響"}',
        "summary": '{"summary":"JetBrainsのCI/CDツール「TeamCity」に深刻な脆弱性「CVE-2024-23917」が発見されました。 この脆弱性を悪用すると、リモートからコードが実行される危険性があります。"}',
        "alert": '{"alert":"!!重要!!JetBrainsのCI/CDツール「TeamCity」に深刻な脆弱性「CVE-2024-23917」が発見されました。このツールを使用している場合、直ちに対策を講じる必要があります。これは、同社のAIと自動化ソリューションの安全性および信頼性に直接的な影響を及ぼす可能性があります。!!"}',
    },
    "ex3": {
        "news": "2024年1月1日16時10分頃、石川県能登地方を震源とするマグニチュード7.6（最大震度7）の地震が発生しました。 被災された方々には謹んでお見舞い申し上げます。 本地震における本投資法人の保有物件所在地の最大震度は4であり、現在のところ人的被害や運用状況に重要な影響を及ぼす被害等は確認されておりません。",
        "company_info": "株式会社ブルーオーシャンフーズ\n\n事業内容:\nブルーオーシャンフーズは、持続可能な海洋資源を利用した健康食品の開発・販売を行っています。特に、高品質な海藻類を使用した商品が高い評価を受けています。\n\n市場情報:\n健康志向と環境保護意識の高まりにより、持続可能な食品市場が拡大しています。ブルーオーシャンフーズは、このトレンドに乗じて市場シェアを拡大しています。\n\nビジネス展開:\n国内販売に加え、アジア圏への輸出を強化しています。2024年度には、新たな健康食品のラインナップを市場に投入する予定です。",
        "risk_list": '{"自然災害":"持続可能な海洋資源を利用した健康食品の開発・販売を行っているため、地震、洪水、台風、異常気象による海洋資源への被害がリスクとなる。","感染症":"国内販売及びアジア圏への輸出を強化しているため、新型ウイルスのパンデミックが業務やサプライチェーンに影響を与えるリスクがある。","環境問題":"持続可能な海洋資源を利用している企業であり、環境保護意識の高まりの中で工場排水の不適切な処理などが発覚すると企業イメージに損害を与える可能性がある。","調達価格":"高品質な海藻類などの原材料を使用しているため、原材料価格の高騰がコスト増加につながり、収益性に影響を与えるリスクがある。"}',
        "risk": '{"ja": "yes", "risk": "yes", "risk_key":"自然災害","reason":"2024年1月1日16時10分頃、石川県能登地方を震源とするマグニチュード7.6（最大震度7）の地震が発生しました。"}',
        "title": '{"title":"[自然災害]震度4の警鐘：持続可能な食品市場における災害リスク"}',
        "summary": '{"summary":"2024年1月1日16時10分頃、石川県能登地方を震源とするマグニチュード7.6（最大震度7）の地震が発生しました。保有物件の最大震度は4で、本地震における本投資法人の保有物件所在地の最大震度は4であり、現在のところ人的被害や運用状況に重要な影響を及ぼす被害等は確認されておりません。"}',
        "alert": '{"alert":"!!石川県能登地方で発生したM7.6の地震、最大震度7。事業活動への直接的な影響は現時点で報告されていませんが、地震が持続可能な海洋資源へ及ぼす可能性のある影響に注意が必要です。震度4の揺れが報告された地域に保有物件が存在するため、引き続き状況を注視し、必要に応じた措置を講じることを推奨します。"}',
    },
}
risk_ex_f = {
    "ex1": {
        "news": "花巻東高（岩手）で通算140本塁打を放った佐々木麟太郎内野手が、米スタンフォード大学へ進学することが決まった。同大学が公式ホームページで発表した。",
        "company_info": "株式会社ブルーオーシャンフーズ \n 事業内容: \n ブルーオーシャンフーズは、持続可能な海洋資源を利用した健康食品の開発・販売を行っています。特に、高品質な海藻類を使用した商品が高い評価を受けています。 \n 市場情報: \n 健康志向と環境保護意識の高まりにより、持続可能な食品市場が拡大しています。ブルーオーシャンフーズは、このトレンドに乗じて市場シェアを拡大しています。 \n ビジネス展開: \n 国内販売に加え、アジア圏への輸出を強化しています。2024年度には、新たな健康食品のラインナップを市場に投入する予定です。",
        "risk_list": '{"自然災害":"持続可能な海洋資源を利用した健康食品の開発・販売を行っているため、地震、洪水、台風、異常気象による海洋資源への被害がリスクとなる。","感染症":"国内販売及びアジア圏への輸出を強化しているため、新型ウイルスのパンデミックが業務やサプライチェーンに影響を与えるリスクがある。","環境問題":"持続可能な海洋資源を利用している企業であり、環境保護意識の高まりの中で工場排水の不適切な処理などが発覚すると企業イメージに損害を与える可能性がある。","調達価格":"高品質な海藻類などの原材料を使用しているため、原材料価格の高騰がコスト増加につながり、収益性に影響を与えるリスクがある。"}',
        "risk": '{"ja": "yes", "risk": "no", "risk_key":"","reason":""}',
    },
    "ex2": {
        "news": "「ももいろクローバーZ」百田夏菜子（29）が14日、東京・渋谷で行われた「カラダうごかせ！ニッポン！」プロジェクト発足記者発表会に登場。「KinKi　Kids」堂本剛（44）と結婚発表後、初めての公の場で、左手に指輪はしていなかったが、幸せいっぱいの笑顔を見せた。",
        "company_info": "株式会社サクラテクノロジー \n 事業内容: サクラテクノロジーは、最先端のAI技術を活用して、製造業向けの自動化ソリューションを提供しています。特に、ロボティクスとデータ分析を組み合わせた製品で市場をリードしています。 \n 市場情報: \n 製造業の自動化ニーズの増加に伴い、サクラテクノロジーの市場は年平均15%の成長が見込まれています。特に、自動車および電子機器製造業界からの強い需要があります。 \n ビジネス展開: \n 国内市場に加えて、北米とヨーロッパ市場への進出を計画しており、2025年までに全世界での事業展開を目指しています。また、新しいAI技術の研究開発にも力を入れており、次世代製品の開発に注力しています。",
        "risk_list": '{"サイバーセキュリティ": "最先端のAI技術を活用しているため、サイバー攻撃やデータ漏洩のリスクがある。","不正会計": "拡大する事業規模と複雑化する経営状況における粉飾決算や横領などの財務不正リスク。","投資活動": "新しいAI技術の研究開発に力を入れており、次世代製品の開発に注力していることから、投資活動の失敗による損失が発生するリスクがある。"}',
        "risk": '{"ja": "yes", "risk": "no", "risk_key":"","reason":""}',
    },
}

# prompt
## company
com_sys_msg = """
You are a helpful business analyst expert designed to \
provide me with the information on risks appearing in the company information and the risk list, \
formatted in JSON.
"""
com_usr_msg = """
Based on the entered company information and the risks provided, \
I want you to select the risks that may occur in the risk list for the entered company. \
This risks should include, but not be limited to, financial, operational, market, and compliance risks associated with the company's specific industry, size, and geographic location. \
Find the original text on which the risk was judged. Output the results in JSON format.
like {{"risk1" : "reason1","risk2":"reason2"}}
Think about it a few more steps \n\n
example:
***
company_information: {c1} \n
risk_list: {r1} \n
output: {o1} \n\n
***
question:
***
company_information: {c2} \n
risk_list: {r2} \n
output:
***
"""
## risk　list
risk_sys_msg = """
You are a helpful business analyst expert designed to \
provide me with the information on risks appearing in the news, formatted in JSON.
"""
risk_usr_msg_old = """
Enter the news and risks company information that company can refer to. \
Determine whether there is risk information in the news. \
If there is, output yes, the risk information contained, \
and the original news text basis that can be judged. \
If there is no, only output no. 
Determine if the news is written in Japanese. Output the results in JSON format. \
like {{'ja': determine if the news is written in Japanese ,'risk': Is there any risk , 'risk_key': Risk type ,'reason': The basis of the original text}} \
Think about it a few more steps \n\n \
news: {n1}  \n
company_information: {c1} \n
risk_list:{r1}\n
output:{o1}  \n\n
news: {n2}  \n
company_information: {c2} \n
risk_list:{r2}\n
output:{o2}  \n\n
news: {n3}  \n
company_information: {c3} \n
risk_list:{r3}\n
output:{o3}  \n\n
news: {n4}\n
company_information: {c4} \n
risk_list:{r4}\n
output:
"""

risk_usr_msg = """
Enter the news and risks company information that company can refer to. \
Determine whether there is risk information in the news. \
If there is, output yes, the risk information contained, \
and the original news text basis that can be judged. \
If there is no, only output no. 
Determine if the news is written in Japanese. Output the results in JSON format. \
like {{'ja': determine if the news is written in Japanese ,'risk': Is there any risk , 'risk_key': Risk type ,'reason': The basis of the original text}} \
Think about it a few more steps \n\n \
example:
***
news: {n1}  \n
company_information: {c1} \n
risk_list:{r1}\n
output:{o1}  \n\n
news: {n3}  \n
company_information: {c3} \n
risk_list:{r3}\n
output:{o3}  \n\n
***
question:
***
news: {n4}\n
company_information: {c4} \n
risk_list:{r4}\n
output:
***
"""

## title
title_usr_msg = """
I need a compelling and insightful title for a report based on input news and company information. \
The news highlights potential risks or challenges in the {risk} sector. Generate a concise, \
attention-grabbing title that encapsulates the essence of the risks discussed in the news article. \
frame the title to capture the attention of company decision-makers. Output the results in JSON format. \
like {{'title':generated title}} \
Think about it a few more steps \n\n \
example:
***
news: {n1} \n
company_information: {c1} \n
output:{o1} \n\n
news: {n2} \n
company_information: {c2} \n
output:{o2} \n\n
***
question:
***
news: {n3} \n
company_information: {c3} \n
output:
***
"""
## sum
summary_usr_msg = """
I need a concise news summary focusing on risks based on the provided news input. \
Please craft a brief summary that highlights potential risks associated with the given news. Within two sentences\
Output the results in JSON format. \
like {{'summary':generated summary}} \
Think about it a few more steps \n\n \
example:
***
news: {n1} \n
output:{o1} \n\n
news: {n2} \n
output:{o2} \n\n
***
question:
***
news: {n3} \n
output:
"""
## alert msg
alert_usr_msg = """
I need assistance in creating an alert message based on the risk-related content of the entered news and company information. \
As a risk analyst, guide me in generating a concise and informative alert message summarizing the potential {risk} risk highlighted for company in the news. \
This alert is crucial for making timely decisions and assessing the impact of the information. \
Output the results in JSON format.\
like {{'alert':generated alert message}} \
Think about it a few more steps \n\n \
example:
***
news: {n1} \n
company_information: {c1} \n
output:{o1} \n\n
news: {n2} \n
company_information: {c2} \n
output:{o2} \n\n
***
question:
***
news: {n3} \n
company_information: {c3} \n
output:
***
"""
## report
report_msg = """
## アラートメッセージ

### {title}

### [サマリ]
{summary}

### [アラート]
{alert}

### [本文]
{news}

#### [リスク]
{risk}

#### [根拠]
{reason}

### [会社リスク]
{company_risk}
"""
## japanese
translate_sys_msg = """
You are a helpful Japanese translation expert designed to \
provide me translate non-Japanese text, formatted in JSON.
"""

translate_usr_msg = """
Translate the following non-Japanese text in a dictionary into formal Japanese for me. \
Do not translate proper nouns. \
If text is Japanese, return unchanged. \
Ensure the correctness of grammar and vocabulary. \
Output the results in JSON format.
like {{'title':translated title,'summary':translated summary,'alert':translated alert,'news':translated news,'risk_key':translated risk_key,'reason':translated reason,'company_risk':translated company_risk}}
input: {dictionary}
output: 
"""

news_e = "The lingering storm system that has battered California for days continues to pose a flooding risk for southern parts of the state. A rare tornado warning was issued for parts of San Diego County. The warning has expired, but heavy rain and lightning were still possible. The San Diego region remains under a flood watch through tonight because of excessive rainfall, the National Weather Service said, and residents in mountain areas are being warned about rockslides and rocks on the roads."
news1 = "14日午後6時33分ごろ、鹿児島市の桜島・南岳山頂火口で爆発的噴火があり、噴煙が火口から5千メートルの高さまで上がった。気象庁は熊本、宮崎、鹿児島各県の一部に降灰予報を出した。鹿児島県によると、けが人や建物被害は確認されていない。5千メートルに達したのは、同じ火口で噴火した"
news2 = "岸田首相（自民党総裁）は１４日午前の衆院予算委員会集中審議で、自民党派閥の政治資金規正法違反事件を受け、規正法違反で会計責任者が逮捕・起訴された場合に議員本人も処分できるよう党則を改正する考えを明らかにした。３月１７日の党大会で決定する意向だ。"
news3 = "楽天グループの去年1年間の連結決算は3394億円の最終赤字になりました。赤字額は過去最大だったおととしより縮小したものの、赤字は5年連続で、その大きな要因となっているのがモバイル事業です。これまで、携帯電話の基地局建設には1兆円以上を投じていますが、その一方で、去年12月時点での契約者数は609万件と大手3社の10分の1以下です。また、基地局建設にあたって発行した社債、いわゆる借金の返済額は今年と来年の2年間でおよそ8000億円に上りますが、三木谷社長は「資金繰りは心配ない」と強調しました。"
news4 = "LINEヤフーは14日、新たに従業員などの個人情報13万件以上が流出した可能性があると発表した。LINEヤフーは2023年11月に大株主である韓国ネット大手・NAVERがサイバー攻撃を受けたことでLINE利用者の個人情報44万件が不正アクセスを受けた可能性があると発表していた。追加調査の結果、NAVERのサーバー経由で従業員などの氏名や顔写真など、個人情報が追加で約8万件、また、他の委託先経由で従業員などの個人データが6万件近く、あわせて13万件以上が流出した可能性が確認された。"
news5 = "韓国人俳優のマ・ドンソクが14日、都内で行われた主演映画『犯罪都市 NO WAY OUT』（2月23日公開）のジャパンプレミアにイ・サンヨン監督、青木崇高、國村隼とともに出席。初の公式来日を果たした。"

com1 = "ネクストエナジーソリューション株式会社\n\n事業内容:\n再生可能エネルギーの普及と利用促進を目的として設立されたネクストエナジーソリューションは、太陽光発電や風力発電などの設置・運営を手掛けています。また、エネルギー管理システムの開発にも注力しています。\n\n市場情報:\n再生可能エネルギーへの関心が高まる中、同社の事業領域は急速に成長しています。特に、企業のCO2排出量削減ニーズが事業拡大の大きなドライバーとなっています。\n\nビジネス展開:\n国内市場での成功を基に、東南アジア市場への展開を進めています。また、エネルギー貯蔵技術の研究開発により、新たなビジネスモデルの構築を目指しています。"
com2 = "アドバンストヘルスケア株式会社\n\n事業内容:\nアドバンストヘルスケアは、医療機器とデジタルヘルスソリューションを融合させた製品を提供しています。特に、在宅医療を支援するデバイスの開発に力を入れています。\n\n市場情報:\n高齢化社会の進展と共に、在宅医療市場は拡大傾向にあります。同社の提供する革新的な製品は、この市場での需要を捉えています。\n\nビジネス展開:\n国内市場での実績を基に、北米およびヨーロッパ市場への進出を計画しています。また、新しい医療技術の研究開発に投資を続け、市場のリーダーとしての地位を確立することを目指しています。"


prompt_dict = {
    "risk_dict": risk_list,
    "company_risk_list": company_risk_list,
    "risk_ex_t": risk_ex_t,
    "risk_ex_f": risk_ex_f,
    "com_sys_msg": com_sys_msg,
    "com_usr_msg": com_usr_msg,
    "risk_sys_msg": risk_sys_msg,
    "risk_usr_msg": risk_usr_msg,
    "title_usr_msg": title_usr_msg,
    "summary_usr_msg": summary_usr_msg,
    "alert_usr_msg": alert_usr_msg,
    "report_msg": report_msg,
    "translate_sys_msg": translate_sys_msg,
    "translate_usr_msg": translate_usr_msg,
}

import pickle


def save_dict(prompt_dict):
    with open("prompt_dict.pkl", "wb") as fi:
        pickle.dump(prompt_dict, fi)


def load_dict():
    with open("prompt_dict.pkl", "rb") as fi:
        prompt_dict = pickle.load(fi)
    return prompt_dict