Update app.py

app.py

@@ -254,7 +254,7 @@ with tab2:
                 st.error(f"🖼️ Image failed: {e}")
 
 
-# ------------------ Tab 3: Visual Queries (Technical Assistant) ------------------
+#------------------ Technical Tab ------------------
 with tab3:
     ASSISTANT_ID = "asst_DjvuWBc7tCvMbAhY7n1em4BZ"
     if "tech_messages" not in st.session_state:
@@ -265,87 +265,89 @@ with tab3:
         st.session_state.tech_results = []
     st.session_state.tech_lightbox = None
 
-    tech_input = st.chat_input("Ask about plans, drawings or components")
-    if tech_input:
-        st.session_state.tech_messages.append({"role": "user", "content": tech_input})
 
-    if st.session_state.tech_messages and st.session_state.tech_messages[-1]["role"] == "user":
-        try:
-            if st.session_state.tech_thread_id is None:
-                thread = client.beta.threads.create()
-                st.session_state.tech_thread_id = thread.id
 
-            client.beta.threads.messages.create(
-                thread_id=st.session_state.tech_thread_id,
-                role="user",
-                content=st.session_state.tech_messages[-1]["content"]
-            )
+tech_input = st.chat_input("Ask about plans, drawings or components")
+if tech_input:
+    st.session_state.tech_messages.append({"role": "user", "content": tech_input})
 
-            run = client.beta.threads.runs.create(
-                thread_id=st.session_state.tech_thread_id,
-                assistant_id=ASSISTANT_ID
-            )
+if st.session_state.tech_messages and st.session_state.tech_messages[-1]["role"] == "user":
+    try:
+        if st.session_state.tech_thread_id is None:
+            thread = client.beta.threads.create()
+            st.session_state.tech_thread_id = thread.id
 
-            with st.spinner("🔍 Searching technical drawings..."):
-                while True:
-                    run_status = client.beta.threads.runs.retrieve(
-                        thread_id=st.session_state.tech_thread_id,
-                        run_id=run.id
-                    )
-                    if run_status.status in ("completed", "failed", "cancelled"):
-                        break
-                    time.sleep(1)
+        client.beta.threads.messages.create(
+            thread_id=st.session_state.tech_thread_id,
+            role="user",
+            content=st.session_state.tech_messages[-1]["content"]
+        )
 
-            if run_status.status == "completed":
-                messages = client.beta.threads.messages.list(thread_id=st.session_state.tech_thread_id)
-                for msg in reversed(messages.data):
-                    if msg.role == "assistant":
-                        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 "))
-                        except:
-                            st.session_state.tech_results = []
-                        break
-        except Exception as e:
-            st.error(f"❌ Technical Assistant Error: {e}")
-
-    with st.expander("🔧 Options (Filter + Pagination)", expanded=False):
-        disciplines = sorted(set(d.get("discipline", "") for d in st.session_state.tech_results))
-        selected = st.selectbox("🌍 Filter by discipline", ["All"] + disciplines)
-        page_size = 8
-        page = st.number_input("Page", min_value=1, step=1, value=1)
-
-    if st.session_state.tech_results:
-        st.subheader("📂 Results")
-        results = [r for r in st.session_state.tech_results if selected == "All" or r.get("discipline") == selected]
-        paged = results[(page - 1) * page_size : page * page_size]
-        cols = st.columns(4)
-        for i, item in enumerate(paged):
-            with cols[i % 4]:
-                st.markdown(f"**📁 {item['drawing_number']} ({item['discipline']})**")
-                st.caption(item.get("summary", ""))
-
-                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}")
+        run = client.beta.threads.runs.create(
+            thread_id=st.session_state.tech_thread_id,
+            assistant_id=ASSISTANT_ID
+        )
 
-                    if st.button("🖼️ View Drawing Details", key=f"thumb_{i}"):
-                        st.session_state.tech_lightbox = url
-
-        if st.session_state.tech_lightbox:
-            st.image(st.session_state.tech_lightbox, caption="🔍 Enlarged Drawing Preview", use_container_width=True)
-            if st.button("❌ Close Preview"):
-                st.session_state.tech_lightbox = None
-                st.rerun()
-    else:
-        for msg in st.session_state.tech_messages:
-            with st.chat_message(msg["role"]):
-                st.markdown(msg["content"], unsafe_allow_html=True)
+        with st.spinner("🔍 Searching technical drawings..."):
+            while True:
+                run_status = client.beta.threads.runs.retrieve(
+                    thread_id=st.session_state.tech_thread_id,
+                    run_id=run.id
+                )
+                if run_status.status in ("completed", "failed", "cancelled"):
+                    break
+                time.sleep(1)
+
+        if run_status.status == "completed":
+            messages = client.beta.threads.messages.list(thread_id=st.session_state.tech_thread_id)
+            for msg in reversed(messages.data):
+                if msg.role == "assistant":
+                    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 "))
+                    except:
+                        st.session_state.tech_results = []
+                    break
+    except Exception as e:
+        st.error(f"❌ Technical Assistant Error: {e}")
+
+with st.expander("🔧 Options (Filter + Pagination)", expanded=False):
+    disciplines = sorted(set(d.get("discipline", "") for d in st.session_state.tech_results))
+    selected = st.selectbox("🌍 Filter by discipline", ["All"] + disciplines)
+    page_size = 8
+    page = st.number_input("Page", min_value=1, step=1, value=1)
+
+if st.session_state.tech_results:
+    st.subheader("📂 Results")
+    results = [r for r in st.session_state.tech_results if selected == "All" or r.get("discipline") == selected]
+    paged = results[(page - 1) * page_size : page * page_size]
+    cols = st.columns(4)
+    for i, item in enumerate(paged):
+        with cols[i % 4]:
+            st.markdown(f"**📁 {item['drawing_number']} ({item['discipline']})**")
+            st.caption(item.get("summary", ""))
+
+            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
+
+    if st.session_state.tech_lightbox:
+        st.image(st.session_state.tech_lightbox, caption="🔍 Enlarged Drawing Preview", use_container_width=True)
+        if st.button("❌ Close Preview"):
+            st.session_state.tech_lightbox = None
+            st.rerun()
+else:
+    for msg in st.session_state.tech_messages:
+        with st.chat_message(msg["role"]):
+            st.markdown(msg["content"], unsafe_allow_html=True)