Update app.py

app.py

@@ -499,6 +499,166 @@ def initialize_rag_pipeline():
 #         logging.error(f"Application error: {str(e)}")
 #         st.error("An unexpected error occurred. Please check the logs and try again.")
 
+# def main():
+#     try:
+#         # Environment check
+#         if not check_environment():
+#             return
+
+#         # Improved CSS styling
+#         st.markdown("""
+#             <style>
+#             /* Container styling */
+#             .block-container {
+#                 padding-top: 2rem;
+#                 padding-bottom: 2rem;
+#             }
+            
+#             /* Text input styling */
+#             .stTextInput > div > div > input {
+#                 width: 100%;
+#             }
+            
+#             /* Button styling */
+#             .stButton > button {
+#                 width: 200px;
+#                 margin: 0 auto;
+#                 display: block;
+#                 background-color: #FF4B4B;
+#                 color: white;
+#                 border-radius: 5px;
+#                 padding: 0.5rem 1rem;
+#             }
+            
+#             /* Title styling */
+#             .main-title {
+#                 text-align: center;
+#                 padding: 1rem 0;
+#                 font-size: 3rem;
+#                 color: #1F1F1F;
+#             }
+            
+#             .sub-title {
+#                 text-align: center;
+#                 padding: 0.5rem 0;
+#                 font-size: 1.5rem;
+#                 color: #4F4F4F;
+#             }
+            
+#             /* Description styling */
+#             .description {
+#                 text-align: center;
+#                 color: #666666;
+#                 padding: 0.5rem 0;
+#                 font-size: 1.1rem;
+#                 line-height: 1.6;
+#                 margin-bottom: 1rem;
+#             }
+
+#             /* Answer container styling */
+#             .stMarkdown {
+#                 max-width: 100%;
+#             }
+
+#             /* Streamlit default overrides */
+#             .st-emotion-cache-16idsys p {
+#                 font-size: 1.1rem;
+#                 line-height: 1.6;
+#             }
+            
+#             /* Container for main content */
+#             .main-content {
+#                 max-width: 1200px;
+#                 margin: 0 auto;
+#                 padding: 0 1rem;
+#             }
+#             </style>
+#         """, unsafe_allow_html=True)
+
+#         # Header section
+#         st.markdown("<h1 class='main-title'>🏆 The Sport Chatbot</h1>", unsafe_allow_html=True)
+#         st.markdown("<h3 class='sub-title'>Using ESPN API</h3>", unsafe_allow_html=True)
+#         st.markdown("""
+#             <p class='description'>
+#                 Hey there! 👋 I can help you with information on Ice Hockey, Baseball, American Football, Soccer, and Basketball. 
+#                 With access to the ESPN API, I'm up to date with the latest details for these sports up until October 2024.
+#             </p>
+#             <p class='description'>
+#                 Got any general questions? Feel free to ask—I'll do my best to provide answers based on the information I've been trained on!
+#             </p>
+#         """, unsafe_allow_html=True)
+
+#         # Initialize the pipeline with better error handling
+#         if 'rag' not in st.session_state:
+#             try:
+#                 with st.spinner("Loading resources..."):
+#                     st.session_state.rag = initialize_rag_pipeline()
+#                     logging.info("Pipeline initialized successfully")
+#             except Exception as e:
+#                 logging.error(f"Pipeline initialization error: {str(e)}")
+#                 st.error("Failed to initialize the system. Please check the logs.")
+#                 st.stop()
+#                 return
+
+#         # Create columns for layout
+#         col1, col2, col3 = st.columns([1, 6, 1])
+        
+#         with col2:
+#             # Query input
+#             query = st.text_input("What would you like to know about sports?")
+            
+#             if st.button("Get Answer"):
+#                 if query:
+#                     response_placeholder = st.empty()
+#                     try:
+#                         # Log query processing start
+#                         logging.info(f"Processing query: {query}")
+                        
+#                         # Process query and get response
+#                         response = st.session_state.rag.process_query(query, response_placeholder)
+                        
+#                         # Log successful response
+#                         logging.info(f"Generated response: {response}")
+#                     except Exception as e:
+#                         # Log error details
+#                         logging.error(f"Query processing error: {str(e)}")
+#                         logging.error("Full error details: ", exc_info=True)
+#                         response_placeholder.warning("Unable to process your question. Please try again.")
+#                 else:
+#                     st.warning("Please enter a question!")
+
+#         # Footer
+#         st.markdown("<br><br>", unsafe_allow_html=True)
+#         st.markdown("---")
+#         st.markdown("""
+#             <p style='text-align: center; color: #666666; padding: 1rem 0;'>
+#                 Powered by ESPN Data & Mistral AI 🚀
+#             </p>
+#         """, unsafe_allow_html=True)
+
+#     except Exception as e:
+#         logging.error(f"Application error: {str(e)}")
+#         logging.error("Full error details: ", exc_info=True)
+#         st.error("An unexpected error occurred. Please check the logs and try again.")
+
+# if __name__ == "__main__":
+#     # Configure logging
+#     logging.basicConfig(
+#         level=logging.INFO,
+#         format='%(asctime)s - %(levelname)s - %(message)s'
+#     )
+    
+#     try:
+#         main()
+#     except Exception as e:
+#         logging.error(f"Fatal error: {str(e)}")
+#         logging.error("Full error details: ", exc_info=True)
+#         st.error("A fatal error occurred. Please check the logs and try again.")
+
+# if __name__ == "__main__":
+#     main()
+
+
 def main():
     try:
         # Environment check
@@ -575,18 +735,24 @@ def main():
             </style>
         """, unsafe_allow_html=True)
 
-        # Header section
-        st.markdown("<h1 class='main-title'>🏆 The Sport Chatbot</h1>", unsafe_allow_html=True)
-        st.markdown("<h3 class='sub-title'>Using ESPN API</h3>", unsafe_allow_html=True)
+        # Header section with unique keys
+        st.markdown("<h1 class='main-title'>🏆 The Sport Chatbot</h1>", unsafe_allow_html=True, key="title")
+        st.markdown("<h3 class='sub-title'>Using ESPN API</h3>", unsafe_allow_html=True, key="subtitle")
         st.markdown("""
             <p class='description'>
                 Hey there! 👋 I can help you with information on Ice Hockey, Baseball, American Football, Soccer, and Basketball. 
                 With access to the ESPN API, I'm up to date with the latest details for these sports up until October 2024.
             </p>
+        """, unsafe_allow_html=True, key="desc1")
+        
+        st.markdown("""
             <p class='description'>
                 Got any general questions? Feel free to ask—I'll do my best to provide answers based on the information I've been trained on!
             </p>
-        """, unsafe_allow_html=True)
+        """, unsafe_allow_html=True, key="desc2")
+
+        # Add spacing
+        st.markdown("<br>", unsafe_allow_html=True, key="spacing1")
 
         # Initialize the pipeline with better error handling
         if 'rag' not in st.session_state:
@@ -596,7 +762,7 @@ def main():
                     logging.info("Pipeline initialized successfully")
             except Exception as e:
                 logging.error(f"Pipeline initialization error: {str(e)}")
-                st.error("Failed to initialize the system. Please check the logs.")
+                st.error("Failed to initialize the system. Please check the logs.", key="init_error")
                 st.stop()
                 return
 
@@ -604,42 +770,54 @@ def main():
         col1, col2, col3 = st.columns([1, 6, 1])
         
         with col2:
-            # Query input
-            query = st.text_input("What would you like to know about sports?")
+            # Query input with unique key
+            query = st.text_input(
+                "What would you like to know about sports?",
+                key="sports_query"
+            )
             
-            if st.button("Get Answer"):
+            # Centered button with unique key
+            if st.button("Get Answer", key="answer_button"):
                 if query:
                     response_placeholder = st.empty()
                     try:
                         # Log query processing start
                         logging.info(f"Processing query: {query}")
                         
-                        # Process query and get response
-                        response = st.session_state.rag.process_query(query, response_placeholder)
-                        
-                        # Log successful response
-                        logging.info(f"Generated response: {response}")
+                        with st.spinner("Processing your question..."):
+                            # Process query and get response
+                            response = st.session_state.rag.process_query(query, response_placeholder)
+                            
+                            # Log successful response
+                            logging.info(f"Generated response: {response}")
+                            
                     except Exception as e:
                         # Log error details
                         logging.error(f"Query processing error: {str(e)}")
                         logging.error("Full error details: ", exc_info=True)
-                        response_placeholder.warning("Unable to process your question. Please try again.")
+                        response_placeholder.warning(
+                            "Unable to process your question. Please try again.",
+                            key="query_error"
+                        )
                 else:
-                    st.warning("Please enter a question!")
+                    st.warning("Please enter a question!", key="empty_query_warning")
 
-        # Footer
-        st.markdown("<br><br>", unsafe_allow_html=True)
-        st.markdown("---")
+        # Footer with unique keys
+        st.markdown("<br><br>", unsafe_allow_html=True, key="spacing2")
+        st.markdown("---", key="divider")
         st.markdown("""
             <p style='text-align: center; color: #666666; padding: 1rem 0;'>
                 Powered by ESPN Data & Mistral AI 🚀
             </p>
-        """, unsafe_allow_html=True)
+        """, unsafe_allow_html=True, key="footer")
 
     except Exception as e:
         logging.error(f"Application error: {str(e)}")
         logging.error("Full error details: ", exc_info=True)
-        st.error("An unexpected error occurred. Please check the logs and try again.")
+        st.error(
+            "An unexpected error occurred. Please check the logs and try again.",
+            key="main_error"
+        )
 
 if __name__ == "__main__":
     # Configure logging
@@ -653,7 +831,7 @@ if __name__ == "__main__":
     except Exception as e:
         logging.error(f"Fatal error: {str(e)}")
         logging.error("Full error details: ", exc_info=True)
-        st.error("A fatal error occurred. Please check the logs and try again.")
-
-if __name__ == "__main__":
-    main()
+        st.error(
+            "A fatal error occurred. Please check the logs and try again.",
+            key="fatal_error"
+        )