Update src/app.py

src/app.py

@@ -313,24 +313,24 @@ if st.button("🔍 Search & Generate Answer"):
                 
                 st.subheader("📝 Answer")
                 
-                col1, col2, col3 = st.columns(3)
+                col1 = st.columns(1)
                 with col1:
                     st.metric("Sources Used", result['sources_used'])
 
                 
                 st.write(result['answer'])
                 
-                #if st.checkbox("📚 Show Source Documents"):
-                #    st.subheader("Sources Used in Answer")
-                #    for idx, source in enumerate(result['formatted_sources'], 1):
-                #        relevance = source['relevance']
-                #        relevance_bar = "█" * int(relevance * 10) + "░" * (10 - int(relevance * 10))
-                #        
-                #        with st.expander(
-                #            f"Source {idx} - {source['type'].upper()} "
-                #            f"[{relevance_bar}] {relevance:.0%}"
-                #        ):
-                #            st.write(source['content'])
+                if st.checkbox("📚 Show Source Documents"):
+                    st.subheader("Sources Used in Answer")
+                    for idx, source in enumerate(result['formatted_sources'], 1):
+                        relevance = source['relevance']
+                        relevance_bar = "█" * int(relevance * 10) + "░" * (10 - int(relevance * 10))
+                        
+                        with st.expander(
+                            f"Source {idx} - {source['type'].upper()} "
+                            f"[{relevance_bar}] {relevance:.0%}"
+                        ):
+                            st.write(source['content'])
                 
                 print(f"\n✅ Answer generation complete!")
                 
@@ -339,4 +339,4 @@ if st.button("🔍 Search & Generate Answer"):
             print(f"Error: {e}")
 
 
-st.divider()
+st.divider()