Update app.py

app.py

@@ -15,7 +15,7 @@ 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.error("❌ Missing secrets. Please ensure both OPENAI_API_KEY and ASSISTANT_ID are set in your Hugging Face Space secrets.")
     st.stop()
 
 client = OpenAI(api_key=OPENAI_API_KEY)
@@ -52,10 +52,7 @@ with col1:
 
 # ------------------ Chat Panel (Right) ------------------
 with col2:
-    for message in st.session_state.messages:
-        role, content = message["role"], message["content"]
-        st.chat_message(role).write(content)
-
+    # 🔼 Chat input stays at the top
     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)
@@ -82,7 +79,7 @@ with col2:
             )
 
             # Wait for assistant response
-            with st.spinner("Assistant is thinking..."):
+            with st.spinner("🤖 Assistant is thinking..."):
                 while True:
                     run_status = client.beta.threads.runs.retrieve(
                         thread_id=thread_id,
@@ -103,7 +100,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
+            # Extract GitHub image URL if present
             image_match = re.search(
                 r'https://raw\.githubusercontent\.com/AndrewLORTech/surgical-pathology-manual/main/[\w\-/]*\.png',
                 assistant_message
@@ -111,7 +108,13 @@ 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)}")
+
+    # 🔽 Show previous messages below the input
+    for message in reversed(st.session_state.messages):
+        role, content = message["role"], message["content"]
+        st.chat_message(role).write(content)
+