Update app.py

app.py

@@ -18,7 +18,10 @@ warnings.filterwarnings('ignore')
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
 if not GROQ_API_KEY:
-    raise ValueError("GROQ_API_KEY environment variable not set. Please add it to your Hugging Face Space secrets.")
+    print("⚠️ WARNING: GROQ_API_KEY environment variable not set!")
+    print("Please add your Groq API key to your Hugging Face Space secrets.")
+    print("For demo purposes, the app will continue but API calls will fail.")
+    GROQ_API_KEY = "dummy-key-for-demo"
 
 # ------------------------------
 # SQL Converter Using Groq API
@@ -26,9 +29,19 @@ if not GROQ_API_KEY:
 
 class EnhancedNL2SQLConverter:
     def __init__(self, model_name: str = "llama3-70b-8192"):
-        self.client = Groq(api_key=GROQ_API_KEY)
         self.model_name = model_name
-        print(f"Using Groq API with model: {self.model_name}")
+        self.client = None
+        
+        try:
+            # Initialize Groq client with proper error handling
+            if GROQ_API_KEY and GROQ_API_KEY != "dummy-key-for-demo":
+                self.client = Groq(api_key=GROQ_API_KEY)
+                print(f"✅ Successfully initialized Groq client with model: {self.model_name}")
+            else:
+                print("⚠️ Groq client not initialized - API key missing")
+        except Exception as e:
+            print(f"❌ Error initializing Groq client: {str(e)}")
+            self.client = None
 
         self.default_schema = """
         Table: employees
@@ -43,6 +56,10 @@ class EnhancedNL2SQLConverter:
 
     def generate_sql(self, query: str, schema: str = None) -> str:
         try:
+            # Check if client is properly initialized
+            if not self.client:
+                return "ERROR: Groq API client not initialized. Please check your API key."
+            
             schema_to_use = schema or self.default_schema
 
             system_prompt = """You are an expert SQL query generator. Convert natural language questions to SQL queries based on the provided database schema.
@@ -157,10 +174,17 @@ class SQLEvaluator:
             return False, str(e)
 
 # ------------------------------
-# Initialize components
+# Initialize components with error handling
 # ------------------------------
-converter = EnhancedNL2SQLConverter()
-evaluator = SQLEvaluator()
+try:
+    converter = EnhancedNL2SQLConverter()
+    evaluator = SQLEvaluator()
+    print("✅ Application components initialized successfully")
+except Exception as e:
+    print(f"❌ Error initializing components: {str(e)}")
+    # Create dummy components for graceful degradation
+    converter = None
+    evaluator = SQLEvaluator()  # Database component should still work
 
 # ------------------------------
 # Gradio Interface Functions
@@ -172,11 +196,15 @@ def process_nl_query(nl_query: str) -> Tuple[str, str, str]:
         return "", "", "Please enter a natural language query."
     
     try:
+        # Check if converter is available
+        if not converter:
+            return "", "", "❌ Error: SQL converter not initialized. Please check API configuration."
+        
         # Generate SQL
         generated_sql = converter.generate_sql(nl_query)
         
         if generated_sql.startswith("ERROR"):
-            return generated_sql, "", "Failed to generate SQL query."
+            return generated_sql, "", "❌ Failed to generate SQL query. Please check your API key."
         
         # Execute SQL
         success, result = evaluator.execute_sql(generated_sql)