Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import logging
 from dotenv import load_dotenv
 import subprocess
+import google.auth
 
 # Force reinstall Gradio
 subprocess.run(["pip", "install", "--upgrade", "gradio==4.44.1"])
@@ -42,18 +43,25 @@ set_debug(True)
 
 # process of getting credentials
 def get_credentials():
-    creds_json_str = os.getenv("BOB") # get json credentials stored as a string
+    """Retrieve Google Cloud credentials from the environment variable and write them to a temporary file."""
+    creds_json_str = os.getenv("BOB")  # Get JSON credentials stored as a string
     if creds_json_str is None:
-        raise ValueError("GOOGLE_APPLICATION_CREDENTIALS_JSON not found in environment")
+        raise ValueError("Environment variable 'BOB' not found. Please set it with the JSON credentials.")
 
-    # create a temporary file
-    with tempfile.NamedTemporaryFile(mode="w+", delete=False, suffix=".json") as temp:
-        temp.write(creds_json_str) # write in json format
-        temp_filename = temp.name 
-
-    return temp_filename
+    try:
+        # Create a temporary file to store the credentials
+        with tempfile.NamedTemporaryFile(mode="w+", delete=False, suffix=".json") as temp:
+            temp.write(creds_json_str)  # Write the JSON string to the file
+            temp_filename = temp.name  # Get the temporary file's name
+            logging.info(f"Temporary credentials file created at: {temp_filename}")
+            return temp_filename
+    except Exception as e:
+        logging.error(f"Error creating temporary credentials file: {e}")
+        raise
     
-os.environ["GOOGLE_APPLICATION_CREDENTIALS"]= get_credentials()
+# Store the temporary file path
+temp_credentials_file = get_credentials()
+os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = temp_credentials_file
 
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
@@ -63,8 +71,10 @@ NEO4J_USERNAME = os.getenv("NEO4J_USERNAME")
 NEO4J_PASSWORD = os.getenv("NEO4J_PASSWORD")
 NEO4J_DATABASE = os.getenv("NEO4J_DATABASE")
 
-project_id = os.getenv("PROJECT_ID")
-credentials = get_credentials()
+# Load credentials using google.auth
+credentials, project_id = google.auth.default()
+logging.info(f"Loaded credentials for project: {project_id}")
+
 embedding_function = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
 
 ### Vector graph search