Update app.py

app.py

@@ -76,15 +76,25 @@ if st.session_state.tech_messages and st.session_state.tech_messages[-1]["role"]
 
                             if isinstance(results, list):
                                 for item in results:
-                                    st.markdown(f"### 📄 {item.get('drawing_number')} ({item.get('discipline')})")
-                                    st.markdown(f"**Summary:** {item.get('summary')}")
-
-                                    if "image" in item:
-                                        st.image(item["image"], use_column_width=True)
-
-                                    elif "images" in item:
-                                        for img in item["images"]:
-                                            st.image(img, use_column_width=True)
+                                    drawing_number = item.get("drawing_number")
+                                    discipline = item.get("discipline")
+                                    summary = item.get("summary")
+                                    images = item.get("images", [])
+                                    single_image = item.get("image")
+                                    question = item.get("question", None)
+
+                                    with st.container(border=True):
+                                        st.markdown(f"### 🗂️ {drawing_number} ({discipline})")
+                                        st.markdown(f"**Summary:** {summary}")
+
+                                        with st.expander("📂 View Drawing Details"):
+                                            if question:
+                                                st.markdown(f"**Question Match:** {question}")
+                                            if single_image:
+                                                st.image(single_image, caption=drawing_number)
+                                            elif images:
+                                                for i, img in enumerate(images, start=1):
+                                                    st.image(img, caption=f"{drawing_number} – Page {i}")
                         except Exception as parse_error:
                             st.warning("🟡 Could not parse assistant response as JSON.")
                         break