Update app.py

app.py

@@ -12,10 +12,12 @@ st.caption("Chat with an Ai Assistant on your Documents")
 with st.sidebar:
     OPENAI_API_KEY = st.text_input("Enter your C2 Group of Technologies Access Key", type="password")
 
+# Exit if no key provided
 if not OPENAI_API_KEY:
     st.error("Please enter your C2 Group of Technologies Access Key to continue.")
     st.stop()
 
+# Initialize OpenAI client
 client = OpenAI(api_key=OPENAI_API_KEY)
 ASSISTANT_ID = "asst_PJjxQftfz2IJUUMvnldK58lB"
 
@@ -31,37 +33,38 @@ if st.button("Clear Chat", use_container_width=True):
     st.session_state.thread_id = None
     st.rerun()
 
-# Display history
+# Show message history
 for message in st.session_state.messages:
     role, content = message["role"], message["content"]
     st.chat_message(role).write(content)
 
-# Handle new input
+# Handle new user message
 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 if none exists
         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 input to OpenAI
+        # Send user prompt to assistant
         client.beta.threads.messages.create(
             thread_id=thread_id,
             role="user",
             content=prompt
         )
 
-        # Start assistant run
+        # Run assistant
         run = client.beta.threads.runs.create(
             thread_id=thread_id,
             assistant_id=ASSISTANT_ID
         )
 
-        # Wait until complete
+        # Wait until completion
         while True:
             run_status = client.beta.threads.runs.retrieve(
                 thread_id=thread_id,
@@ -71,7 +74,7 @@ if prompt := st.chat_input():
                 break
             time.sleep(1)
 
-        # Get the latest assistant message
+        # Retrieve assistant message
         messages = client.beta.threads.messages.list(thread_id=thread_id)
         assistant_message = None
         for message in reversed(messages.data):
@@ -79,13 +82,13 @@ if prompt := st.chat_input():
                 assistant_message = message.content[0].text.value
                 break
 
-        # Display response text
+        # Display assistant message
         st.chat_message("assistant").write(assistant_message)
         st.session_state.messages.append({"role": "assistant", "content": assistant_message})
 
-        # Check if assistant already gave a full image URL
+        # Inject page image if assistant didn't include it
         if "![Page Image](" not in assistant_message:
-            # Try to extract the referenced page number
+            # Find the page number
             page_match = re.search(r'Page (\d+)', assistant_message)
             if page_match:
                 page_number = page_match.group(1).zfill(3)