Update app.py

app.py

@@ -1,87 +1,90 @@
 import streamlit as st
 from openai import OpenAI
 import time
+import re
 
+# Streamlit page setup
 st.set_page_config(page_title="documentaitest")
-
 st.title("documentaitest")
 st.caption("Chat with an Ai Assistant on your Documents")
 
-# Sidebar for API Key input
+# Sidebar for API key input
 with st.sidebar:
     OPENAI_API_KEY = st.text_input("Enter your C2 Group of Technologies Access Key", type="password")
 
-if OPENAI_API_KEY:
-    client = OpenAI(api_key=OPENAI_API_KEY)
-else:
+if not OPENAI_API_KEY:
     st.error("Please enter your C2 Group of Technologies Access Key to continue.")
     st.stop()
 
+client = OpenAI(api_key=OPENAI_API_KEY)
 ASSISTANT_ID = "asst_PJjxQftfz2IJUUMvnldK58lB"
 
-# Initialize session state
+# Session state
 if "messages" not in st.session_state:
     st.session_state.messages = []
 if "thread_id" not in st.session_state:
     st.session_state.thread_id = None
 
-# Clear chat button
+# Clear chat
 if st.button("Clear Chat", use_container_width=True):
     st.session_state.messages = []
     st.session_state.thread_id = None
     st.rerun()
 
-# Display chat history
+# Show history
 for message in st.session_state.messages:
     role, content = message["role"], message["content"]
     st.chat_message(role).write(content)
 
-# Process user input
+# Handle new input
 if prompt := st.chat_input():
     st.session_state.messages.append({"role": "user", "content": prompt})
     st.chat_message("user").write(prompt)
 
     try:
-        # Create a new thread only if one doesn't exist yet
         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 user message to OpenAI API in the existing thread
+        # Send user message
         client.beta.threads.messages.create(
             thread_id=thread_id,
             role="user",
             content=prompt
         )
 
-        # Run the assistant to generate a response
+        # Run assistant
         run = client.beta.threads.runs.create(
             thread_id=thread_id,
             assistant_id=ASSISTANT_ID
         )
 
-        # Wait for response
+        # Poll until complete
         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)
 
-        # Retrieve assistant response
+        # Get assistant response
         messages = client.beta.threads.messages.list(thread_id=thread_id)
-        # Get the most recent assistant message
         for message in reversed(messages.data):
             if message.role == "assistant":
                 assistant_message = message.content[0].text.value
                 break
 
-        # Display assistant's response
+        # Show response
         st.chat_message("assistant").write(assistant_message)
-
-        # Store in session state
         st.session_state.messages.append({"role": "assistant", "content": assistant_message})
 
+        # 🔍 Detect and display GitHub-hosted image
+        page_match = re.search(r'Page (\d+)', assistant_message)
+        if page_match:
+            page_number = page_match.group(1).zfill(3)
+            image_url = f"https://raw.githubusercontent.com/AndrewLORTech/surgical-pathology-manual/main/51940670-Manual-of-Surgical-Pathology-Third-Edition_1_page_{page_number}.png"
+            st.image(image_url, caption=f"Page {int(page_number)}", use_column_width=True)
+
     except Exception as e:
         st.error(f"Error: {str(e)}")