Update app.py

app.py

@@ -254,7 +254,7 @@ with tab2:
                 st.error(f"🖼️ Image failed: {e}")
 
 
-#------------------ Technical Tab ------------------
+# ------------------ Technical Tab ------------------
 with tab3:
     ASSISTANT_ID = "asst_DjvuWBc7tCvMbAhY7n1em4BZ"
     if "tech_messages" not in st.session_state:
@@ -265,89 +265,88 @@ with tab3:
         st.session_state.tech_results = []
     st.session_state.tech_lightbox = None
 
+    # ✅ Input moved inside tab3 block
+    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
 
-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"]
+            )
 
-        client.beta.threads.messages.create(
-            thread_id=st.session_state.tech_thread_id,
-            role="user",
-            content=st.session_state.tech_messages[-1]["content"]
-        )
+            run = client.beta.threads.runs.create(
+                thread_id=st.session_state.tech_thread_id,
+                assistant_id=ASSISTANT_ID
+            )
 
-        run = client.beta.threads.runs.create(
-            thread_id=st.session_state.tech_thread_id,
-            assistant_id=ASSISTANT_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)
 
-        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})
+            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.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)
+                        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)