add gemini traceability

app.py

@@ -491,6 +491,36 @@ def main():
                         if rag_result:
                             sources = rag_result.get('sources', []) if isinstance(rag_result, dict) else (rag_result.sources if hasattr(rag_result, 'sources') else [])
                             
+                            # For Gemini, also check gemini_result for sources
+                            if not sources or len(sources) == 0:
+                                gemini_result = chat_result.get('gemini_result')
+                                print(f"🔍 DEBUG: Checking gemini_result for sources...")
+                                print(f"   gemini_result exists: {gemini_result is not None}")
+                                if gemini_result:
+                                    print(f"   gemini_result type: {type(gemini_result)}")
+                                    print(f"   has sources attr: {hasattr(gemini_result, 'sources')}")
+                                    if hasattr(gemini_result, 'sources'):
+                                        print(f"   sources length: {len(gemini_result.sources) if gemini_result.sources else 0}")
+                                
+                                if gemini_result and hasattr(gemini_result, 'sources'):
+                                    # Format Gemini sources for display
+                                    if hasattr(st.session_state.chatbot, 'gemini_client'):
+                                        sources = st.session_state.chatbot.gemini_client.format_sources_for_display(gemini_result)
+                                        print(f"✅ Formatted {len(sources)} sources from gemini_client")
+                                    elif hasattr(st.session_state.chatbot, '_format_gemini_sources'):
+                                        sources = st.session_state.chatbot._format_gemini_sources(gemini_result)
+                                        print(f"✅ Formatted {len(sources)} sources from _format_gemini_sources")
+                            
+                            # Update rag_result with sources if we found them
+                            if sources and len(sources) > 0:
+                                if isinstance(rag_result, dict):
+                                    rag_result['sources'] = sources
+                                elif hasattr(rag_result, 'sources'):
+                                    rag_result.sources = sources
+                                # Update last_rag_result with sources
+                                st.session_state.last_rag_result = rag_result
+                                print(f"✅ Updated rag_result with {len(sources)} sources")
+                            
                             # Get the actual RAG query
                             actual_rag_query = chat_result.get('actual_rag_query', '')
                             if actual_rag_query:
@@ -516,6 +546,9 @@ def main():
                                 "timestamp": time.time()
                             }
                             st.session_state.rag_retrieval_history.append(retrieval_entry)
+                            
+                            # Debug logging
+                            print(f"📊 RETRIEVAL TRACKING: {len(sources)} sources stored in retrieval history")
                     else:
                         response = chat_result
                         st.session_state.last_rag_result = None
@@ -545,6 +578,16 @@ def main():
                 # Dictionary format from multi-agent system
                 sources = rag_result['sources']
             
+            # For Gemini, also check if we need to format sources from gemini_result
+            if (not sources or len(sources) == 0) and isinstance(rag_result, dict):
+                gemini_result = rag_result.get('gemini_result')
+                if gemini_result and hasattr(gemini_result, 'sources'):
+                    # Format Gemini sources for display
+                    if hasattr(st.session_state.chatbot, 'gemini_client'):
+                        sources = st.session_state.chatbot.gemini_client.format_sources_for_display(gemini_result)
+                    elif hasattr(st.session_state.chatbot, '_format_gemini_sources'):
+                        sources = st.session_state.chatbot._format_gemini_sources(gemini_result)
+            
             if sources and len(sources) > 0:
                 # Count unique filenames
                 unique_filenames = set()
@@ -772,7 +815,11 @@ def main():
                                         logger.info("📤 SNOWFLAKE UI: Attempting to save feedback to Snowflake...")
                                         print("📤 SNOWFLAKE UI: Attempting to save feedback to Snowflake...")
                                         
-                                        snowflake_success = feedback_manager.save_to_snowflake(feedback_obj)
+                                        # Show spinner while saving to Snowflake (can take 10-15 seconds)
+                                        # This includes: connection establishment (~5s), data preparation, and SQL execution (~5s)
+                                        with st.spinner("💾 Saving feedback to Snowflake... This may take 10-15 seconds (connecting to database, preparing data, and executing query)"):
+                                            snowflake_success = feedback_manager.save_to_snowflake(feedback_obj)
+                                        
                                         if snowflake_success:
                                             logger.info("✅ SNOWFLAKE UI: Successfully saved to Snowflake")
                                             print("✅ SNOWFLAKE UI: Successfully saved to Snowflake")