Update app.py

app.py

@@ -181,47 +181,42 @@ 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 = []
-    if "tech_lightbox" not in st.session_state:
-        st.session_state.tech_lightbox = None
+    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:
-            # Always create a new thread for each new query
-            thread = client.beta.threads.create()
-            tech_thread_id = thread.id
+            if st.session_state.tech_thread_id is None:
+                thread = client.beta.threads.create()
+                st.session_state.tech_thread_id = thread.id
 
-            # Send user message
             client.beta.threads.messages.create(
-                thread_id=tech_thread_id,
+                thread_id=st.session_state.tech_thread_id,
                 role="user",
-                content=tech_input
+                content=st.session_state.tech_messages[-1]["content"]
             )
 
-            # Run assistant
             run = client.beta.threads.runs.create(
-                thread_id=tech_thread_id,
+                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=tech_thread_id,
+                        thread_id=st.session_state.tech_thread_id,
                         run_id=run.id
                     )
                     if run_status.status in ("completed", "failed", "cancelled"):
@@ -229,29 +224,25 @@ with tab2:
                     time.sleep(1)
 
             if run_status.status == "completed":
-                messages = client.beta.threads.messages.list(thread_id=tech_thread_id)
+                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})
-
-                        # Parse JSON from assistant response
                         try:
-                            st.session_state.tech_results = json.loads(content.strip("json ").strip())
-                        except Exception as e:
-                            st.warning(f"⚠️ Failed to parse drawing data: {e}")
+                            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}")
 
-    # 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]
@@ -272,7 +263,6 @@ 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)