Update app.py

app.py

@@ -34,10 +34,22 @@ daily_queries = [0, 0, 0, 0, 0, 6, 8, 10, 7, 9, 12, 15, 11, 14]  # Mock data for
 
 # Initialize Gemini API with service account credentials
 def init_gemini_api():
-    """Initialize Google Gemini API with service account credentials"""
+    """Initialize Google Gemini API with service account credentials from Hugging Face Secrets"""
     try:
-        # Load credentials from service account JSON file
-        credentials_path = os.getenv("GOOGLE_APPLICATION_CREDENTIALS")
+        # Retrieve service account JSON from Hugging Face Secrets
+        service_account_json = os.getenv("SERVICE_ACCOUNT_JSON")
+        if not service_account_json:
+            raise ValueError("SERVICE_ACCOUNT_JSON is not set in environment variables")
+            
+        # Convert the single-line string back to JSON format
+        service_account_dict = json.loads(service_account_json)
+        
+        # Write it to a temporary JSON file
+        credentials_path = "service_account.json"
+        with open(credentials_path, "w") as f:
+            json.dump(service_account_dict, f)
+        
+        # Load credentials from the temporary file
         credentials = service_account.Credentials.from_service_account_file(
             credentials_path,
             scopes=["https://www.googleapis.com/auth/cloud-platform"]
@@ -45,15 +57,18 @@ def init_gemini_api():
         
         # Configure Gemini API with credentials
         genai.configure(credentials=credentials)
-        print("Gemini API initialized with service account credentials")
+        print("Gemini API initialized with service account credentials from Hugging Face Secrets")
         return True
+        
     except Exception as e:
-        print(f"Error initializing Gemini API: {e}")
+        print(f"Error initializing Gemini API with service account: {e}")
+        
         # Fallback to API key method if service account fails
         try:
             genai.configure(api_key=os.getenv("GEMINI_API_KEY", ""))
             print("Gemini API initialized with API key")
             return True
+            
         except Exception as e2:
             print(f"Fallback to API key also failed: {e2}")
             return False