norhan12 commited on
Commit
e5f0834
·
verified ·
1 Parent(s): 05245a6

Update process_interview.py

Browse files
Files changed (1) hide show
  1. 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
- # Initialize services
 
 
 
 
 
 
 
 
 
58
  def initialize_services():
59
  try:
60
  pc = Pinecone(api_key=PINECONE_KEY)
61
  index_name = "interview-speaker-embeddings"
62
- if index_name not in pc.list_indexes().names():
 
 
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
- gemini_model = genai.GenerativeModel('gemini-1.5-flash')
 
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()