Update app.py

app.py

@@ -36,8 +36,8 @@ if "search_history" not in st.session_state:
 if "run_in_progress" not in st.session_state:
     st.session_state.run_in_progress = False
 
-# ------------------ Sidebar ------------------
-st.sidebar.header("⚙️ Settings")
+# ------------------ Sidebar (Left Utility Panel) ------------------
+st.sidebar.header("ℹ️ Information")
 if st.sidebar.button("🧹 Clear Chat"):
     st.session_state.messages = []
     st.session_state.thread_id = None
@@ -49,113 +49,106 @@ if st.sidebar.button("🧹 Clear Chat"):
 
 show_image = st.sidebar.toggle("📑 Show Page Image", value=True)
 
-# ------------------ Horizontal Layout ------------------
-left, center = st.columns([1, 2])
-
-# ------------------ Expandable Left Section ------------------
-with left.expander("📂 Navigate or Query Document", expanded=False):
-    keyword = st.text_input("Jump to clause keyword or term", placeholder="e.g. defects, payment, WHS")
-    if st.button("🔎 Search Keyword") and keyword:
-        st.session_state.search_history.append(keyword)
-        if not st.session_state.run_in_progress:
-            st.session_state.messages.append({"role": "user", "content": f"Find clauses or references related to: {keyword}"})
-
-    section_options = [
-        "Select a section...",
-        "1. Formal Instrument of Contract",
-        "2. Offer and Acceptance",
-        "3. Key Personnel",
-        "4. Contract Pricing",
-        "5. Contract Specifications",
-        "6. WHS Policies",
-        "7. Penalties and Delays",
-        "8. Dispute Resolution",
-        "9. Principal Obligations"
-    ]
-    selected_section = st.selectbox("📄 Browse Sections", section_options)
-    if selected_section and selected_section != section_options[0]:
-        st.session_state.messages.append({"role": "user", "content": f"Summarize or list key points from section: {selected_section}"})
-
-    actions = [
-        "Select an action...",
-        "List all contractual obligations",
-        "Summarize payment terms",
-        "List WHS responsibilities",
-        "Find delay-related penalties",
-        "Extract dispute resolution steps"
-    ]
-    selected_action = st.selectbox("⚙️ Common Actions", actions)
-    if selected_action and selected_action != actions[0]:
-        st.session_state.messages.append({"role": "user", "content": selected_action})
-
-    if show_image and st.session_state.image_url:
-        st.markdown("---")
-        st.markdown("### 📄 Page Preview")
-        try:
-            response = requests.get(st.session_state.image_url)
-            response.raise_for_status()
-            img = Image.open(BytesIO(response.content))
-            st.image(img, caption="📄 OCR Page Image", use_container_width=True)
-        except Exception as e:
-            st.error(f"❗ Failed to load image: {e}")
-
-    if st.session_state.search_history:
-        st.markdown("---")
-        st.markdown("#### 🔁 Recent Searches")
-        for term in reversed(st.session_state.search_history[-5:]):
-            st.markdown(f"- {term}")
-
-# ------------------ Chat Section ------------------
-with center:
-    st.markdown("### 🧠 Ask a Document-Specific Question")
-    prompt = st.chat_input("Example: What is the defects liability period?")
-
-    for msg in st.session_state.messages:
-        with st.chat_message(msg["role"]):
-            st.markdown(msg["content"], unsafe_allow_html=True)
-
-    if prompt and not st.session_state.run_in_progress:
-        st.session_state.messages.append({"role": "user", "content": prompt})
-        st.session_state.run_in_progress = True
-
-        try:
-            if st.session_state.thread_id is None:
-                thread = client.beta.threads.create()
-                st.session_state.thread_id = thread.id
-
-            client.beta.threads.messages.create(thread_id=st.session_state.thread_id, role="user", content=prompt)
-
-            run = client.beta.threads.runs.create(thread_id=st.session_state.thread_id, assistant_id=ASSISTANT_ID)
-
-            with st.spinner("🤖 Parsing and responding with referenced content..."):
-                while True:
-                    run_status = client.beta.threads.runs.retrieve(thread_id=st.session_state.thread_id, run_id=run.id)
-                    if run_status.status in ("completed", "failed", "cancelled"):
-                        break
-                    time.sleep(1)
-
-            if run_status.status != "completed":
-                st.error(f"⚠️ Assistant failed: {run_status.status}")
-            else:
-                messages = client.beta.threads.messages.list(thread_id=st.session_state.thread_id)
-                for message in reversed(messages.data):
-                    if message.role == "assistant":
-                        assistant_message = message.content[0].text.value
-                        st.session_state.messages.append({"role": "assistant", "content": assistant_message})
-                        break
-
-                match = re.search(r'Document Reference:\s+(.+?),\s+Page\s+(\d+)', assistant_message)
-                if match:
-                    doc_name = match.group(1).strip()
-                    page = int(match.group(2))
-                    page_str = f"{page:04d}"
-                    image_url = f"https://raw.githubusercontent.com/AndrewLORTech/c2ozschlaegerforrestdale/main/{doc_name}/{doc_name}_page_{page_str}.png"
-                    st.session_state.image_url = image_url
-                    st.session_state.image_updated = True
-
-            st.session_state.run_in_progress = False
-            st.rerun()
-
-        except Exception as e:
-            st.session_state.run_in_progress = False
-            st.error(f"❌ Error: {e}")
+st.sidebar.markdown("### 🔍 Navigate or Query Document")
+keyword = st.sidebar.text_input("Jump to clause keyword or term", placeholder="e.g. defects, payment, WHS")
+if st.sidebar.button("🔎 Search Keyword") and keyword:
+    st.session_state.search_history.append(keyword)
+    if not st.session_state.run_in_progress:
+        st.session_state.messages.append({"role": "user", "content": f"Find clauses or references related to: {keyword}"})
+
+section_options = [
+    "Select a section...",
+    "1. Formal Instrument of Contract",
+    "2. Offer and Acceptance",
+    "3. Key Personnel",
+    "4. Contract Pricing",
+    "5. Contract Specifications",
+    "6. WHS Policies",
+    "7. Penalties and Delays",
+    "8. Dispute Resolution",
+    "9. Principal Obligations"
+]
+selected_section = st.sidebar.selectbox("📄 Browse Sections", section_options)
+if selected_section and selected_section != section_options[0]:
+    st.session_state.messages.append({"role": "user", "content": f"Summarize or list key points from section: {selected_section}"})
+
+actions = [
+    "Select an action...",
+    "List all contractual obligations",
+    "Summarize payment terms",
+    "List WHS responsibilities",
+    "Find delay-related penalties",
+    "Extract dispute resolution steps"
+]
+selected_action = st.sidebar.selectbox("⚙️ Common Actions", actions)
+if selected_action and selected_action != actions[0]:
+    st.session_state.messages.append({"role": "user", "content": selected_action})
+
+if st.session_state.search_history:
+    st.sidebar.markdown("---")
+    st.sidebar.markdown("#### 🔁 Recent Searches")
+    for term in reversed(st.session_state.search_history[-5:]):
+        st.sidebar.markdown(f"- {term}")
+
+if show_image and st.session_state.image_url:
+    st.sidebar.markdown("---")
+    st.sidebar.markdown("#### 🖼️ Page Image (Toggle)")
+    try:
+        response = requests.get(st.session_state.image_url)
+        response.raise_for_status()
+        img = Image.open(BytesIO(response.content))
+        st.sidebar.image(img, caption="OCR Page", use_container_width=True)
+    except Exception as e:
+        st.sidebar.error(f"❗ Failed to load image: {e}")
+
+# ------------------ Center Panel: Chat ------------------
+st.markdown("### 🧠 Ask a Document-Specific Question")
+prompt = st.chat_input("Example: What is the defects liability period?")
+
+for msg in st.session_state.messages:
+    with st.chat_message(msg["role"]):
+        st.markdown(msg["content"], unsafe_allow_html=True)
+
+if prompt and not st.session_state.run_in_progress:
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    st.session_state.run_in_progress = True
+
+    try:
+        if st.session_state.thread_id is None:
+            thread = client.beta.threads.create()
+            st.session_state.thread_id = thread.id
+
+        client.beta.threads.messages.create(thread_id=st.session_state.thread_id, role="user", content=prompt)
+        run = client.beta.threads.runs.create(thread_id=st.session_state.thread_id, assistant_id=ASSISTANT_ID)
+
+        with st.spinner("🤖 Parsing and responding with referenced content..."):
+            while True:
+                run_status = client.beta.threads.runs.retrieve(thread_id=st.session_state.thread_id, run_id=run.id)
+                if run_status.status in ("completed", "failed", "cancelled"):
+                    break
+                time.sleep(1)
+
+        if run_status.status != "completed":
+            st.error(f"⚠️ Assistant failed: {run_status.status}")
+        else:
+            messages = client.beta.threads.messages.list(thread_id=st.session_state.thread_id)
+            for message in reversed(messages.data):
+                if message.role == "assistant":
+                    assistant_message = message.content[0].text.value
+                    st.session_state.messages.append({"role": "assistant", "content": assistant_message})
+                    break
+
+            match = re.search(r'Document Reference:\s+(.+?),\s+Page\s+(\d+)', assistant_message)
+            if match:
+                doc_name = match.group(1).strip()
+                page = int(match.group(2))
+                page_str = f"{page:04d}"
+                image_url = f"https://raw.githubusercontent.com/AndrewLORTech/c2ozschlaegerforrestdale/main/{doc_name}/{doc_name}_page_{page_str}.png"
+                st.session_state.image_url = image_url
+                st.session_state.image_updated = True
+
+        st.session_state.run_in_progress = False
+        st.rerun()
+    except Exception as e:
+        st.session_state.run_in_progress = False
+        st.error(f"❌ Error: {e}")