import streamlit as st
import google.generativeai as genai

# Configure Gemini API
genai.configure(api_key=st.secrets["GEMINI_API_KEY"])

st.title("Embedding Test")

# 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("📌 Generated Embedding")
                st.write(f"**Embedding Dimension:** {len(embedding)}")
                st.code(str(embedding))
                
                st.success("Embedding generated successfully!")
                
            except Exception as e:
                st.error(f"Error generating embedding: {str(e)}")