Fix: Update API key handling for HF Spaces environment variables

- Modified app.py to properly read GROQ_API_KEY from environment variables
- Updated classifier.py with better error handling and logging
- Added fallback for both HF Spaces and local development
- Improved user-friendly error messages with setup instructions
- Switch to llama-3.1-70b-versatile model for better reliability

app.py

@@ -22,32 +22,50 @@ load_dotenv()
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-try:
-    # Try Streamlit Cloud secrets first, then fall back to .env
-    if hasattr(st, 'secrets') and 'GROQ_API_KEY' in st.secrets:
-        os.environ['GROQ_API_KEY'] = st.secrets['GROQ_API_KEY']
-        st.success("🔑 API key loaded from Streamlit Cloud secrets")
-    elif 'GROQ_API_KEY' not in os.environ:
-        st.error("⚠️ GROQ_API_KEY not found!")
-        st.info("**For Streamlit Cloud deployment:**")
-        st.info("Add your API key in the Streamlit Cloud app settings > Secrets tab")
-        st.code("""
-        # In Streamlit Cloud Secrets:
-        GROQ_API_KEY = "your_groq_api_key_here"
-        """)
-        st.info("**For local development:**")
-        st.info("Add GROQ_API_KEY to your .env file")
-        st.code("""
-        # In .env file:
-        GROQ_API_KEY=your_groq_api_key_here
-        """)
-        st.stop()
-    else:
-        st.success("🔑 API key loaded from environment")
-except Exception as e:
-    st.error(f"⚠️ Error accessing API key: {e}")
-    st.error("Please check your configuration")
+# Function to get API key from multiple sources
+def get_groq_api_key():
+    """Get GROQ API key from environment variables or Streamlit secrets"""
+    # Try environment variable first (HF Spaces and Docker way)
+    api_key = os.getenv('GROQ_API_KEY')
+    if api_key:
+        return api_key
+    
+    # Try Streamlit secrets as fallback (for Streamlit Cloud)
+    try:
+        if hasattr(st, 'secrets') and 'GROQ_API_KEY' in st.secrets:
+            return st.secrets['GROQ_API_KEY']
+    except Exception as e:
+        logger.warning(f"Could not access Streamlit secrets: {e}")
+    
+    return None
+
+# Check API key availability
+groq_api_key = get_groq_api_key()
+
+if not groq_api_key:
+    st.error("🚨 **GROQ API Key Missing!**")
+    st.markdown("""
+    ### Please add your GROQ API key:
+    
+    **For Hugging Face Spaces:**
+    1. Go to your Space settings tab
+    2. Scroll to "Repository secrets" 
+    3. Add secret: `GROQ_API_KEY` = `your_actual_key`
+    4. Restart the space
+    
+    **For local development:**
+    Add to `.streamlit/secrets.toml`:
+    ```toml
+    GROQ_API_KEY = "your_groq_api_key_here"
+    ```
+    
+    **Get your API key:** https://console.groq.com/keys
+    """)
     st.stop()
+else:
+    # Set the API key in environment for other modules
+    os.environ['GROQ_API_KEY'] = groq_api_key
+    st.success("🔑 API key loaded successfully")
 
 try:
     from models import Ticket, TicketClassification, TopicTagEnum, SentimentEnum, PriorityEnum

classifier.py

@@ -10,15 +10,23 @@ logger = logging.getLogger(__name__)
 
 class TicketClassifier:
     def __init__(self):
+        # Get API key from environment
         api_key = os.getenv("GROQ_API_KEY")
         if not api_key:
-            raise ValueError("GROQ_API_KEY environment variable is required")
+            logger.error("GROQ_API_KEY environment variable not found")
+            raise ValueError("GROQ_API_KEY environment variable is required. Please set it in your HF Spaces secrets or local environment.")
         
-        self.client = Groq(api_key=api_key)
-        self.models = [
-            "moonshotai/kimi-k2-instruct"
-        ]
-        self.model = "moonshotai/kimi-k2-instruct"
+        try:
+            self.client = Groq(api_key=api_key)
+            # Use the more reliable model for classification
+            self.models = [
+                "llama-3.1-70b-versatile"
+            ]
+            self.model = "llama-3.1-70b-versatile"
+            logger.info("TicketClassifier initialized successfully with Groq client")
+        except Exception as e:
+            logger.error(f"Failed to initialize Groq client: {e}")
+            raise
         
     def _create_classification_prompt(self, ticket: Ticket) -> str: