Update app.py

app.py

@@ -53,6 +53,60 @@ if st.sidebar.button("🔄 Clear Chat"):
 
 show_image = st.sidebar.checkbox("📖 Show Document Image", value=True)
 
+# ------------------ Assistant Query Function ------------------
+def query_assistant(prompt):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    st.chat_message("user").write(prompt)
+
+    try:
+        if st.session_state.thread_id is None:
+            thread = client.beta.threads.create()
+            st.session_state.thread_id = thread.id
+
+        thread_id = st.session_state.thread_id
+
+        client.beta.threads.messages.create(
+            thread_id=thread_id,
+            role="user",
+            content=prompt
+        )
+
+        run = client.beta.threads.runs.create(
+            thread_id=thread_id,
+            assistant_id=ASSISTANT_ID
+        )
+
+        with st.spinner("Assistant is thinking..."):
+            while True:
+                run_status = client.beta.threads.runs.retrieve(
+                    thread_id=thread_id,
+                    run_id=run.id
+                )
+                if run_status.status == "completed":
+                    break
+                time.sleep(1)
+
+        messages = client.beta.threads.messages.list(thread_id=thread_id)
+        for message in reversed(messages.data):
+            if message.role == "assistant":
+                assistant_message = message.content[0].text.value
+                st.chat_message("assistant").write(assistant_message)
+                st.session_state.messages.append({"role": "assistant", "content": assistant_message})
+
+                # Extract GitHub image URL if available
+                image_match = re.search(
+                    r'https://raw\.githubusercontent\.com/AndrewLORTech/surgical-pathology-manual/main/[\w\-/]*\.png',
+                    assistant_message
+                )
+                if image_match:
+                    st.session_state.image_url = image_match.group(0)
+                    st.session_state.image_updated = True
+                return assistant_message
+
+    except Exception as e:
+        st.error(f"❌ Error: {str(e)}")
+        return None
+
 # ------------------ Layout ------------------
 left, center, right = st.columns([1, 2, 1])
 
@@ -70,101 +124,26 @@ with left:
 # ------------------ Center Column: Chat UI ------------------
 with center:
     st.subheader("💬 Document AI Assistant")
-    for message in st.session_state.messages:
+    for message in reversed(st.session_state.messages):
         role, content = message["role"], message["content"]
         st.chat_message(role).write(content)
 
     if prompt := st.chat_input("Type your question about the document..."):
-        st.session_state.messages.append({"role": "user", "content": prompt})
-        st.chat_message("user").write(prompt)
-
-        try:
-            if st.session_state.thread_id is None:
-                thread = client.beta.threads.create()
-                st.session_state.thread_id = thread.id
-
-            thread_id = st.session_state.thread_id
-
-            client.beta.threads.messages.create(
-                thread_id=thread_id,
-                role="user",
-                content=prompt
-            )
-
-            run = client.beta.threads.runs.create(
-                thread_id=thread_id,
-                assistant_id=ASSISTANT_ID
-            )
-
-            with st.spinner("Assistant is thinking..."):
-                while True:
-                    run_status = client.beta.threads.runs.retrieve(
-                        thread_id=thread_id,
-                        run_id=run.id
-                    )
-                    if run_status.status == "completed":
-                        break
-                    time.sleep(1)
-
-            messages = client.beta.threads.messages.list(thread_id=thread_id)
-            assistant_message = None
-            for message in reversed(messages.data):
-                if message.role == "assistant":
-                    assistant_message = message.content[0].text.value
-                    break
+        query_assistant(prompt)
 
-            st.chat_message("assistant").write(assistant_message)
-            st.session_state.messages.append({"role": "assistant", "content": assistant_message})
-
-            # Extract GitHub image URL
-            image_match = re.search(
-                r'https://raw\.githubusercontent\.com/AndrewLORTech/surgical-pathology-manual/main/[\w\-/]*\.png',
-                assistant_message
-            )
-            if image_match:
-                st.session_state.image_url = image_match.group(0)
-                st.session_state.image_updated = True
-                st.rerun()
-
-        except Exception as e:
-            st.error(f"❌ Error: {str(e)}")
-
-# ------------------ Right Column: Structured Summary + FAQ (Button-based) ------------------
+# ------------------ Right Column: Structured Summary + FAQ (Trigger Assistant) ------------------
 with right:
-    st.subheader("📌 Summary & FAQ (from Structured Data)")
+    st.subheader("��� Summary & FAQ (from Assistant)")
 
     col1, col2 = st.columns(2)
     show_summary = col1.button("📝 Load Summary")
     show_faq = col2.button("❓ Load FAQ")
 
-    summary_text = "Click the button to load summary."
-    faq_list = []
-
     if st.session_state.image_url:
         match = re.search(r'/(\d{3})\.png', st.session_state.image_url)
         if match:
             page_number = int(match.group(1))
-            page_entry = next((entry for entry in structured_data if entry.get("page_number") == page_number), None)
-
-            if page_entry:
-                if show_summary:
-                    summary_text = page_entry.get("summary", "No summary available.")
-                if show_faq:
-                    faq_list = page_entry.get("faqs", []) or page_entry.get("questions", [])
-
-    # Display Summary
-    if show_summary:
-        st.subheader("📝 Summary")
-        st.markdown(summary_text)
-
-    # Display FAQs
-    if show_faq:
-        st.subheader("❓ Auto-Generated FAQ")
-        if faq_list:
-            for faq in faq_list:
-                if isinstance(faq, dict):
-                    st.markdown(f"**Q:** {faq.get('question', '')}\n\n**A:** {faq.get('answer', '')}")
-                else:
-                    st.markdown(f"**Q:** {faq}")
-        else:
-            st.info("No FAQs available for this page.")
+            if show_summary:
+                query_assistant(f"Summarize the section on page {page_number}.")
+            if show_faq:
+                query_assistant(f"Provide FAQs for the section on page {page_number}.")