Update app.py

app.py

@@ -298,7 +298,8 @@ class SearchAgentUI:
                                     elif message.name == "read_text":
                                         st.expander(f"📝 טקסט: {message.content}")
                                     
-                elif message.type == "ai" and message.content:                                   
+                elif message.type == "ai": 
+                    if  message.content:
                         with st.chat_message(message.type):
                             if isinstance(message.content, list):
                                 for item in message.content:                                   
@@ -308,8 +309,10 @@ class SearchAgentUI:
                                 
                     for tool_call in message.tool_calls:
                         with st.expander(f"🛠️ שימוש בכלי: {tool_call['name']}"):
-                            st.json(tool_call["args"])                                
-                else: 
+                            st.json(tool_call["args"])     
+
+                            
+                else: # user message
                     with st.chat_message(message.type):
                         st.write(message.content)