Spaces:
Sleeping
Sleeping
Update Gradio_UI.py
Browse files- Gradio_UI.py +7 -3
Gradio_UI.py
CHANGED
|
@@ -155,11 +155,13 @@ 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=
|
| 163 |
)
|
| 164 |
elif isinstance(final_answer, AgentImage):
|
| 165 |
yield gr.ChatMessage(
|
|
@@ -172,7 +174,9 @@ def stream_to_gradio(
|
|
| 172 |
content={"path": final_answer.to_string(), "mime_type": "audio/wav"},
|
| 173 |
)
|
| 174 |
else:
|
| 175 |
-
|
|
|
|
|
|
|
| 176 |
|
| 177 |
|
| 178 |
class GradioUI:
|
|
|
|
| 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 |
+
# Strip any wrapper or formatting for exact match evaluation
|
| 160 |
if isinstance(final_answer, AgentText):
|
| 161 |
+
content = final_answer.to_string().strip()
|
| 162 |
yield gr.ChatMessage(
|
| 163 |
role="assistant",
|
| 164 |
+
content=content,
|
| 165 |
)
|
| 166 |
elif isinstance(final_answer, AgentImage):
|
| 167 |
yield gr.ChatMessage(
|
|
|
|
| 174 |
content={"path": final_answer.to_string(), "mime_type": "audio/wav"},
|
| 175 |
)
|
| 176 |
else:
|
| 177 |
+
# Fall back to raw string with no prefix
|
| 178 |
+
yield gr.ChatMessage(role="assistant", content=str(final_answer).strip())
|
| 179 |
+
|
| 180 |
|
| 181 |
|
| 182 |
class GradioUI:
|