Update utils/ai_engine.py

utils/ai_engine.py

@@ -3,19 +3,17 @@ from groq import Groq
 from openai import OpenAI
 
 # -----------------------------------------------------------
-# 🔑 Load API Keys (HuggingFace Secrets)
+# 🔑 Load API Keys
 # -----------------------------------------------------------
 
 GROQ_KEY = os.getenv("GROQ_API_KEY")
 OPENAI_KEY = os.getenv("OPENAI_API_KEY")
 
 # -----------------------------------------------------------
-# 🚀 Initialize Clients (safely)
+# 🚀 Initialize Clients
 # -----------------------------------------------------------
 
 groq_client = Groq(api_key=GROQ_KEY) if GROQ_KEY else None
-
-# ⚠️ Disable OpenAI automatically if key exists but quota is empty
 openai_client = OpenAI(api_key=OPENAI_KEY) if OPENAI_KEY else None
 
 
@@ -26,27 +24,27 @@ openai_client = OpenAI(api_key=OPENAI_KEY) if OPENAI_KEY else None
 def generate_text(prompt: str) -> str:
     """
     AI text generation:
-    1️⃣ Groq (primary, free, fast)
-    2️⃣ OpenAI (fallback ONLY if available)
+    1️⃣ Groq (primary)
+    2️⃣ OpenAI (fallback)
     """
 
     # ----------------------------
-    # 1️⃣ GROQ (PRIMARY)
+    # 1️⃣ GROQ
     # ----------------------------
     if groq_client:
         try:
             resp = groq_client.chat.completions.create(
-    model="llama-3.1-8b-instant",
-    messages=[{"role": "user", "content": prompt}],
-    max_tokens=700,
-    temperature=0.3
-)
-return resp.choices[0].message.content
+                model="llama-3.1-8b-instant",
+                messages=[{"role": "user", "content": prompt}],
+                max_tokens=700,
+                temperature=0.3
+            )
+            return resp.choices[0].message.content
         except Exception as e:
             return f"🚫 Groq failed: {e}"
 
     # ----------------------------
-    # 2️⃣ OPENAI (OPTIONAL FALLBACK)
+    # 2️⃣ OPENAI (fallback)
     # ----------------------------
     if openai_client:
         try:
@@ -57,11 +55,10 @@ return resp.choices[0].message.content
                 temperature=0.3
             )
             return resp.choices[0].message.content
-
         except Exception as e:
             return f"🚫 OpenAI failed: {e}"
 
     # ----------------------------
     # 3️⃣ NO KEYS
     # ----------------------------
-    return "❌ No AI keys found. Please add GROQ_API_KEY in HuggingFace Secrets."
+    return "❌ No AI keys found. Please add GROQ_API_KEY to HuggingFace Secrets."