Update generator.py

generator.py

@@ -10,19 +10,68 @@ import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# def get_api_keys(service_name: str, key_names: List[str]) -> List[str]:
+#     """Get API keys from multiple sources"""
+#     keys = []
+    
+#     # 1. HuggingFace Spaces Secrets (Primary)
+#     for key_name in key_names:
+#         # Try HF-specific naming first
+#         hf_key_name = f"HF_{service_name.upper()}_{key_name}"
+#         key = os.getenv(hf_key_name)
+#         if key and key.strip():
+#             keys.append(key.strip())
+#             logger.info(f"✅ Found {service_name} key in HuggingFace secrets")
+#             break  # Use first valid key
+    
+#     # 2. Standard Environment Variables (fallback)
+#     if not keys:
+#         for key_name in key_names:
+#             key = os.getenv(key_name) or os.getenv(key_name.upper())
+#             if key and key.strip():
+#                 keys.append(key.strip())
+#                 logger.info(f"✅ Found {service_name} key in environment")
+#                 break
+    
+#     # 3. Streamlit Secrets (last resort)
+#     if not keys:
+#         try:
+#             import streamlit as st
+#             if hasattr(st, 'secrets') and service_name in st.secrets:
+#                 secrets = st.secrets[service_name]
+#                 for key_name in key_names:
+#                     key = secrets.get(key_name)
+#                     if key and key.strip():
+#                         keys.append(key.strip())
+#                         logger.info(f"✅ Found {service_name} key in Streamlit secrets")
+#                         break
+#         except Exception:
+#             pass
+    
+#     if not keys:
+#         logger.warning(f"❌ No {service_name} API keys found")
+    
+#     return keys
+
 def get_api_keys(service_name: str, key_names: List[str]) -> List[str]:
-    """Get API keys from multiple sources"""
+    """Get API keys from multiple sources with priority for HuggingFace Spaces"""
     keys = []
     
-    # 1. HuggingFace Spaces Secrets (Primary)
-    for key_name in key_names:
-        # Try HF-specific naming first
-        hf_key_name = f"HF_{service_name.upper()}_{key_name}"
+    # 1. HuggingFace Spaces Secrets (Primary) - multiple keys
+    for i in range(1, 6):  # Check for HF_GROQ_API_KEY_1 through HF_GROQ_API_KEY_5
+        hf_key_name = f"HF_{service_name.upper()}_API_KEY_{i}"
         key = os.getenv(hf_key_name)
+        if key and key.strip():
+            keys.append(key.strip())
+            logger.info(f"✅ Found {service_name} key {i} in HuggingFace secrets")
+    
+    # Also check for single key (backward compatibility)
+    if not keys:
+        single_key_name = f"HF_{service_name.upper()}_API_KEY"
+        key = os.getenv(single_key_name)
         if key and key.strip():
             keys.append(key.strip())
             logger.info(f"✅ Found {service_name} key in HuggingFace secrets")
-            break  # Use first valid key
     
     # 2. Standard Environment Variables (fallback)
     if not keys:
@@ -31,7 +80,6 @@ def get_api_keys(service_name: str, key_names: List[str]) -> List[str]:
             if key and key.strip():
                 keys.append(key.strip())
                 logger.info(f"✅ Found {service_name} key in environment")
-                break
     
     # 3. Streamlit Secrets (last resort)
     if not keys:
@@ -44,18 +92,24 @@ def get_api_keys(service_name: str, key_names: List[str]) -> List[str]:
                     if key and key.strip():
                         keys.append(key.strip())
                         logger.info(f"✅ Found {service_name} key in Streamlit secrets")
-                        break
         except Exception:
             pass
     
     if not keys:
         logger.warning(f"❌ No {service_name} API keys found")
+    else:
+        logger.info(f"✅ Found {len(keys)} {service_name} API keys")
     
     return keys
 
+# def get_groq_api_keys():
+#     """Get Groq API keys for all environments"""
+#     return get_api_keys("groq", ["api_key", "api_key_1", "api_key_2"])
+
 def get_groq_api_keys():
     """Get Groq API keys for all environments"""
-    return get_api_keys("groq", ["api_key", "api_key_1", "api_key_2"])
+    # Look for multiple HF keys first, then fallback
+    return get_api_keys("groq", ["api_key", "api_key_1", "api_key_2", "api_key_3"])
 
 def get_ollama_url():
     """Get Ollama URL from multiple sources"""