Spaces:
Sleeping
Sleeping
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)}") |