Update engine.py

engine.py

@@ -1,5 +1,5 @@
 # ==============================================================================
-# engine.py - [FINAL PRO VERSION FOR HUGGING FACE]
+# engine.py - [FINAL PRO VERSION FOR HUGGING FACE & LOCAL]
 # FEATURE UPGRADE: Cinematic Zoom (Ken Burns), Random Transitions, High Quality Render
 # ==============================================================================
 
@@ -17,6 +17,13 @@ import re
 from gtts import gTTS
 from werkzeug.utils import secure_filename
 
+# स्थानीय परीक्षण के लिए dotenv लोड करें (अगर उपलब्ध हो)
+try:
+    from dotenv import load_dotenv
+    load_dotenv()
+except ImportError:
+    pass
+
 # ==============================================================================
 # 1. Global Setup and Database Functions
 # ==============================================================================
@@ -79,12 +86,21 @@ def update_task_final_status(task_id, status, error_message=None, output_filenam
     conn.close()
 
 def load_api_keys(prefix):
+    """
+    prefix (जैसे 'Gemini_Key') से शुरू होने वाली सभी environment variables को लोड करता है।
+    यह 'Gemini_Key_1', 'Gemini_Key_2' आदि को अपने आप पहचान लेगा।
+    """
     try:
         prefix_lower = prefix.lower()
-        keys = [v for k, v in os.environ.items() if k.lower().startswith(prefix_lower)]
-        if not keys:
-            print(f"🚨 चेतावनी: '{prefix}' से शुरू होने वाला कोई एनवायरनमेंट वेरिएबल नहीं मिला।")
-            return keys
+        # Environment variables को चेक करें (case-insensitive search)
+        keys = [v.strip() for k, v in os.environ.items() if k.lower().startswith(prefix_lower) and v.strip()]
+        
+        if keys:
+            print(f"✅ API Key Check: '{prefix}' के लिए {len(keys)} कीज़ मिलीं (Example: ...{keys[0][-4:]})")
+        else:
+            print(f"❌ API Key Check: '{prefix}' के लिए कोई कीज़ नहीं मिलीं!")
+            
+        return keys
     except Exception as e:
         print(f"🚨 एनवायरनमेंट वेरिएबल्स लोड करते समय त्रुटि: {e}")
         return []
@@ -331,8 +347,22 @@ def run_ai_engine_worker(task_id, script_text, script_file_path, orientation, ma
     
     try:
         log("Step 0: API Keys की पुष्टि...", 2)
-        gemini_keys, pexels_keys, pixabay_keys, groq_keys = load_api_keys("Gemini_Key"), load_api_keys("Pexels_Key"), load_api_keys("Pixabay_Key"), load_api_keys("Groq_Key")
-        if not all([gemini_keys, pexels_keys, pixabay_keys, groq_keys]): raise Exception("API Key Error: Missing one or more required keys.")
+        # कीज़ लोड करें
+        gemini_keys = load_api_keys("Gemini_Key")
+        pexels_keys = load_api_keys("Pexels_Key")
+        pixabay_keys = load_api_keys("Pixabay_Key")
+        groq_keys = load_api_keys("Groq_Key")
+
+        # अगर कोई की गायब है तो सटीक Error दें
+        missing = []
+        if not gemini_keys: missing.append("Gemini_Key")
+        if not pexels_keys: missing.append("Pexels_Key")
+        if not pixabay_keys: missing.append("Pixabay_Key")
+        if not groq_keys: missing.append("Groq_Key")
+        
+        if missing:
+            raise Exception(f"API Key Error: ये कीज़ नहीं मिलीं: {', '.join(missing)}. कृपया सुनिश्चित करें कि आपने उन्हें 'Gemini_Key_1' या 'Gemini_Key' जैसे नामों से सेट किया है।")
+
         gemini = GeminiTeam(api_keys=gemini_keys); log("-> सभी जरूरी API कीज मौजूद हैं।", 5)
 
         log("Step 1: स्क्रिप्ट तैयार की जा रही है...", 8)