Update app.py

app.py

@@ -318,56 +318,25 @@ async def run_query(query: str):
             # Start the handler
             handler = web_agent.run(query, ctx=ctx)
             
-            # Stream content
+            # Just stream the text content naturally, with minimal additions
             async for event in handler.stream_events():
-                # Add some debugging info to understand event structure
-                # print(f"Event type: {type(event)}")
-                # print(f"Event attrs: {dir(event)}")
-                
                 if isinstance(event, AgentStream):
-                    # This is the text being generated
+                    # This is the text being generated - pass it through directly
                     if hasattr(event, 'delta') and event.delta:
                         yield event.delta
                 
+                # We can optionally add minimal annotations for tool calls and results
+                # But we'll keep it simple to avoid interfering with the agent's natural output
                 elif isinstance(event, ToolCall):
-                    # Handle ToolCall differently - check available attributes
-                    tool_name = "unknown tool"
-                    
-                    # Try different possible attribute locations
-                    if hasattr(event, 'name'):
-                        tool_name = event.name
-                    elif hasattr(event, 'function_name'):
-                        tool_name = event.function_name
-                    elif hasattr(event, 'tool_name'):
-                        tool_name = event.tool_name
-                    
-                    yield f"\n\n🔧 Using tool: {tool_name}...\n"
+                    # We could optionally add a subtle marker here
+                    pass  # Or comment this out if you want nothing added
                 
                 elif isinstance(event, ToolCallResult):
-                    # Handle ToolCallResult - check available attributes
-                    result = "Result not available"
-                    
-                    # Try different possible attribute locations
-                    if hasattr(event, 'result'):
-                        result = event.result
-                    elif hasattr(event, 'output'):
-                        result = event.output
-                    elif hasattr(event, 'data') and hasattr(event.data, 'get'):
-                        result = event.data.get("result", "")
-                    
-                    # Truncate long results for display
-                    if isinstance(result, str) and len(result) > 200:
-                        result_preview = result[:200] + "... (truncated)"
-                    else:
-                        result_preview = str(result)
-                    
-                    yield f"\n📊 Got result from tool\n"
+                    # We could optionally add a subtle marker here
+                    pass  # Or comment this out if you want nothing added
             
     except Exception as e:
         yield f"\n\n❌ Error: {str(e)}\n"
-        # For debugging:
-        import traceback
-        yield f"Traceback: {traceback.format_exc()}"
     finally:
         instrumentor.flush()