Update utils.py

utils.py

@@ -5,7 +5,23 @@ import streamlit as st
 import openai
 
 def setup_gcp_auth():
-    """Setup GCP authentication from HF Spaces, environment variables, or Streamlit secrets."""
+    """
+    Setup Google Cloud Platform authentication using various methods.
+    
+    This function tries multiple authentication methods in order of preference:
+    1. HF Spaces environment variable (GCP_CREDENTIALS)
+    2. Local environment variable pointing to a credentials file (GOOGLE_APPLICATION_CREDENTIALS)
+    3. Streamlit secrets (gcp_credentials)
+    
+    Each method is tried in sequence until one succeeds.
+    
+    Returns:
+        google.oauth2.service_account.Credentials: GCP credentials object
+        
+    Raises:
+        ValueError: If no valid credentials are found
+        Exception: For any authentication errors
+    """
     try:
         # Option 1: HF Spaces environment variable
         if "GCP_CREDENTIALS" in os.environ:
@@ -59,7 +75,20 @@ def setup_gcp_auth():
         raise
 
 def setup_openai_auth():
-    """Setup OpenAI API authentication from environment variables or Streamlit secrets."""
+    """
+    Setup OpenAI API authentication using various methods.
+    
+    This function tries multiple authentication methods in order of preference:
+    1. Standard environment variable (OPENAI_API_KEY)
+    2. HF Spaces environment variable (OPENAI_KEY)
+    3. Streamlit secrets (openai_api_key)
+    
+    Each method is tried in sequence until one succeeds.
+    
+    Raises:
+        ValueError: If no valid API key is found
+        Exception: For any authentication errors
+    """
     try:
         # Option 1: Standard environment variable
         if "OPENAI_API_KEY" in os.environ:
@@ -89,7 +118,15 @@ def setup_openai_auth():
         raise
 
 def setup_all_auth():
-    """Setup all authentication in one call"""
+    """
+    Setup all authentication in one call.
+    
+    This is a convenience function that sets up both GCP and OpenAI authentication.
+    It's typically called during Anveshak's initialization.
+    
+    Returns:
+        google.oauth2.service_account.Credentials: GCP credentials object
+    """
     gcp_creds = setup_gcp_auth()
     setup_openai_auth()
     return gcp_creds