Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+ import streamlit as st
 import os
 import time
 import re
@@ -55,6 +55,7 @@ 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:
@@ -66,6 +67,7 @@ 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"):
@@ -110,6 +112,7 @@ 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")
@@ -179,43 +182,21 @@ with tab1:
                 st.error(f"🖼️ Image failed: {e}")
 
 # ------------------ Technical Tab ------------------
-
 with tab2:
-    with st.sidebar:
-        st.header("🛠️ Technical Tools")
-        if st.button("🧹 Clear Technical Chat"):
-            st.session_state.tech_messages = []
-            st.session_state.tech_thread_id = None
-            st.session_state.tech_results = []
-            st.session_state.tech_lightbox = None
-            st.session_state.tech_new_input = False
-            st.rerun()
-
     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_last_input" not in st.session_state:
-    st.session_state.tech_last_input = None
         st.session_state.tech_results = []
-    if "tech_lightbox" not in st.session_state:
-        st.session_state.tech_lightbox = None
-    if "tech_new_input" not in st.session_state:
-        st.session_state.tech_new_input = False
+    st.session_state.tech_lightbox = None
 
     tech_input = st.chat_input("Ask about plans, drawings or components")
     if tech_input:
-    st.session_state.tech_last_input = tech_input
         st.session_state.tech_messages.append({"role": "user", "content": tech_input})
-        st.session_state.tech_new_input = True
 
-    if st.session_state.tech_new_input:
-        # Always start fresh thread for each query
-        thread = client.beta.threads.create()
-        st.session_state.tech_thread_id = thread.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()
@@ -234,7 +215,10 @@ if "tech_last_input" not in st.session_state:
 
             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)
+                    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)
@@ -245,24 +229,13 @@ if "tech_last_input" not in st.session_state:
                     if msg.role == "assistant":
                         content = msg.content[0].text.value
                         st.session_state.tech_messages.append({"role": "assistant", "content": content})
-
-                        # Always clear previous results
-                        st.session_state.tech_results = []
-
                         try:
-                            parsed = json.loads(content.strip("`json ").strip())
-                            if isinstance(parsed, list):
-                                st.session_state.tech_results = parsed
-                            else:
-                                st.warning("Assistant response was not a valid list of results.")
-                        except Exception as parse_error:
-                            st.warning("⚠️ Failed to parse assistant response as JSON.")
-                            st.code(content, language="markdown")
+                            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}")
-        finally:
-            st.session_state.tech_new_input = False
 
     with st.expander("🔧 Options (Filter + Pagination)", expanded=False):
         disciplines = sorted(set(d.get("discipline", "") for d in st.session_state.tech_results))
@@ -270,9 +243,6 @@ if "tech_last_input" not in st.session_state:
         page_size = 8
         page = st.number_input("Page", min_value=1, step=1, value=1)
 
-    if st.session_state.get("tech_last_input"):
-        st.markdown(f"#### 🧑 You asked: `{st.session_state.tech_last_input}`")
-
     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]
@@ -295,5 +265,4 @@ if "tech_last_input" not in st.session_state:
     else:
         for msg in st.session_state.tech_messages:
             with st.chat_message(msg["role"]):
-                st.markdown(msg["content"], unsafe_allow_html=True)
-        st.info("No drawing results found for the last query. Try rephrasing or narrowing the search.")
+                st.markdown(msg["content"], unsafe_allow_html=True)