commit

app.py

@@ -0,0 +1,115 @@
+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",
+                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 = """
+<h5>注意事項</h5>
+<p><small>＊当商品は、GPT API[gpt-3.5-turbo]を利用した株式会社KandaQuantumが提供するGenAI（ジェネレーションAI）ソリューションです。</small></p>
+<p><small>＊本プロダクトに内蔵されるGPTのAPI提供元OpenAIはAPIを介してユーザーから送信されたデータをモデルのトレーニングや改善に使わないことを明示しました。</small></p>
+<p><small>＊当社もデータ保持に関してOpenAIと同様の方針です（ユーザービリティ向上のため変更する可能性はございます）。</small></p>
+<p><small>＊当社は、製品の使用により発生したいかなる損害についても一切の責任を負いません。</small></p>
+"""
+
+iface = gr.Interface(
+    fn=greet,
+    inputs=[input_offer, input_company, input_depart],
+    outputs=[output_offer],
+    title="求人票作成AI CalqOffer", 
+    description=description,
+    article=article
+
+    )
+iface.launch()

requirements.txt

@@ -0,0 +1 @@
+openai==0.27.0