Create app.py

app.py

@@ -0,0 +1,56 @@
+import openai
+from openai import AsyncOpenAI
+import os
+import json 
+import pandas as pd
+
+async def text2theme(p, openai_key=os.environ.get('OPENAI_KEY'), target="ユーザ"):
+    """
+    input1 (text): 「ペットボトル」という項目は何に仕分けしたら良いですか？
+    input2 (text): default
+    input3 (text): ユーザ
+    output1 (dataframe): 仕分け候補
+    """
+    if openai_key == "default":
+        os.environ['OPENAI_API_KEY'] = os.environ.get('OPENAI_KEY')
+    else:
+        os.environ['OPENAI_API_KEY'] = openai_key
+        
+    header = '{"1":"'
+    prompt=f'{p}\n{target}にとっての順位をkeyとして1~の数値をintに、訴求要素をvalueに、１階層のjsonだけを返し、バッククオートなどは含めないでください。\njson:{header}'
+    #print(prompt)1
+    client = AsyncOpenAI()
+    response = await client.chat.completions.create(
+    model='gpt-4',
+    messages=[
+        {
+        "role": "system",
+        "content": "あなたは会計士と税理士の資格を持っており、全ての質問にjsonで回答します。"
+        },
+        {
+        "role": "user",
+        "content":  prompt
+            
+        },
+    ],
+    top_p=1,
+    frequency_penalty=0,
+    presence_penalty=0
+    )
+    
+    json_data = header + response.choices[0].message.content
+    try:
+        # JSONデータを読み込む
+        j = json.loads(json_data)
+    except json.decoder.JSONDecodeError as e:
+        # JSON解析エラーが発生した場合、エラーの位置を特定
+        error_position = e.pos  # エラーが発生した位置
+        print("JSONデコードエラー:", e)
+        print("エラーの周辺の文字:", json_data[max(0, error_position-10):error_position+10])
+    
+    
+    data = []
+    for k,v in j.items():
+        data.append([k,v])
+
+    return pd.DataFrame(data, columns=['rank', 'theme'])