sonoisa commited on
Commit
1396cfc
·
verified ·
1 Parent(s): 4213e17

Update index.html

Browse files
Files changed (1) hide show
  1. index.html +10 -6
index.html CHANGED
@@ -205,6 +205,7 @@ def process_prompt(prompt, history, context, platform, endpoint, azure_deploymen
205
  messages = []
206
  for user_message, assistant_message in history:
207
  if user_message is not None and assistant_message is not None:
 
208
  messages.append({ "role": "user", "content": user_message })
209
  messages.append({ "role": "assistant", "content": assistant_message })
210
 
@@ -226,6 +227,7 @@ def process_prompt(prompt, history, context, platform, endpoint, azure_deploymen
226
  http_client=http_client
227
  )
228
 
 
229
  completion = openai_client.chat.completions.create(
230
  messages=messages,
231
  model=model_name,
@@ -238,7 +240,8 @@ def process_prompt(prompt, history, context, platform, endpoint, azure_deploymen
238
  raise gr.Error(completion.error["message"])
239
  else:
240
  message = completion.choices[0].message
241
- return message.content
 
242
 
243
  except Exception as e:
244
  if hasattr(e, "message"):
@@ -408,9 +411,12 @@ def main():
408
  chatbot = gr.Chatbot(
409
  CHAT_HISTORY,
410
  elem_id="chatbot", render=False, height=500, show_copy_button=True,
411
- render_markdown=False, likeable=False, layout="bubble",
412
  avatar_images=[None, Path("robot.png")])
413
 
 
 
 
414
  chatbot.change(None, inputs=[chatbot, save_chat_history_to_url], outputs=None,
415
  # チャット履歴をクエリパラメータに保存する。
416
  js=save_or_delete_chat_history, show_progress="hidden")
@@ -420,10 +426,8 @@ def main():
420
 
421
  chat = gr.ChatInterface(process_prompt,
422
  title="Chat with your PDF",
423
- chatbot=chatbot,
424
- textbox=gr.Textbox(
425
- placeholder="Type a message...",
426
- render=False, container=False, scale=7),
427
  additional_inputs=[context, platform, endpoint, azure_deployment, azure_api_version, api_key, model_name, max_tokens, temperature],
428
  examples=[['''制約条件に従い、以下の研究論文で提案されている技術や手法について要約してください。
429
 
 
205
  messages = []
206
  for user_message, assistant_message in history:
207
  if user_message is not None and assistant_message is not None:
208
+ user_message = user_message.replace("{context}", context)
209
  messages.append({ "role": "user", "content": user_message })
210
  messages.append({ "role": "assistant", "content": assistant_message })
211
 
 
227
  http_client=http_client
228
  )
229
 
230
+ bot_response = ""
231
  completion = openai_client.chat.completions.create(
232
  messages=messages,
233
  model=model_name,
 
240
  raise gr.Error(completion.error["message"])
241
  else:
242
  message = completion.choices[0].message
243
+ bot_response += message.content
244
+ yield bot_response
245
 
246
  except Exception as e:
247
  if hasattr(e, "message"):
 
411
  chatbot = gr.Chatbot(
412
  CHAT_HISTORY,
413
  elem_id="chatbot", render=False, height=500, show_copy_button=True,
414
+ sanitize_html=False, render_markdown=False, likeable=False, layout="bubble",
415
  avatar_images=[None, Path("robot.png")])
416
 
417
+ chat_message_textbox = gr.Textbox(placeholder="Type a message...",
418
+ render=False, container=False, scale=7)
419
+
420
  chatbot.change(None, inputs=[chatbot, save_chat_history_to_url], outputs=None,
421
  # チャット履歴をクエリパラメータに保存する。
422
  js=save_or_delete_chat_history, show_progress="hidden")
 
426
 
427
  chat = gr.ChatInterface(process_prompt,
428
  title="Chat with your PDF",
429
+ chatbot=chatbot,
430
+ textbox=chat_message_textbox,
 
 
431
  additional_inputs=[context, platform, endpoint, azure_deployment, azure_api_version, api_key, model_name, max_tokens, temperature],
432
  examples=[['''制約条件に従い、以下の研究論文で提案されている技術や手法について要約してください。
433