Update app.py

app.py

@@ -9,8 +9,8 @@ from openai import OpenAI
 
 # ------------------ App Configuration ------------------
 st.set_page_config(page_title="Schlaeger Forrestdale DocAIA", layout="wide")
-st.title("📄 Document AI Assistant")
-st.caption("Chat with an AI Assistant on your construction documents")
+st.title("📄 Schlaeger Forrestdale Document Assistant")
+st.caption("Chat with construction documents using OCR-based AI")
 
 # ------------------ Load API Key and Assistant ID ------------------
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
@@ -32,113 +32,90 @@ if "image_url" not in st.session_state:
 if "image_updated" not in st.session_state:
     st.session_state.image_updated = False
 
-# ------------------ Sidebar Controls ------------------
-st.sidebar.header("🔧 Settings")
-if st.sidebar.button("🔄 Clear Chat"):
+# ------------------ Sidebar ------------------
+st.sidebar.header("⚙️ Settings")
+if st.sidebar.button("🧹 Clear Chat"):
     st.session_state.messages = []
     st.session_state.thread_id = None
     st.session_state.image_url = None
     st.session_state.image_updated = False
     st.rerun()
 
-show_image = st.sidebar.checkbox("📖 Show Document Image", value=True)
+show_image = st.sidebar.toggle("📑 Show Page Image", value=True)
 
-# ------------------ Layout: Image + Chat ------------------
-col1, col2 = st.columns([1, 2])
+# ------------------ Layout ------------------
+image_col, chat_col = st.columns([1, 2])
 
-# ------------------ Left Panel: Image ------------------
-with col1:
+# ------------------ Image Viewer ------------------
+with image_col:
     if show_image and st.session_state.image_url:
-        with st.spinner("🖼️ Loading image preview..."):
+        with st.spinner("Loading document preview..."):
             try:
                 response = requests.get(st.session_state.image_url)
                 response.raise_for_status()
                 img = Image.open(BytesIO(response.content))
-                st.image(img, caption="📑 Extracted Page", use_container_width=True)
+                st.image(img, caption="📄 OCR Page Image", use_container_width=True)
                 st.session_state.image_updated = False
             except Exception as e:
-                st.warning(f"⚠️ Failed to load image from URL:\n{st.session_state.image_url}\n\nError: {e}")
-
-# ------------------ Right Panel: Chat ------------------
-with col2:
-    prompt = st.chat_input("Type your question about the document or choose from FAQs below...")
-
-    # Display chat history in a cleaner bubble layout
-    paired_messages = []
-    buffer = []
+                st.error(f"❗ Failed to load image: {e}")
+
+# ------------------ Chat Interface ------------------
+with chat_col:
+    st.markdown("### 🧠 Ask a Question or Choose Below")
+    prompt = st.chat_input("What would you like to ask about the construction documents?")
+
+    # Quick Action Buttons
+    st.markdown("#### ⚡ Quick Prompts")
+    quick_cols = st.columns(3)
+    with quick_cols[0]:
+        if st.button("🔍 Summary Mode"):
+            prompt = "Summarize this section."
+    with quick_cols[1]:
+        if st.button("📘 FAQ Mode"):
+            prompt = "Generate FAQs from this section."
+    with quick_cols[2]:
+        if st.button("🧾 Compliance Info"):
+            prompt = "What are the safety and compliance measures?"
+
+    # Display Message History
     for msg in st.session_state.messages:
-        buffer.append(msg)
-        if msg["role"] == "assistant" and len(buffer) == 2:
-            paired_messages.append(buffer.copy())
-            buffer.clear()
-    if buffer:
-        paired_messages.append(buffer.copy())
-
-    for pair in reversed(paired_messages):
-        for msg in pair:
-            with st.chat_message(msg["role"]):
-                st.markdown(f"{msg['content']}", unsafe_allow_html=True)
+        with st.chat_message(msg["role"]):
+            st.markdown(msg["content"])
 
     if prompt:
         st.session_state.messages.append({"role": "user", "content": prompt})
-
         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
-            )
+            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
-            )
+            run = client.beta.threads.runs.create(thread_id=st.session_state.thread_id, assistant_id=ASSISTANT_ID)
 
-            with st.spinner("🤖 Assistant is thinking..."):
+            with st.spinner("🤖 Processing your query..."):
                 while True:
-                    run_status = client.beta.threads.runs.retrieve(
-                        thread_id=st.session_state.thread_id,
-                        run_id=run.id
-                    )
+                    run_status = client.beta.threads.runs.retrieve(thread_id=st.session_state.thread_id, run_id=run.id)
                     if run_status.status == "completed":
                         break
                     time.sleep(1)
 
             messages = client.beta.threads.messages.list(thread_id=st.session_state.thread_id)
-            assistant_message = None
             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
 
-            st.session_state.messages.append({"role": "assistant", "content": assistant_message})
-
             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}"
-                corrected_url = (
-                    f"https://raw.githubusercontent.com/AndrewLORTech/c2ozschlaegerforrestdale/main/"
-                    f"{doc_name}/{doc_name}_page_{page_str}.png"
-                )
-                st.session_state.image_url = corrected_url
+                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.rerun()
-
         except Exception as e:
-            st.error(f"❌ Error: {str(e)}")
-
-# ------------------ Suggested FAQs Section ------------------
-st.markdown("""
-<hr>
-<b>🧠 Suggested FAQs:</b>
-- What are the mechanical services on page 3?
-- Summarize the demolition schedule.
-- List the safety compliance steps.
-""")
+            st.error(f"❌ Error: {e}")