Update src/services/utils.py

src/services/utils.py

@@ -6,7 +6,15 @@ import nltk
 from nltk.stem import *
 nltk.download("punkt_tab")
 
-FILE_PATH = "/app/src/ressources/technologies_database.xlsx"
+from pathlib import Path
+import os
+import google.generativeai as genai
+import json
+from google.genai import Client, types
+
+BASE_DIR = Path(__file__).resolve().parent.parent
+
+FILE_PATH = BASE_DIR / 'ressources' / 'global_tech_embeddings.pkl'
 
 def set_prompt(problem):
     prompt = """
@@ -48,7 +56,7 @@ def load_technologies_excel():
     return df
 
 def load_technologies():
-    EMBEDDINGS_FILE = '/app/src/ressources/global_tech_embeddings.pkl'
+    EMBEDDINGS_FILE = FILE_PATH
 
     try:
         with open(EMBEDDINGS_FILE, 'rb') as f:
@@ -147,4 +155,21 @@ def save_to_pickle(result_similarites):
         pickle.dump(data_to_save, f)
 
     print(f"\nMatrix and labels saved to {output_filename}")
-    return output_filename
+    return output_filename
+
+    
+def set_gemini():
+    gemini_api = os.getenv("GEMINI_API")
+    client = Client(api_key=gemini_api)
+
+    # Define the grounding tool
+    grounding_tool = types.Tool(
+        google_search=types.GoogleSearch()
+    )
+
+    # Configure generation settings
+    config = types.GenerateContentConfig(
+        tools=[grounding_tool]
+    )
+
+    return client,config