Update app.py

app.py

@@ -1,19 +1,27 @@
+import os
 import streamlit as st
 import pandas as pd
-import os
 from groq import Groq
+from dotenv import load_dotenv
+
+# Load environment variables from .env file (optional)
+load_dotenv()
 
-# Set the GROQ API key from environment variable
-groq_api_key = os.getenv("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")  # Make sure the GROQ API key is set in your environment
+# Get the API key from environment variable or set directly
+groq_api_key = os.getenv("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")  # First try from environment variable
+if not groq_api_key:
+    # Optionally, set the API key directly for testing
+    groq_api_key = "your_actual_groq_api_key_here"
+
+if not groq_api_key:
+    st.error("GROQ_API_KEY is not set. Please provide a valid API key.")
+    st.stop()
 
 # Initialize the GROQ client
-try:
-    groq_client = Groq(api_key=groq_api_key)
-    
-    if not groq_client:
-        raise ValueError("Failed to initialize the GROQ client. Please check your API key.")
-except Exception as e:
-    st.error(f"Error initializing GROQ client: {e}")
+groq_client = Groq(api_key=groq_api_key)
+
+if not groq_client:
+    st.error("Failed to initialize the GROQ client. Please check your API key.")
     st.stop()
 
 # Function to load and preprocess data