Update app.py

app.py

@@ -286,6 +286,9 @@ def main():
     if 'chat_history' not in st.session_state:
         st.session_state.chat_history = []
         
+    if 'context_memory' not in st.session_state:
+        st.session_state.context_memory = []
+        
     if 'query' not in st.session_state:
         st.session_state.query = ''
 
@@ -332,26 +335,23 @@ def main():
                         {assistant_response}
                     </div>
                 """, unsafe_allow_html=True)
-                
-                # Add expanders right after the answer
-                with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
-                    for i, doc in enumerate(content['documents'], 1):
-                        st.markdown(f"""
-                            <div style="padding: 1rem; background-color: #000000; border-radius: 8px; margin: 0.5rem 0;">
-                                <strong>เอกสารที่ {i}:</strong><br>
-                                {doc.content}
-                            </div>
-                        """, unsafe_allow_html=True)
-                
-                # with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม", expanded=False):
-                #     st.json(content['query_info'])
+
+                # Add expanders right after the answer (if additional document references exist)
+                if isinstance(content, dict) and 'documents' in content:
+                    with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
+                        for i, doc in enumerate(content['documents'], 1):
+                            st.markdown(f"""
+                                <div style="padding: 1rem; background-color: #F9FAFB; border-radius: 8px; margin: 0.5rem 0;">
+                                    <strong>เอกสารที่ {i}:</strong><br>
+                                    {doc.get('content', 'ไม่มีข้อมูล')}
+                                </div>
+                            """, unsafe_allow_html=True)
 
         # Input section at the bottom
         with st.container():
-            # Replace this section in your code:
             st.markdown("""
                 <label for="query_input" style="font-size: 1.2rem; font-weight: 600; margin-bottom: 1rem; display: block;">
-                    <span style="color: #ffffff; border-left: 4px solid #ffffff; padding-left: 0.8rem;">
+                    <span style="color: #1E3A8A; border-left: 4px solid #1E3A8A; padding-left: 0.8rem;">
                         โปรดระบุคำถามเกี่ยวกับปฏิทินการศึกษา:
                     </span>
                 </label>
@@ -377,7 +377,7 @@ def main():
                     st.session_state.chat_history = []
                     st.rerun()
 
-    # Info column remains the same
+    # Sidebar info column
     with info_col:
         st.markdown("""
             <div style="background-color: #F9FAFB; padding: 1.5rem; border-radius: 12px; margin-bottom: 2rem;">