Update app.py

app.py

@@ -4,7 +4,7 @@ import json, os, vertexai, wandb
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
-credentials = os.environ["CREDENTIALS"]
+//credentials = os.environ["CREDENTIALS"]
 project = os.environ["PROJECT"]
 wandb_api_key = os.environ["WANDB_API_KEY"]
 
@@ -16,29 +16,37 @@ wandb.login(key = wandb_api_key)
 wandb.init(project = "vertext-ai-txt", config = config)
 config = wandb.config
 
-credentials = json.loads(credentials)
+//credentials = json.loads(credentials)
 
 from google.oauth2 import service_account
-credentials = service_account.Credentials.from_service_account_info(credentials)
+//credentials = service_account.Credentials.from_service_account_info(credentials)
 
-if credentials.expired:
-    credentials.refresh(Request())
+//if credentials.expired:
+//    credentials.refresh(Request())
 
-vertexai.init(project = project, 
-              location = "us-central1",
-              credentials = credentials)
+//vertexai.init(project = project, 
+//              location = "us-central1",
+//              credentials = credentials)
 
 from vertexai.language_models import TextGenerationModel
 generation_model = TextGenerationModel.from_pretrained("text-bison@001")
 
-def invoke(prompt):
+def invoke(prompt, credentials):
+    //
+    credentials = service_account.Credentials.from_service_account_info(credentials)
+    if credentials.expired:
+        credentials.refresh(Request())
+    vertexai.init(project = project, 
+                  location = "us-central1",
+                  credentials = credentials)
+    //
     completion = generation_model.predict(prompt = prompt).text
     wandb.log({"prompt": prompt, "completion": completion})
     return completion
 
 gr.close_all()
 demo = gr.Interface(fn=invoke, 
-                    inputs = [gr.Textbox(label = "Prompt", lines = 1)],
+                    inputs = [gr.Textbox(label = "Prompt", lines = 1), gr.Textbox(label = "Credentials", lines = 1)],
                     outputs = [gr.Textbox(label = "Completion", lines = 1)],
                     title = "Generative AI - Language",
                     description = "<a href='https://www.gradio.app/'>Gradio</a> UI using <a href='https://cloud.google.com/vertex-ai?hl=en/'>Google Vertex AI</a> API with Bison foundation model")