Update app.py

app.py

@@ -3,7 +3,7 @@ from openai import OpenAI
 import time
 import re
 
-# Streamlit page setup
+# Streamlit setup
 st.set_page_config(page_title="documentaitest")
 st.title("documentaitest")
 st.caption("Chat with an Ai Assistant on your Documents")
@@ -12,6 +12,7 @@ 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")
 
+# Stop execution if no API key provided
 if not OPENAI_API_KEY:
     st.error("Please enter your C2 Group of Technologies Access Key to continue.")
     st.stop()
@@ -19,36 +20,37 @@ if not OPENAI_API_KEY:
 client = OpenAI(api_key=OPENAI_API_KEY)
 ASSISTANT_ID = "asst_PJjxQftfz2IJUUMvnldK58lB"
 
-# Session state
+# Initialize 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 to clear chat
 if st.button("Clear Chat", use_container_width=True):
     st.session_state.messages = []
     st.session_state.thread_id = None
     st.rerun()
 
-# Show history
+# Display 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 input
 if prompt := st.chat_input():
     st.session_state.messages.append({"role": "user", "content": prompt})
     st.chat_message("user").write(prompt)
 
     try:
+        # Create thread if not yet created
         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
+        # Send message
         client.beta.threads.messages.create(
             thread_id=thread_id,
             role="user",
@@ -63,19 +65,23 @@ if prompt := st.chat_input():
 
         # Poll until complete
         while True:
-            run_status = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
+            run_status = client.beta.threads.runs.retrieve(
+                thread_id=thread_id,
+                run_id=run.id
+            )
             if run_status.status == "completed":
                 break
             time.sleep(1)
 
         # Get assistant response
         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
 
-        # Show response
+        # Display assistant message
         st.chat_message("assistant").write(assistant_message)
         st.session_state.messages.append({"role": "assistant", "content": assistant_message})
 
@@ -84,7 +90,7 @@ if prompt := st.chat_input():
         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)
+            st.image(image_url, caption=f"Page {int(page_number)}", use_container_width=True)
 
     except Exception as e:
         st.error(f"Error: {str(e)}")