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)}")