Update sherlock2.py

sherlock2.py

@@ -1,4 +1,5 @@
 import google.generativeai as genai
+import google.ai.generativelanguage as glm
 import streamlit as st
 from bs4 import BeautifulSoup
 import wikipedia
@@ -35,19 +36,19 @@ sherlock_guidelines = """
 * Be confident in your deductions but remain open to new information and alternative perspectives.
 """
 
-# Generate embeddings for Sherlock Holmes corpus (models/embedding-001)
-embedding_model = genai.EmbeddingModel('models/embedding-001')
+# Generate embeddings using the Gemini Embedding API
+embed_model = 'models/embedding-001'
 
 # Function for embedding generation (using models/embedding-001)
 def generate_embeddings_from_documents(extracted_text):
-    """Generates embeddings for a list of extracted text documents using the 'models/embedding-001' model 
+    """Generates embeddings for a list of extracted text documents using the 'models/embedding-001' model
     and the appropriate task type."""
     embeddings = []
     for text in extracted_text:
         try:
             # Determine the appropriate task type (e.g., "RETRIEVAL_DOCUMENT" for search/similarity)
-            task_type = "RETRIEVAL_DOCUMENT" 
-            response = embedding_model.embed_text(text, task_type=task_type)
+            task_type = "RETRIEVAL_DOCUMENT"
+            response = genai.embed_content(model=embed_model, content=text, task_type=task_type)
             embeddings.append(response["embedding"])
         except Exception as e:
             st.error(f"Error generating embeddings: {e}")