Spaces:
Sleeping
Sleeping
File size: 3,845 Bytes
5bd0c5e 1fa6e70 5bd0c5e fa6de7f 9209303 1fa6e70 9209303 1fa6e70 5bd0c5e fa6de7f 086f49f 308b460 086f49f 308b460 086f49f fa6de7f 086f49f 9209303 086f49f fa6de7f 086f49f 308b460 5fb6963 1fa6e70 5bd0c5e 9209303 5bd0c5e 1fa6e70 9209303 5bd0c5e fa6de7f 9209303 1fa6e70 5bd0c5e 1fa6e70 5bd0c5e fa6de7f 1fa6e70 5bd0c5e 9209303 5bd0c5e 9209303 5bd0c5e fa6de7f 1fa6e70 5bd0c5e 1fa6e70 |
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 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
import streamlit as st
from google import genai
# ---------------- Page Config ----------------
st.set_page_config(page_title="Gemini API Tester", page_icon="⚡")
st.title("⚡ Gemini API Tester")
# ---------------- Session State Defaults ----------------
defaults = {
"api_key_input": "",
"api_key": None,
"api_validated": False,
"client": None,
"models": [],
"selected_model": None,
"messages": []
}
for k, v in defaults.items():
if k not in st.session_state:
st.session_state[k] = v
# ---------------- Sidebar ----------------
with st.sidebar:
st.header("🔐 API Configuration")
if not st.session_state.api_validated:
with st.form("api_form", clear_on_submit=False):
api_key_input = st.text_input(
"Enter Google API Key",
type="password",
key="api_key_input"
)
submitted = st.form_submit_button("Validate API Key")
if submitted:
if not api_key_input:
st.warning("Please enter an API key.")
else:
try:
client = genai.Client(api_key=api_key_input)
try:
models = [
m.name for m in client.models.list()
if "gemini" in m.name.lower()
]
except Exception:
models = []
st.session_state.api_key = api_key_input
st.session_state.client = client
st.session_state.models = models or ["gemini-2.5-flash"]
st.session_state.selected_model = st.session_state.models[0]
st.session_state.api_validated = True
st.success("✅ API key validated successfully")
except Exception as e:
error_msg = str(e).lower()
if "401" in error_msg or "403" in error_msg or "unauthorized" in error_msg:
st.error("❌ Invalid API key")
else:
st.warning("⚠️ API key accepted, but model access may be restricted")
st.caption(str(e))
else:
st.success("🔒 API Key validated (hidden)")
st.selectbox(
"Available Models",
st.session_state.models,
key="selected_model"
)
if st.button("🔄 Change API Key"):
for k in defaults:
st.session_state[k] = defaults[k]
# ---------------- Chat History ----------------
for msg in st.session_state.messages:
with st.chat_message(msg["role"]):
st.markdown(msg["content"])
# ---------------- Chat Input ----------------
if prompt := st.chat_input("Ask something..."):
if not st.session_state.api_validated:
st.error("Please validate an API key first.")
st.stop()
# User message
st.session_state.messages.append(
{"role": "user", "content": prompt}
)
with st.chat_message("user"):
st.markdown(prompt)
# Assistant response
with st.chat_message("assistant"):
placeholder = st.empty()
placeholder.markdown("Thinking...")
try:
response = st.session_state.client.models.generate_content(
model=st.session_state.selected_model,
contents=prompt
)
reply = response.text
placeholder.markdown(reply)
st.session_state.messages.append(
{"role": "assistant", "content": reply}
)
except Exception as e:
placeholder.error(f"Error: {e}")
|