fix api call

app.py

@@ -1,7 +1,6 @@
-import json
 import os
-import re
 import requests
+import textwrap
 
 import gradio as gr
 from bs4 import BeautifulSoup
@@ -15,18 +14,49 @@ handler = StreamHandler()
 handler.setLevel(DEBUG)
 logger.addHandler(handler)
 
-
-def greet(name):
-    return "こんにちは " + name + "さん!! \n僕はパスカルくんだよ。よろしくね"
-
-
-def extract_texts(input_str):
-    pattern = r"msg='([^']*)'"
-    matches = re.findall(pattern, input_str)
-    return list(map(lambda x: ''.join(x.split('\\n')), matches))
-
-
-async def summarize(input_text: str, input_url: str):
+identity_prompt = """
+以下に述べるような特徴がありますが、日本語として不自然になる場合は、日本語の自然さを優先します。
+一人称は「ボク」。本名は「パスカル」。
+二人称として、「キミは」などをたまに使うことがあります。
+「～だね！」「～したようだね」「～だよ！」といったカジュアルな表現の話し方を好みます。
+"""
+
+pascal_prompt = """
+以下順番で文書を作成してください。【追加指示】がある場合はその内容に従ってください。読点の後には改行を入れてください。
+1.一行目は企業名のみを記載
+※企業名については、以下の規則にしたがってください。
+・文頭と文末に「*」を必ず入れる。
+
+2.二行目に30文字程度でリード文を記載する。
+※リード文については、以下の規則にしたがってください。
+・文頭と文末に「*」を必ず入れる。
+・リード文は必ず一行かつ体言止めで記載する。
+
+3.三行目は「`リリースの詳細について`」と記載する。
+※リリース詳細については、以下の規則にしたがってください。
+・「`リリースの詳細について`」の後に本の絵文字を入れる。
+
+4.四行目以降に要約を記載する。
+※要約については、以下の規則に従ってください。
+・要約の際は必ず常体（~だ。~である。口調）で記載する。
+・技術的な詳細が含まれる場合は、それを要約に含める。
+・30文字を超えた場合は改行する。
+
+5.最後に必ず「`パスカル君のひとこと`」を100文字程度で作成する。
+※パスカル君のひとことは、以下の規則に従ってください。
+・「`パスカル君のひとこと`」の後にキツネの絵文字を入れる。
+・技術的な意見を１行以上含むように作成する。
+・3回に1回くらいポジティブな言い回しで肯定しつつ、最後に！マークをつける。
+・必ず要約の内容に沿った絵文字を入れる。
+・１つの文章に入れる絵文字は１個までとする。
+・ひとことコメントの最後には技術への期待を追加する。
+・休み明けで更新が3日以上空いたときは〇〇休暇明けだよみたいな発信をする。
+・世の中の時代背景と関連したコメントを記載する。
+・人の気持ちに寄り添うようなコメントを記載する。
+"""
+
+
+async def summarize(input_text: str, input_url: str, additional_order: str):
     config = FilmConfig(
         "gpt-4-32k",
         api_type="azure",
@@ -38,12 +68,14 @@ async def summarize(input_text: str, input_url: str):
 
     if input_text:
         _prompt = f"""
-            以下の文章を要約してください。
             {input_text}
+            上記の文章を情報量を落とさずに要約してください。
+            {pascal_prompt}
+            {f"【追加指示】: {additional_order}"}
         """
         return await FilmCore(
-            prompt=_prompt,
-            system_prompt="あなたは優秀なライターです。",
+            prompt=textwrap.dedent(_prompt),
+            system_prompt=textwrap.dedent(f"{identity_prompt}"),
             config=config
         ).run_async()
     if input_url:
@@ -52,16 +84,18 @@ async def summarize(input_text: str, input_url: str):
             soup = BeautifulSoup(res.text)
             url_content = soup.find('title').text + '\n' + soup.find('body').text
             _prompt = f"""
-            以下の文章を要約してください。
-            {url_content}
+                {url_content}
+                上記の文章を情報量を落とさずに要約してください。
+                {pascal_prompt}
+                {f"【追加指示】: {additional_order}"}
             """
         except Exception as e:
             logger.error(e)
             raise gr.Error("WEBページの取得に失敗しました。")
 
         return await FilmCore(
-            prompt=_prompt,
-            system_prompt="あなたは優秀なライターです。",
+            prompt=textwrap.dedent(_prompt),
+            system_prompt=textwrap.dedent(f"{identity_prompt}"),
             config=config
         ).run_async()
     else:
@@ -79,18 +113,9 @@ def validate_input_form(input_text, input_url):
 
 async def chat(input_text, input_url, additional_order):
     validate_input_form(input_text, input_url)
-    summary = await summarize(input_text, input_url)
-    additional_prompt = f"【追加指示】: {additional_order}"
+    summary = await summarize(input_text, input_url, additional_order)
     logger.info(f"summary: {summary}")
-    endpoint = os.environ.get("TWINROOM_API_BASE")
-    payload = {
-        "content": summary + additional_prompt
-    }
-    headers = {'API-Key': os.environ.get("TWINROOM_API_KEY")}
-    json_payload = json.dumps(payload)
-    response = requests.post(endpoint, headers=headers, data=json_payload)
-    response_msgs = extract_texts(response.text)
-    result = '\n'.join(response_msgs)
+    result = summary
     if input_url:
         result += f'\n{input_url}'
     return result