Update app.py

app.py

@@ -5,7 +5,7 @@ import json
 import re
 from openai import OpenAI
 
-# Configuration
+# Basic config
 st.set_page_config(page_title="Forrestdale Drawing Viewer", layout="wide")
 st.title("📐 Forrestdale Technical Drawing Assistant")
 
@@ -25,17 +25,15 @@ if "tech_thread_id" not in st.session_state:
 if "tech_messages" not in st.session_state:
     st.session_state.tech_messages = []
 
-# Prompt input
 prompt = st.chat_input("Ask about plans, drawings or components (e.g. Show me all electrical plans)")
 if prompt:
     st.session_state.tech_messages.append({"role": "user", "content": prompt})
 
-# Display assistant messages
 for msg in st.session_state.tech_messages:
     with st.chat_message(msg["role"]):
         st.markdown(msg["content"])
 
-# Trigger assistant
+# Fetch assistant reply
 if st.session_state.tech_messages and st.session_state.tech_messages[-1]["role"] == "user":
     with st.spinner("⏳ Fetching results from assistant..."):
         try:
@@ -68,33 +66,28 @@ if st.session_state.tech_messages and st.session_state.tech_messages[-1]["role"]
                         reply_content = message.content[0].text.value.strip()
                         st.session_state.tech_messages.append({"role": "assistant", "content": reply_content})
 
-                        # Try to parse response as JSON block (```json ... ```)
                         try:
                             match = re.search(r"```json\s*(.*?)```", reply_content, re.DOTALL)
                             json_str = match.group(1) if match else reply_content
                             results = json.loads(json_str)
 
                             if isinstance(results, list):
-                                for item in results:
-                                    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}")
+                                cols = st.columns(4)
+                                for idx, item in enumerate(results):
+                                    with cols[idx % 4]:
+                                        with st.container(border=True):
+                                            st.markdown(f"### 📁 {item.get('drawing_number')} ({item.get('discipline')})", help=item.get("summary"))
+                                            st.caption(item.get("summary"))
+
+                                            with st.expander("📂 View Drawing Details"):
+                                                if item.get("question"):
+                                                    st.markdown(f"**Question Match:** {item.get('question')}")
+                                                if item.get("image"):
+                                                    st.image(item.get("image"), caption=item.get("drawing_number"))
+                                                elif item.get("images"):
+                                                    for i, img in enumerate(item["images"]):
+                                                        if img.startswith("http"):
+                                                            st.image(img, caption=f"{item['drawing_number']} – Page {i+1}")
                         except Exception as parse_error:
                             st.warning("🟡 Could not parse assistant response as JSON.")
                         break