Update app.py

app.py

@@ -49,20 +49,21 @@ col1, col2 = st.columns([1, 2])
 # ------------------ Left Panel: Image ------------------
 with col1:
     if show_image and st.session_state.image_url:
-        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.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}")
+        with st.spinner("🖼️ Loading image 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.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...")
+    prompt = st.chat_input("Type your question about the document or choose from FAQs below...")
 
-    # Display chat history
+    # Display chat history in a cleaner bubble layout
     paired_messages = []
     buffer = []
     for msg in st.session_state.messages:
@@ -76,31 +77,28 @@ with col2:
     for pair in reversed(paired_messages):
         for msg in pair:
             with st.chat_message(msg["role"]):
-                st.write(msg["content"])
+                st.markdown(f"{msg['content']}", unsafe_allow_html=True)
 
     if prompt:
         st.session_state.messages.append({"role": "user", "content": prompt})
 
         try:
-            # Create thread
             if st.session_state.thread_id is None:
                 thread = client.beta.threads.create()
                 st.session_state.thread_id = thread.id
 
-            # Send user message
             client.beta.threads.messages.create(
                 thread_id=st.session_state.thread_id,
                 role="user",
                 content=prompt
             )
 
-            # Run assistant
             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("🤖 Assistant is thinking..."):
                 while True:
                     run_status = client.beta.threads.runs.retrieve(
                         thread_id=st.session_state.thread_id,
@@ -110,7 +108,6 @@ with col2:
                         break
                     time.sleep(1)
 
-            # Retrieve assistant reply
             messages = client.beta.threads.messages.list(thread_id=st.session_state.thread_id)
             assistant_message = None
             for message in reversed(messages.data):
@@ -120,7 +117,6 @@ with col2:
 
             st.session_state.messages.append({"role": "assistant", "content": assistant_message})
 
-            # ✅ Parse Document Reference line and generate correct image URL
             match = re.search(r'Document Reference:\s+(.+?),\s+Page\s+(\d+)', assistant_message)
             if match:
                 doc_name = match.group(1).strip()
@@ -136,4 +132,13 @@ with col2:
             st.rerun()
 
         except Exception as e:
-            st.error(f"❌ Error: {str(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.
+""")