Update app.py

app.py

@@ -181,42 +181,47 @@ with tab1:
             except Exception as e:
                 st.error(f"🖼️ Image failed: {e}")
 
+# ------------------ Technical Tab ------------------
 # ------------------ Technical Tab ------------------
 with tab2:
     ASSISTANT_ID = "asst_DjvuWBc7tCvMbAhY7n1em4BZ"
+
     if "tech_messages" not in st.session_state:
         st.session_state.tech_messages = []
-    if "tech_thread_id" not in st.session_state:
-        st.session_state.tech_thread_id = None
     if "tech_results" not in st.session_state:
         st.session_state.tech_results = []
-    st.session_state.tech_lightbox = None
+    if "tech_lightbox" not in st.session_state:
+        st.session_state.tech_lightbox = None
 
     tech_input = st.chat_input("Ask about plans, drawings or components")
+
     if tech_input:
+        # Clear old thread/results each time
         st.session_state.tech_messages.append({"role": "user", "content": tech_input})
+        st.session_state.tech_results = []
 
-    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
+            # Always create a new thread for each new query
+            thread = client.beta.threads.create()
+            tech_thread_id = thread.id
 
+            # Send user message
             client.beta.threads.messages.create(
-                thread_id=st.session_state.tech_thread_id,
+                thread_id=tech_thread_id,
                 role="user",
-                content=st.session_state.tech_messages[-1]["content"]
+                content=tech_input
             )
 
+            # Run assistant
             run = client.beta.threads.runs.create(
-                thread_id=st.session_state.tech_thread_id,
+                thread_id=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,
+                        thread_id=tech_thread_id,
                         run_id=run.id
                     )
                     if run_status.status in ("completed", "failed", "cancelled"):
@@ -224,25 +229,29 @@ with tab2:
                     time.sleep(1)
 
             if run_status.status == "completed":
-                messages = client.beta.threads.messages.list(thread_id=st.session_state.tech_thread_id)
+                messages = client.beta.threads.messages.list(thread_id=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})
+
+                        # Parse JSON from assistant response
                         try:
-                            st.session_state.tech_results = json.loads(content.strip("json "))
-                        except:
-                            st.session_state.tech_results = []
+                            st.session_state.tech_results = json.loads(content.strip("json ").strip())
+                        except Exception as e:
+                            st.warning(f"⚠️ Failed to parse drawing data: {e}")
                         break
         except Exception as e:
             st.error(f"❌ Technical Assistant Error: {e}")
 
+    # Expander for filtering and pagination
     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)
 
+    # Display results
     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]
@@ -263,6 +272,7 @@ with tab2:
                 st.session_state.tech_lightbox = None
                 st.rerun()
     else:
+        # Only show chat messages if no results
         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)