Lar159 commited on
Commit
e848884
·
verified ·
1 Parent(s): 8c5c24b

Update Gradio_UI.py

Browse files
Files changed (1) hide show
  1. Gradio_UI.py +6 -3
Gradio_UI.py CHANGED
@@ -155,12 +155,15 @@ def stream_to_gradio(
155
 
156
  final_answer = step_log # Last log is the run's final_answer
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(
166
  role="assistant",
 
155
 
156
  final_answer = step_log # Last log is the run's final_answer
157
  final_answer = handle_agent_output_types(final_answer)
158
+ # Adicionamos uma verificação para formatar a saída caso seja um dicionário (nosso caso de uso!)
159
+ if isinstance(final_answer, dict):
160
+ # Formata o dicionário como uma string JSON bonita, com indentação e caracteres em português
161
+ json_output = json.dumps(final_answer, indent=2, ensure_ascii=False)
162
  yield gr.ChatMessage(
163
  role="assistant",
164
+ content=f"**Resposta Final:**\n```json\n{json_output}\n```",
165
  )
166
+
167
  elif isinstance(final_answer, AgentImage):
168
  yield gr.ChatMessage(
169
  role="assistant",