Remove API key text inputs, use environment variables instead

streamlit_app.py

@@ -77,10 +77,12 @@ def generate_aoe_response(model, tokenizer, prompt, max_length=512):
     except Exception as e:
         return f"Error generating AoE response: {str(e)}"
 
-def query_gpt4_api(prompt: str, api_key: Optional[str] = None) -> str:
-    """Query GPT-4 API (placeholder - requires API key)"""
+def query_gpt4_api(prompt: str) -> str:
+    """Query GPT-4 API using environment variable for API key"""
+    api_key = os.getenv('OPENAI_API_KEY')
+    
     if not api_key:
-        return "❌ GPT-4 API key not configured. Please add your OpenAI API key to use GPT-4."
+        return "❌ GPT-4 API key not found in environment variables. Please set OPENAI_API_KEY environment variable to use GPT-4."
     
     try:
         # This is a placeholder implementation - would need actual OpenAI API integration
@@ -88,10 +90,12 @@ def query_gpt4_api(prompt: str, api_key: Optional[str] = None) -> str:
     except Exception as e:
         return f"Error querying GPT-4: {str(e)}"
 
-def query_gemini_api(prompt: str, api_key: Optional[str] = None) -> str:
-    """Query Gemini API (placeholder - requires API key)"""
+def query_gemini_api(prompt: str) -> str:
+    """Query Gemini API using environment variable for API key"""
+    api_key = os.getenv('GOOGLE_API_KEY')
+    
     if not api_key:
-        return "❌ Gemini API key not configured. Please add your Google API key to use Gemini."
+        return "❌ Gemini API key not found in environment variables. Please set GOOGLE_API_KEY environment variable to use Gemini."
     
     try:
         # This is a placeholder implementation - would need actual Google Gemini API integration
@@ -125,23 +129,9 @@ def main():
     st.markdown("---")
     st.subheader("🔧 Configuration")
     
-    col1, col2, col3 = st.columns(3)
+    col1, col2 = st.columns(2)
     
     with col1:
-        openai_api_key = st.text_input(
-            "OpenAI API Key (for GPT-4)",
-            type="password",
-            help="Enter your OpenAI API key to enable GPT-4 responses"
-        )
-    
-    with col2:
-        google_api_key = st.text_input(
-            "Google API Key (for Gemini)",
-            type="password",
-            help="Enter your Google API key to enable Gemini responses"
-        )
-    
-    with col3:
         max_length = st.slider(
             "Max Response Length",
             min_value=100,
@@ -151,6 +141,13 @@ def main():
             help="Maximum length for generated responses"
         )
     
+    with col2:
+        # Display API key status
+        openai_key_status = "✅ Found" if os.getenv('OPENAI_API_KEY') else "❌ Missing"
+        google_key_status = "✅ Found" if os.getenv('GOOGLE_API_KEY') else "❌ Missing"
+        
+        st.info(f"**API Key Status:**\n\nOpenAI API Key: {openai_key_status}\n\nGoogle API Key: {google_key_status}")
+    
     # Main comparison interface
     st.markdown("---")
     st.subheader("💬 Compare LLM Responses")
@@ -174,14 +171,14 @@ def main():
             with col1:
                 st.markdown("### 🤖 GPT-4")
                 with st.spinner("Generating GPT-4 response..."):
-                    gpt4_response = query_gpt4_api(user_prompt, openai_api_key)
+                    gpt4_response = query_gpt4_api(user_prompt)
                 st.markdown("**Response:**")
                 st.write(gpt4_response)
             
             with col2:
                 st.markdown("### 🌟 Gemini")
                 with st.spinner("Generating Gemini response..."):
-                    gemini_response = query_gemini_api(user_prompt, google_api_key)
+                    gemini_response = query_gemini_api(user_prompt)
                 st.markdown("**Response:**")
                 st.write(gemini_response)
             
@@ -205,11 +202,13 @@ def main():
         st.header("ℹ️ Model Information")
         
         st.markdown("**🤖 GPT-4**")
-        st.write(f"Status: {'✅ Configured' if openai_api_key else '❌ API key needed'}")
+        openai_status = "✅ Configured" if os.getenv('OPENAI_API_KEY') else "❌ Environment variable OPENAI_API_KEY not set"
+        st.write(f"Status: {openai_status}")
         st.write("Provider: OpenAI")
         
         st.markdown("**🌟 Gemini**")
-        st.write(f"Status: {'✅ Configured' if google_api_key else '❌ API key needed'}")
+        google_status = "✅ Configured" if os.getenv('GOOGLE_API_KEY') else "❌ Environment variable GOOGLE_API_KEY not set"
+        st.write(f"Status: {google_status}")
         st.write("Provider: Google")
         
         st.markdown("**🏰 AOE (Local)**")
@@ -228,14 +227,14 @@ def main():
         
         st.markdown("---")
         st.markdown("**📋 Instructions:**")
-        st.markdown("1. Configure API keys for GPT-4 and Gemini")
+        st.markdown("1. Set OPENAI_API_KEY and GOOGLE_API_KEY environment variables")
         st.markdown("2. Enter your prompt in the text area")
         st.markdown("3. Click 'Generate All Responses'")
         st.markdown("4. Compare responses side by side")
         
         st.markdown("---")
         st.markdown("**⚠️ Notes:**")
-        st.markdown("- GPT-4 and Gemini require valid API keys")
+        st.markdown("- GPT-4 and Gemini require valid API keys in environment variables")
         st.markdown("- AOE model runs locally from outputs/student/")
         st.markdown("- Responses are generated independently")