Implement robust API key handling and set gpt-4o as default model

app.py

@@ -23,28 +23,41 @@ def main():
     st.subheader("Process customer feedback with custom LLM instructions")
     
     # Initialize session state for API key
-    if 'api_key_set' not in st.session_state:
-        st.session_state.api_key_set = False
+    if 'openai_api_key' not in st.session_state:
+        st.session_state.openai_api_key = ""
     
     # Sidebar for configuration
     with st.sidebar:
         st.header("Configuration")
         
         # API Key input
-        api_key = st.text_input("OpenAI API Key", type="password")
-        if api_key:
-            # Set API key directly in the openai module
-            openai.api_key = api_key
-            # Also set it in environment variables as a backup
-            os.environ["OPENAI_API_KEY"] = api_key
-            st.session_state.api_key_set = True
-            st.success("API Key set successfully!")
-            logger.info("API key configured")
+        api_key = st.text_input("OpenAI API Key", type="password", key="api_key_input", value=st.session_state.openai_api_key)
+        
+        # Set API key button
+        if st.button("Set API Key"):
+            if api_key:
+                # Store API key in session state
+                st.session_state.openai_api_key = api_key
+                # Set API key directly in the openai module
+                openai.api_key = api_key
+                # Also set it in environment variables as a backup
+                os.environ["OPENAI_API_KEY"] = api_key
+                st.success("API Key set successfully!")
+                logger.info("API key configured")
+            else:
+                st.error("Please enter an API key")
+        
+        # Display API key status
+        if st.session_state.openai_api_key:
+            st.info(f"API Key Status: ✅ Set")
+        else:
+            st.warning(f"API Key Status: ❌ Not Set")
         
         # Model selection
         model = st.selectbox(
             "Select LLM Model",
-            ["gpt-3.5-turbo", "gpt-4", "gpt-4-turbo"]
+            ["gpt-4o", "gpt-3.5-turbo", "gpt-4", "gpt-4-turbo"],
+            index=0  # Set gpt-4o as default
         )
         
         # Temperature setting
@@ -57,7 +70,30 @@ def main():
         
         # Batch processing settings
         batch_size = st.number_input("Batch Size", min_value=1, max_value=100, value=10, step=1)
-        
+    
+    # Test API key connection
+    if st.session_state.openai_api_key:
+        if st.button("Test API Connection"):
+            with st.spinner("Testing API connection..."):
+                try:
+                    # Ensure API key is set
+                    openai.api_key = st.session_state.openai_api_key
+                    
+                    # Make a simple API call
+                    response = openai.chat.completions.create(
+                        model="gpt-3.5-turbo",
+                        messages=[
+                            {"role": "system", "content": "You are a helpful assistant."},
+                            {"role": "user", "content": "Hello, this is a test."}
+                        ],
+                        max_tokens=10
+                    )
+                    st.success("✅ API connection successful!")
+                    logger.info("API connection test successful")
+                except Exception as e:
+                    st.error(f"❌ API connection failed: {str(e)}")
+                    logger.error(f"API connection test failed: {str(e)}")
+    
     # Main area
     uploaded_file = st.file_uploader("Upload Excel file with customer feedback", type=["xlsx"])
     
@@ -91,16 +127,16 @@ def main():
             
             # Process button
             if st.button("Process Feedback"):
-                if not st.session_state.api_key_set or not api_key:
-                    st.error("Please enter your OpenAI API key in the sidebar first.")
+                if not st.session_state.openai_api_key:
+                    st.error("Please set your OpenAI API key in the sidebar first.")
                     logger.error("Processing attempted without API key")
                 else:
-                    # Double-check API key is set
-                    if not openai.api_key:
-                        openai.api_key = api_key
+                    # Ensure API key is set in the openai module
+                    openai.api_key = st.session_state.openai_api_key
                     
                     # Log the API key status (without revealing the key)
                     logger.info(f"API key status: {'Set' if openai.api_key else 'Not set'}")
+                    logger.info(f"Using model: {model}")
                     
                     # Create configuration
                     config = FeedbackAnalysisConfig(