Spaces:
Sleeping
Sleeping
Update process_interview.py
Browse files- process_interview.py +18 -4
process_interview.py
CHANGED
|
@@ -54,12 +54,23 @@ ASSEMBLYAI_KEY = os.getenv("ASSEMBLYAI_KEY")
|
|
| 54 |
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|
| 55 |
|
| 56 |
|
| 57 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
def initialize_services():
|
| 59 |
try:
|
| 60 |
pc = Pinecone(api_key=PINECONE_KEY)
|
| 61 |
index_name = "interview-speaker-embeddings"
|
| 62 |
-
|
|
|
|
|
|
|
| 63 |
pc.create_index(
|
| 64 |
name=index_name,
|
| 65 |
dimension=192,
|
|
@@ -69,12 +80,15 @@ def initialize_services():
|
|
| 69 |
index = pc.Index(index_name)
|
| 70 |
|
| 71 |
genai.configure(api_key=GEMINI_API_KEY)
|
| 72 |
-
|
|
|
|
| 73 |
|
|
|
|
| 74 |
return index, gemini_model
|
|
|
|
| 75 |
except Exception as e:
|
| 76 |
logger.error(f"Error initializing services: {str(e)}")
|
| 77 |
-
raise
|
| 78 |
|
| 79 |
|
| 80 |
index, gemini_model = initialize_services()
|
|
|
|
| 54 |
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|
| 55 |
|
| 56 |
|
| 57 |
+
import os
|
| 58 |
+
import logging
|
| 59 |
+
from pinecone import Pinecone, ServerlessSpec
|
| 60 |
+
import google.generativeai as genai
|
| 61 |
+
|
| 62 |
+
logger = logging.getLogger(__name__)
|
| 63 |
+
|
| 64 |
+
PINECONE_KEY = os.getenv("PINECONE_API_KEY")
|
| 65 |
+
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|
| 66 |
+
|
| 67 |
def initialize_services():
|
| 68 |
try:
|
| 69 |
pc = Pinecone(api_key=PINECONE_KEY)
|
| 70 |
index_name = "interview-speaker-embeddings"
|
| 71 |
+
|
| 72 |
+
existing_indexes = [index_info["name"] for index_info in pc.list_indexes()]
|
| 73 |
+
if index_name not in existing_indexes:
|
| 74 |
pc.create_index(
|
| 75 |
name=index_name,
|
| 76 |
dimension=192,
|
|
|
|
| 80 |
index = pc.Index(index_name)
|
| 81 |
|
| 82 |
genai.configure(api_key=GEMINI_API_KEY)
|
| 83 |
+
|
| 84 |
+
gemini_model = genai.GenerativeModel('gemini-1.5-flash-latest')
|
| 85 |
|
| 86 |
+
logger.info("Services initialized successfully.")
|
| 87 |
return index, gemini_model
|
| 88 |
+
|
| 89 |
except Exception as e:
|
| 90 |
logger.error(f"Error initializing services: {str(e)}")
|
| 91 |
+
raise e
|
| 92 |
|
| 93 |
|
| 94 |
index, gemini_model = initialize_services()
|