Update app.py

app.py

@@ -54,8 +54,10 @@ def display_result(result):
 
     if 'images' in result:
         st.write("### Images")
-        for image in result['images']:
-            st.image(image['url'], caption=image['description'])
+        for idx, image in enumerate(result['images']):
+            img_path = f"image_{idx}.png"
+            WebSearchTools.download_image(image, img_path)
+            st.image(img_path, caption=f"Image {idx+1}")
 
     if 'content' in result:
         st.markdown(result['content'])