Update app.py

app.py

@@ -7,11 +7,23 @@ from torchvision import models, transforms
 import json
 import requests
 
-# Load credentials from Hugging Face's Secret Manager
-hf_token = os.environ.get("HF_TOKEN")  # Assuming the Hugging Face API token is set in environment
+# Load credentials (stringified JSON) from environment variable
+credentials_string = os.environ.get("GOOGLE_APPLICATION_CREDENTIALS")
+if not credentials_string:
+    raise ValueError("GOOGLE_APPLICATION_CREDENTIALS is not set in the environment!")
 
-# Initialize Gemini model using Hugging Face
-llm = ChatGoogleGenerativeAI(model='gemini-1.5-pro')  # You can change 'gemini-1.5-pro' to the specific model you need
+# Parse the stringified JSON back to a Python dictionary
+credentials = json.loads(credentials_string)
+
+# Save the credentials to a temporary JSON file (required by Google SDKs)
+with open("service_account.json", "w") as f:
+    json.dump(credentials, f)
+
+# Set the GOOGLE_APPLICATION_CREDENTIALS environment variable to the temporary file
+os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "service_account.json"
+
+# Initialize Gemini model
+llm = ChatGoogleGenerativeAI(model='gemini-1.5-pro')
 
 # Load a pre-trained ResNet50 model for image classification
 model = models.resnet50(pretrained=True)