embedding_testq / app.py
Sebbe33's picture
Create app.py
f9b4a02 verified
raw
history blame
1.53 kB
import streamlit as st
import google.generativeai as genai
# Configure Gemini API
genai.configure(api_key=st.secrets["GEMINI_API_KEY"])
st.title("🦾 Gemini Text Embedding Generator")
st.markdown("Generate text embeddings using Google's Gemini AI")
# Text input area
input_text = st.text_area("Enter your text to generate embedding:",
height=150,
placeholder="Type your text here...")
# Button to generate embedding
if st.button("Generate Embedding"):
if not input_text.strip():
st.warning("Please enter some text to generate embedding.")
else:
with st.spinner("Generating embedding..."):
try:
# Generate embedding
result = genai.embed_content(
model="models/text-embedding-004",
content=input_text
)
embedding = result['embedding']
# Display results
st.subheader("πŸ“ Input Text")
st.write(input_text)
st.subheader("πŸ“Œ Generated Embedding")
st.write(f"**Embedding Dimension:** {len(embedding)}")
st.write("**First 10 values preview:**")
st.code(str(embedding[:10]))
st.success("Embedding generated successfully!")
except Exception as e:
st.error(f"Error generating embedding: {str(e)}")