Update app.py

app.py

@@ -55,7 +55,6 @@ 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:
@@ -67,7 +66,6 @@ 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"):
@@ -112,7 +110,6 @@ 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")
@@ -184,26 +181,22 @@ with tab1:
 # ------------------ Technical Tab ------------------
 with tab2:
     ASSISTANT_ID = "asst_DjvuWBc7tCvMbAhY7n1em4BZ"
-
-    # Initialize session state variables
     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_new_input" not in st.session_state:
-        st.session_state.tech_new_input = False
     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
 
-    # Input box
     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})
-        st.session_state.tech_new_input = True  # Flag to process this new input
+        st.session_state.tech_new_input = True
 
-    # Run assistant only if new input was entered
     if st.session_state.tech_new_input:
         try:
             if st.session_state.tech_thread_id is None:
@@ -223,10 +216,7 @@ with tab2:
 
             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)
@@ -242,21 +232,17 @@ with tab2:
                         except:
                             st.session_state.tech_results = []
                         break
-            else:
-                st.error("❌ Assistant run failed.")
         except Exception as e:
             st.error(f"❌ Technical Assistant Error: {e}")
         finally:
-            st.session_state.tech_new_input = False  # Reset the flag
+            st.session_state.tech_new_input = False
 
-    # Filtering & Pagination UI
     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)
 
-    # Results display
     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]
@@ -279,4 +265,4 @@ with tab2:
     else:
         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)