Update app.py

app.py

@@ -10,7 +10,7 @@ from openai import OpenAI
 # ------------------ App Configuration ------------------
 st.set_page_config(page_title="Schlaeger Forrestdale DocAIA", layout="wide")
 st.title("📄 Schlaeger Forrestdale Document Assistant")
-st.caption("Chat with construction documents using OCR-based AI")
+st.caption("Explore technical construction documents via AI with OCR references")
 
 # ------------------ Load API Key and Assistant ID ------------------
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
@@ -61,26 +61,21 @@ with image_col:
 
 # ------------------ 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:
-        with st.chat_message(msg["role"]):
-            st.markdown(msg["content"])
+    st.markdown("### 🧠 Ask a Document-Specific Question")
+    prompt = st.chat_input("Example: What is the defects liability period?")
+
+    st.markdown("<hr style='margin:0.5em 0;'>", unsafe_allow_html=True)
+    st.markdown("#### ⚡ Prompt Templates")
+    col_btn1, col_btn2, col_btn3 = st.columns(3)
+    with col_btn1:
+        if st.button("Summarize Current Page"):
+            prompt = "Summarize the current page."
+    with col_btn2:
+        if st.button("List Contractual Obligations"):
+            prompt = "List the contractual obligations from this section."
+    with col_btn3:
+        if st.button("Find Related AS Clauses"):
+            prompt = "Which clauses reference AS 4000-1997 or similar standards?"
 
     if prompt:
         st.session_state.messages.append({"role": "user", "content": prompt})
@@ -93,7 +88,7 @@ with chat_col:
 
             run = client.beta.threads.runs.create(thread_id=st.session_state.thread_id, assistant_id=ASSISTANT_ID)
 
-            with st.spinner("🤖 Processing your query..."):
+            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 == "completed":
@@ -119,3 +114,7 @@ with chat_col:
             st.rerun()
         except Exception as e:
             st.error(f"❌ Error: {e}")
+
+    for msg in st.session_state.messages:
+        with st.chat_message(msg["role"]):
+            st.markdown(msg["content"], unsafe_allow_html=True)