import os
import gradio as gr
import openai

class OpenAIUtils:
    
    @classmethod
    def use_openai_chatgpt_base(cls, prompt):
        """
        Use OpenAI's GPT-3 model, Davinci, to generate text based on the given prompt
        """
        openai.api_key = os.getenv('OPENAI_API_KEY')
        print("prompt")
        # print(prompt)

        try:
            res = openai.ChatCompletion.create(
            # res = openai.Completion.create(
                model="gpt-3.5-turbo",
                # model="gpt-4",
                messages=[
                        {"role": "system", "content": "あなたは企業の採用担当です"},
                        # {"role": "user", "content": prompt},
                        {"role": "user", "content": f"{prompt}"},
                    ]
                )
            string = res.choices[0]['message']['content']
        except openai.error.InvalidRequestError as e:
            print(f"Error: {e}")
            return None
        return string
    @classmethod
    def use_openai_davinci_base(cls, prompt):
        """
        Use OpenAI's GPT-3 model, Davinci, to generate text based on the given prompt
        """
        openai.api_key = os.getenv('OPENAI_API_KEY')
        try:
            res = openai.Completion.create(
                model="text-davinci-003",
                prompt=prompt,
                max_tokens=1024,
                temperature=0.5
            )
            string = res.choices[0].text
        except openai.error.InvalidRequestError as e:
            print(f"Error: {e}")
            return None
        return string

class CalqOffer:
    @classmethod
    def generate_reply_mail_prompt(cls, offer, company, depart):
        prompt = f"""
        {offer}
上記の内容をもとにして以下を必須項目として、求人票を作成してください（1000文字程度）。
【法人名】：{company}
【部門】：{depart}
【求人票のタイトル案】
[落ち着いたタイトル案（3つ提案して）]
[ポップなタイトル案（3つ提案して）]
【仕事内容】
【必要なスキル・経験（箇条書き）】
【歓迎するスキル・経験（箇条書き）】
【給与】
【勤務地】
【雇用形態】
【待遇】
        """
        # print(prompt)
        # return OpenAIUtils.use_openai_davinci_base(prompt)
        return OpenAIUtils.use_openai_chatgpt_base(prompt)

    def post_process_mail_reply(offer_reply):
        remove_list = ["（箇条書き）"]

        for item in remove_list:
            offer_reply = offer_reply.replace(item, "")

        offer_reply = offer_reply.replace("（3つ提案して）", "（3つ）")
        return offer_reply


def greet(offer, company, depart):
    offer_reply = CalqOffer.generate_reply_mail_prompt(offer, company, depart)
    # print(offer_reply)
    offer_reply = CalqOffer.post_process_mail_reply(offer_reply)
    print(offer_reply)
    return offer_reply

input_offer = gr.Textbox(label="採用したい人材の情報を列挙してください（例：マネージャー、年収500万円、東京）")
input_company = gr.Textbox(label="あなたの会社名（オプション）")
input_depart = gr.Textbox(label="部門（オプション）")
output_offer = gr.Textbox(label="求人票提案文章(最大1分ほどお待ちください)")
description="""
"""


article = """

"""

iface = gr.Interface(
    fn=greet,
    inputs=[input_offer, input_company, input_depart],
    outputs=[output_offer],
    title="", 
    description=description,
    article=article

    )
iface.launch()