larrysim commited on
Commit
3fa0141
Β·
verified Β·
1 Parent(s): 1a54c01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -32,6 +32,7 @@ with st.sidebar:
32
  key="api_key_input"
33
  )
34
 
 
35
  if st.button("Validate API Key"):
36
  if not api_key_input:
37
  st.warning("Please enter an API key.")
@@ -41,14 +42,12 @@ with st.sidebar:
41
  client = genai.Client(api_key=api_key_input)
42
 
43
  # Attempt to list models (authentication test)
44
- models = []
45
  try:
46
  models = [
47
  m.name for m in client.models.list()
48
  if "gemini" in m.name.lower()
49
  ]
50
  except Exception:
51
- # Listing models may fail even if key is valid
52
  models = []
53
 
54
  # Persist validated state
@@ -72,12 +71,14 @@ with st.sidebar:
72
  else:
73
  st.success("πŸ”’ API Key validated (hidden)")
74
 
 
75
  st.selectbox(
76
  "Available Models",
77
  st.session_state.models,
78
  key="selected_model"
79
  )
80
 
 
81
  if st.button("πŸ”„ Change API Key"):
82
  for k in defaults:
83
  st.session_state[k] = defaults[k]
 
32
  key="api_key_input"
33
  )
34
 
35
+ # Validate API Key immediately when button is clicked
36
  if st.button("Validate API Key"):
37
  if not api_key_input:
38
  st.warning("Please enter an API key.")
 
42
  client = genai.Client(api_key=api_key_input)
43
 
44
  # Attempt to list models (authentication test)
 
45
  try:
46
  models = [
47
  m.name for m in client.models.list()
48
  if "gemini" in m.name.lower()
49
  ]
50
  except Exception:
 
51
  models = []
52
 
53
  # Persist validated state
 
71
  else:
72
  st.success("πŸ”’ API Key validated (hidden)")
73
 
74
+ # Model selector is now immediately visible
75
  st.selectbox(
76
  "Available Models",
77
  st.session_state.models,
78
  key="selected_model"
79
  )
80
 
81
+ # Allow changing API key
82
  if st.button("πŸ”„ Change API Key"):
83
  for k in defaults:
84
  st.session_state[k] = defaults[k]