Update app.py

app.py

@@ -40,18 +40,18 @@ def extract_text_from_pdf(pdf_path):
         text += page.get_text()
     return text
 
-def summarize_text_with_chat(text, max_length=10000):
+def summarize_text_with_chat(text, max_length=20000):
     """OpenAIのChat APIを使ってテキストを要約。"""
     openai.api_key = os.getenv('OPEN_AI_API_KEYS')
     trimmed_text = text[:max_length]
     response = openai.chat.completions.create(
-        model="gpt-3.5-turbo-0125",
+        model="gpt-4-0125-preview",
         messages=[
             {"role": "system", "content": "次の文書を要約してください。必ず'## タイトル', '## 要約', '## 専門用語解説'を記載してください。"},
             {"role": "user", "content": trimmed_text}
         ],
         temperature=0.7,
-        max_tokens=1000
+        max_tokens=2000
     )
     summary_text = response.choices[0].message.content
     total_token = response.usage.total_tokens
@@ -111,7 +111,7 @@ def gradio_interface():
         summaries.append(summary_info)
 
     summaries_markdown = "\n---\n".join(summaries)  # 要約を水平線で区切る
-    return summaries_markdown + f"\n全ての要約で使用されたトータルトークン数: {total_tokens_used}"
+    return summaries_markdown + f"\n---\n全ての要約で使用されたトータルトークン数: {total_tokens_used}"
 
 iface = gr.Interface(
     fn=gradio_interface,