Update app.py

app.py

@@ -356,10 +356,6 @@ def main():
     with chat_col:
         # Display chat history first
         for i, (role, content) in enumerate(st.session_state.chat_history):
-            # Debug print
-            st.write("Debug - Content type:", type(content))
-            st.write("Debug - Content:", content)
-            
             if role == "user":
                 st.markdown(f"""
                     <div class="chat-message user-message">
@@ -368,49 +364,29 @@ def main():
                     </div>
                 """, unsafe_allow_html=True)
             else:
-                # Handle both string and dictionary responses
-                if isinstance(content, dict):
-                    # Try different possible structures
-                    assistant_response = content.get('answer', content.get('response', ''))
-                    documents = content.get('documents', content.get('refs', content.get('sources', [])))
-                    
-                    st.markdown(f"""
-                        <div class="chat-message assistant-message">
-                            <strong>🤖 คำตอบ:</strong><br>
-                            {assistant_response}
-                        </div>
-                    """, unsafe_allow_html=True)
-                    
-                    # Debug print documents
-                    st.write("Debug - Documents:", documents)
-                    
-                    if documents:
-                        with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=True):
-                            for idx, doc in enumerate(documents, 1):
-                                st.write(f"Debug - Document {idx}:", doc)
-                                
-                                # Try different ways to get document content
-                                if isinstance(doc, dict):
-                                    doc_content = doc.get('content', doc.get('text', str(doc)))
-                                else:
-                                    doc_content = str(doc)
-                                
-                                st.markdown(f"""
-                                    <div style="padding: 1rem; background-color: #F3F4F6; border-radius: 8px; margin: 0.5rem 0; border: 1px solid #E5E7EB;">
-                                        <strong>เอกสารที่ {idx}:</strong><br>
-                                        {doc_content}
-                                    </div>
-                                """, unsafe_allow_html=True)
-                else:
-                    # If content is a string
-                    st.markdown(f"""
-                        <div class="chat-message assistant-message">
-                            <strong>🤖 คำตอบ:</strong><br>
-                            {content}
-                        </div>
-                    """, unsafe_allow_html=True)
+                if isinstance(content, dict):  
+                    assistant_response = content.get('answer', '❌ ไม่มีข้อมูลคำตอบ')
+                else:  
+                    assistant_response = content
+
+                st.markdown(f"""
+                    <div class="chat-message assistant-message">
+                        <strong>🤖 คำตอบ:</strong><br>
+                        {assistant_response}
+                    </div>
+                """, unsafe_allow_html=True)
 
-        # Create chat input using the implementation
+                if isinstance(content, dict) and content.get('documents'):
+                    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)
+        
+        # Create chat input using the new implementation
         create_chat_input()
 
     # Sidebar info column