Update app.py

app.py

@@ -55,7 +55,6 @@ tab1, tab2 = st.tabs(["📑 Contract", "📁 Technical"])
 with tab1:
     ASSISTANT_ID = "asst_KsQRedoJUnEeStzfox1o06lO"
 
-    # Correct session state initialization
     if "messages" not in st.session_state:
         st.session_state.messages = []
     if "thread_id" not in st.session_state:
@@ -67,7 +66,6 @@ with tab1:
     if "pending_prompt" not in st.session_state:
         st.session_state.pending_prompt = None
 
-    # Sidebar tools
     with st.sidebar:
         st.header("ℹ️ Contract Tools")
         if st.button("🧹 Clear Chat"):
@@ -79,7 +77,6 @@ with tab1:
             st.rerun()
 
         show_image = st.toggle("📑 Show Page Image", value=True)
-
         keyword = st.text_input("Search by Keyword", placeholder="e.g. defects, WHS, delay")
         if st.button("🔎 Search Keyword") and keyword:
             st.session_state.pending_prompt = f"Find clauses or references related to: {keyword}"
@@ -112,7 +109,6 @@ with tab1:
         if action != actions[0]:
             st.session_state.pending_prompt = action
 
-    # Chat + Image layout
     chat_col, image_col = st.columns([2, 1])
     with chat_col:
         st.markdown("### 🧠 Ask a Document-Specific Question")
@@ -230,7 +226,7 @@ with tab2:
                         content = msg.content[0].text.value
                         st.session_state.tech_messages.append({"role": "assistant", "content": content})
                         try:
-                            st.session_state.tech_results = json.loads(content.strip("json "))
+                            st.session_state.tech_results = json.loads(content.strip("`json "))
                         except:
                             st.session_state.tech_results = []
                         break
@@ -252,10 +248,20 @@ with tab2:
             with cols[i % 4]:
                 st.markdown(f"**📁 {item['drawing_number']} ({item['discipline']})**")
                 st.caption(item.get("summary", ""))
-                for url in item.get("images", [])[:1]:
+
+                image_urls = item.get("images", [])
+                if not image_urls:
+                    st.warning("⚠️ No image available.")
+                else:
+                    url = image_urls[0]
+                    st.caption(f"🔗 Image URL: {url}")
+                    try:
+                        st.image(url, caption=f"{item['drawing_number']} - Page 1", use_container_width=True)
+                    except Exception as e:
+                        st.error(f"❌ Could not load image: {e}")
+
                     if st.button("🖼️ View Drawing Details", key=f"thumb_{i}"):
                         st.session_state.tech_lightbox = url
-                    st.image(url, caption=f"{item['drawing_number']} - Page 1", use_container_width=True)
 
         if st.session_state.tech_lightbox:
             st.image(st.session_state.tech_lightbox, caption="🔍 Enlarged Drawing Preview", use_container_width=True)
@@ -265,4 +271,4 @@ with tab2:
     else:
         for msg in st.session_state.tech_messages:
             with st.chat_message(msg["role"]):
-                st.markdown(msg["content"], unsafe_allow_html=True) 
+                st.markdown(msg["content"], unsafe_allow_html=True)