Update app.py

app.py

@@ -2,6 +2,9 @@ import streamlit as st
 import os
 import time
 import re
+import json
+import requests
+from PIL import Image
 from openai import OpenAI
 
 # ------------------ App Configuration ------------------
@@ -13,7 +16,6 @@ st.caption("Chat with an AI Assistant on your medical/pathology documents")
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 ASSISTANT_ID = os.environ.get("ASSISTANT_ID")
 
-# ------------------ Error Handling for Missing Secrets ------------------
 if not OPENAI_API_KEY or not ASSISTANT_ID:
     st.error("Missing secrets. Please ensure both OPENAI_API_KEY and ASSISTANT_ID are set in your Hugging Face Space secrets.")
     st.stop()
@@ -41,17 +43,27 @@ if st.sidebar.button("🔄 Clear Chat"):
 
 show_image = st.sidebar.checkbox("📖 Show Document Image", value=True)
 
-# ------------------ Split Layout ------------------
-col1, col2 = st.columns([1, 2])  # Adjust ratio as needed
+# ------------------ Load Structured Summary/FAQ ------------------
+with open("51940670-Manual-of-Surgical-Pathology-Third-Edition_1_structured_output.json", "r") as f:
+    structured_data = json.load(f)
 
-# ------------------ Image Panel (Left) ------------------
-with col1:
+# ------------------ Three-Column Layout ------------------
+left, center, right = st.columns([1, 2, 1])  # adjust as needed
+
+# ------------------ Left Column: Document Image ------------------
+with left:
+    st.subheader("📄 Document Image")
     if show_image and st.session_state.image_url:
-        st.image(st.session_state.image_url, caption="📑 Extracted Page", use_container_width=True)
-        st.session_state.image_updated = False  # Reset flag after rendering
+        try:
+            image = Image.open(requests.get(st.session_state.image_url, stream=True).raw)
+            st.image(image, caption="📑 Extracted Page", use_column_width=True)
+            st.session_state.image_updated = False
+        except Exception as e:
+            st.warning("⚠️ Could not load image.")
 
-# ------------------ Chat Panel (Right) ------------------
-with col2:
+# ------------------ Center Column: Chat UI ------------------
+with center:
+    st.subheader("💬 Document AI Assistant")
     for message in st.session_state.messages:
         role, content = message["role"], message["content"]
         st.chat_message(role).write(content)
@@ -61,14 +73,13 @@ with col2:
         st.chat_message("user").write(prompt)
 
         try:
-            # Initialize thread if needed
             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
 
-            # Send message to assistant
+            # Send user prompt
             client.beta.threads.messages.create(
                 thread_id=thread_id,
                 role="user",
@@ -81,7 +92,7 @@ with col2:
                 assistant_id=ASSISTANT_ID
             )
 
-            # Wait for assistant response
+            # Poll until done
             with st.spinner("Assistant is thinking..."):
                 while True:
                     run_status = client.beta.threads.runs.retrieve(
@@ -92,7 +103,7 @@ with col2:
                         break
                     time.sleep(1)
 
-            # Get assistant response
+            # Get assistant message
             messages = client.beta.threads.messages.list(thread_id=thread_id)
             assistant_message = None
             for message in reversed(messages.data):
@@ -103,7 +114,7 @@ with col2:
             st.chat_message("assistant").write(assistant_message)
             st.session_state.messages.append({"role": "assistant", "content": assistant_message})
 
-            # Extract GitHub image from response if available
+            # Detect GitHub image in response
             image_match = re.search(
                 r'https://raw\.githubusercontent\.com/AndrewLORTech/surgical-pathology-manual/main/[\w\-/]*\.png',
                 assistant_message
@@ -111,7 +122,28 @@ with col2:
             if image_match:
                 st.session_state.image_url = image_match.group(0)
                 st.session_state.image_updated = True
-                st.rerun()  # Trigger rerun to refresh image display
+                st.rerun()
 
         except Exception as e:
             st.error(f"❌ Error: {str(e)}")
+
+# ------------------ Right Column: Summary and FAQ ------------------
+with right:
+    st.subheader("📌 Summary")
+
+    if st.session_state.image_url:
+        match = re.search(r'page_(\d+)', st.session_state.image_url)
+        page_number = int(match.group(1)) if match else None
+    else:
+        page_number = 151  # default
+
+    summary_text = structured_data.get(str(page_number), {}).get("summary", "No summary available.")
+    st.markdown(summary_text)
+
+    st.subheader("❓ Auto-Generated FAQ")
+    faq_list = structured_data.get(str(page_number), {}).get("faqs", [])
+    if faq_list:
+        for faq in faq_list:
+            st.markdown(f"**Q:** {faq.get('question', '')}\n\n**A:** {faq.get('answer', '')}")
+    else:
+        st.info("No FAQs available for this page.")