YKajima commited on
Commit
5d4d634
1 Parent(s): 3044ba7

add textbox for additional order

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -77,13 +77,14 @@ def validate_input_form(input_text, input_url):
77
  return
78
 
79
 
80
- async def chat(input_text, input_url):
81
  validate_input_form(input_text, input_url)
82
  summary = await summarize(input_text, input_url)
 
83
  logger.info(f"summary: {summary}")
84
  endpoint = os.environ.get("TWINROOM_API_BASE")
85
  payload = {
86
- "content": summary
87
  }
88
  headers = {'API-Key': os.environ.get("TWINROOM_API_KEY")}
89
  json_payload = json.dumps(payload)
@@ -97,18 +98,20 @@ async def chat(input_text, input_url):
97
 
98
  with gr.Blocks() as iface:
99
  # UI
100
- gr.Markdown("# パスカルくん \n## 使い方 \nテキスト、もしくはURLにパスカルくんに分析させたい記事の内容やURLを入力して「回答生成」ボタンを押すと回答が出力されます。 \n※テキスト、URLの両方に入力して実行した場合は、**URLが優先されます。** \n※テキスト、URLの両方が**空の場合は実行できません。** 必ずいずれかを入力してから実行してください。")
101
  with gr.Row():
102
  with gr.Column():
103
  input_text = gr.Textbox(label="テキスト")
104
  input_url = gr.Textbox(label="URL")
105
 
 
 
106
  chat_btn = gr.Button("回答生成")
107
  with gr.Column():
108
  output_text = gr.Textbox(label="回答")
109
 
110
  # Event handler
111
- chat_btn.click(fn=chat, inputs=[input_text, input_url], outputs=output_text)
112
 
113
  gr.Markdown("## トラブルシューティング")
114
  with gr.Accordion(label="「LLM APIの呼び出しに失敗しました。」というエラーが表示された場合", open=False):
 
77
  return
78
 
79
 
80
+ async def chat(input_text, input_url, additional_order):
81
  validate_input_form(input_text, input_url)
82
  summary = await summarize(input_text, input_url)
83
+ additional_prompt = f"【追加指示】: {additional_order}"
84
  logger.info(f"summary: {summary}")
85
  endpoint = os.environ.get("TWINROOM_API_BASE")
86
  payload = {
87
+ "content": summary + additional_prompt
88
  }
89
  headers = {'API-Key': os.environ.get("TWINROOM_API_KEY")}
90
  json_payload = json.dumps(payload)
 
98
 
99
  with gr.Blocks() as iface:
100
  # UI
101
+ gr.Markdown("# パスカルくん \n## 使い方 \nテキスト、もしくはURLにパスカルくんに分析させたい記事の内容やURLを入力して「回答生成」ボタンを押すと回答が出力されます。 \n※テキスト、URLの両方に入力して実行した場合は、**URLが優先されます。** \n※テキスト、URLの両方が**空の場合は実行できません。** 必ずいずれかを入力してから実行してください。 \n※追加指示から、パスカルくんに追加で指示を与えることができます。 \n例)敬語で話してください")
102
  with gr.Row():
103
  with gr.Column():
104
  input_text = gr.Textbox(label="テキスト")
105
  input_url = gr.Textbox(label="URL")
106
 
107
+ additional_order = gr.Textbox(label="追加指示")
108
+
109
  chat_btn = gr.Button("回答生成")
110
  with gr.Column():
111
  output_text = gr.Textbox(label="回答")
112
 
113
  # Event handler
114
+ chat_btn.click(fn=chat, inputs=[input_text, input_url, additional_order], outputs=output_text)
115
 
116
  gr.Markdown("## トラブルシューティング")
117
  with gr.Accordion(label="「LLM APIの呼び出しに失敗しました。」というエラーが表示された場合", open=False):