ryanrwatkins commited on
Commit
295d089
·
verified ·
1 Parent(s): f3cc8e9

Update Gradio_UI.py

Browse files
Files changed (1) hide show
  1. Gradio_UI.py +3 -1
Gradio_UI.py CHANGED
@@ -157,9 +157,11 @@ def stream_to_gradio(
157
  final_answer = handle_agent_output_types(final_answer)
158
 
159
  if isinstance(final_answer, AgentText):
 
 
160
  yield gr.ChatMessage(
161
  role="assistant",
162
- content=f"**Final answer:**\n{final_answer.to_string()}\n",
163
  )
164
  elif isinstance(final_answer, AgentImage):
165
  yield gr.ChatMessage(
 
157
  final_answer = handle_agent_output_types(final_answer)
158
 
159
  if isinstance(final_answer, AgentText):
160
+ formatted_final_answer = f"**Query:** {task}\n\n**Final answer:**\n{final_answer.to_string()}\n"
161
+
162
  yield gr.ChatMessage(
163
  role="assistant",
164
+ content=f"**Final answer:**\n{formatted_final_answer.to_string()}\n",
165
  )
166
  elif isinstance(final_answer, AgentImage):
167
  yield gr.ChatMessage(