Sebbe33 commited on
Commit
f9b4a02
·
verified ·
1 Parent(s): 3707a4f

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +42 -0
app.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import google.generativeai as genai
3
+
4
+ # Configure Gemini API
5
+ genai.configure(api_key=st.secrets["GEMINI_API_KEY"])
6
+
7
+ st.title("🦾 Gemini Text Embedding Generator")
8
+ st.markdown("Generate text embeddings using Google's Gemini AI")
9
+
10
+ # Text input area
11
+ input_text = st.text_area("Enter your text to generate embedding:",
12
+ height=150,
13
+ placeholder="Type your text here...")
14
+
15
+ # Button to generate embedding
16
+ if st.button("Generate Embedding"):
17
+ if not input_text.strip():
18
+ st.warning("Please enter some text to generate embedding.")
19
+ else:
20
+ with st.spinner("Generating embedding..."):
21
+ try:
22
+ # Generate embedding
23
+ result = genai.embed_content(
24
+ model="models/text-embedding-004",
25
+ content=input_text
26
+ )
27
+
28
+ embedding = result['embedding']
29
+
30
+ # Display results
31
+ st.subheader("📝 Input Text")
32
+ st.write(input_text)
33
+
34
+ st.subheader("📌 Generated Embedding")
35
+ st.write(f"**Embedding Dimension:** {len(embedding)}")
36
+ st.write("**First 10 values preview:**")
37
+ st.code(str(embedding[:10]))
38
+
39
+ st.success("Embedding generated successfully!")
40
+
41
+ except Exception as e:
42
+ st.error(f"Error generating embedding: {str(e)}")