File size: 2,579 Bytes
cff1e0e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | """
Step 1: API Configuration
"""
import streamlit as st
from core.workflow import set_openai_api_key
from services.key_manager import get_key_manager
def render_step1():
"""Render Step 1: Input API Keys"""
st.header("Step 1: Input Your API Keys")
st.markdown("Configure API keys for the models you want to use for evaluation. (Optional)")
st.markdown("### 🔑 OpenAI Configuration")
st.session_state.openai_key_input = st.text_input(
"OpenAI API Key",
type="password",
value=st.session_state.get("openai_key_input", ""),
help="Enter your OpenAI API key for GPT model evaluation (optional)",
)
if st.session_state.openai_key_input:
if st.button("Test OpenAI Connection", key="test_openai"):
with st.spinner("Testing connection..."):
try:
from providers.openai_client import OpenAIClient
client = OpenAIClient(st.session_state.openai_key_input)
# Simple test - check if key is valid format
if st.session_state.openai_key_input.startswith("sk-"):
st.success("✅ OpenAI API key format is valid!")
st.session_state.openai_configured = True
st.session_state.openai_key = (
st.session_state.openai_key_input
)
# Save to KeyManager
key_manager = get_key_manager()
key_manager.set_key("openai", st.session_state.openai_key)
set_openai_api_key(
st.session_state.openai_key
) # <-- initialize core
else:
st.error("❌ Invalid OpenAI API key format")
st.session_state.openai_configured = False
except Exception as e:
st.error(f"❌ Error: {str(e)}")
st.session_state.openai_configured = False
if st.session_state.get("openai_configured", False):
st.success("✅ OpenAI is configured and ready!")
# Validation
st.divider()
if st.session_state.get("openai_configured", False):
st.success("✅ API configured! You can proceed.")
else:
st.info("💡 You can proceed without configuring an API key, or configure OpenAI above.")
if st.button("Next: Upload File →", type="primary", use_container_width=True):
st.session_state.step = 2
st.rerun()
|