Update app.py

app.py

@@ -337,15 +337,23 @@ def main():
                 """, unsafe_allow_html=True)
 
                 # Add expanders right after the answer (if additional document references exist)
-                if isinstance(content, dict) and 'documents' in content:
-                    with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
+                if isinstance(content, dict) and content.get('documents'):
+                    with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=True):
                         for i, doc in enumerate(content['documents'], 1):
+                            doc_content = doc.get('content', 'ไม่มีข้อมูล')  # Ensure 'content' key exists
+                            doc_source = doc.get('source', 'ไม่ทราบแหล่งที่มา')  # Add document source info if available
+                
                             st.markdown(f"""
                                 <div style="padding: 1rem; background-color: #F9FAFB; border-radius: 8px; margin: 0.5rem 0;">
-                                    <strong>เอกสารที่ {i}:</strong><br>
-                                    {doc.get('content', 'ไม่มีข้อมูล')}
+                                    <strong>📄 เอกสารที่ {i}:</strong> <br>
+                                    <em>🔗 แหล่งที่มา: {doc_source}</em> <br><br>
+                                    {doc_content}
                                 </div>
                             """, unsafe_allow_html=True)
+                else:
+                    with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
+                        st.markdown("<em>ไม่มีข้อมูลอ้างอิงสำหรับคำถามนี้</em>", unsafe_allow_html=True)
+
 
         # Input section at the bottom
         with st.container():