updated gemini service creds

.gitignore

@@ -181,4 +181,4 @@ __pycache__/
 gradio_cached_examples/
 models/__pycache__/
 setup/__pycache__/
-setup/gcp-default-creds.json
+setup/gcp_default_creds.json

app.py

@@ -20,7 +20,7 @@ def generate_text(query_text, model_name: Optional[str] = "google/gemma-2b-it"):
 
 examples = [["I'm planning a vacation to France. Can you suggest a one-week itinerary including must-visit places and "
              "local cuisines to try?", "google/gemma-2b-it"],
-            # ["I want to explore off-the-beaten-path destinations in Europe, any suggestions?", "gemini-1.0-pro"],
+            ["I want to explore off-the-beaten-path destinations in Europe, any suggestions?", "gemini-1.0-pro"],
             ["Suggest some cities that can be visited from London and are very rich in history and culture.",
              "google/gemma-2b-it"],
             ]

setup/convert_creds.py

@@ -0,0 +1,26 @@
+import json
+import base64
+import os
+import dotenv
+
+# load the service account key
+with open(os.getcwd() +'/gcp_default_creds.json') as f:
+    service_key = json.load(f)
+
+dotenv_file = dotenv.find_dotenv()
+dotenv.load_dotenv(dotenv_file)
+
+# convert json to a string
+service_key = json.dumps(service_key)
+
+# encode service key
+encoded_service_key = base64.b64encode(service_key.encode('utf-8'))
+encoded_service_key = str(encoded_service_key)[2:-1]
+
+print(encoded_service_key)
+# b'many_characters_here'
+os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = encoded_service_key
+print(os.environ['GOOGLE_APPLICATION_CREDENTIALS'])  # outputs 'newvalue'
+
+# Write changes to .env file.
+dotenv.set_key(dotenv_file, "GOOGLE_APPLICATION_CREDENTIALS", os.environ["GOOGLE_APPLICATION_CREDENTIALS"])

setup/vertex_ai_setup.py

@@ -3,16 +3,24 @@ from dotenv import load_dotenv
 from google.oauth2 import service_account
 import vertexai
 import os
+import json
+import base64
 
 load_dotenv()
 
 # TODO: fix it in spaces
-os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "gcp-default-creds.json"
-GOOGLE_APPLICATION_CREDENTIALS = os.environ["GOOGLE_APPLICATION_CREDENTIALS"]
+
 VERTEXAI_PROJECT = os.environ["VERTEXAI_PROJECT"]
 
 
+def decode_service_key():
+    encoded_key = os.environ["GOOGLE_APPLICATION_CREDENTIALS"]
+    original_service_key = json.loads(base64.b64decode(encoded_key).decode('utf-8'))
+    return original_service_key
+
+
 def initialize_vertexai_params(location: Optional[str] = "us-central1"):
+    GOOGLE_APPLICATION_CREDENTIALS = decode_service_key()
     service_account.Credentials.from_service_account_file(
         filename=GOOGLE_APPLICATION_CREDENTIALS,
         scopes=["https://www.googleapis.com/auth/cloud-platform"],