Update core_creator/intent_parser.py

core_creator/intent_parser.py

@@ -1,17 +1,11 @@
-# intent_parser.py - Extracts purpose and domain of the robotics app idea
+# intent_parser.py - Smart fallback to Gemini or OpenAI for intent classification
 
 import os
-import google.generativeai as genai
+import json
 
-# Use Gemini API Key from environment variable
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
-if not GEMINI_API_KEY:
-    raise EnvironmentError("GEMINI_API_KEY not set in environment variables.")
+USE_GEMINI = bool(os.getenv("GEMINI_API_KEY"))
+USE_OPENAI = bool(os.getenv("OPENAI_API_KEY"))
 
-genai.configure(api_key=GEMINI_API_KEY)
-model = genai.GenerativeModel("gemini-pro")
-
-# Categories the system understands
 INTENT_CATEGORIES = [
     "educational",
     "assistive",
@@ -23,7 +17,16 @@ INTENT_CATEGORIES = [
     "creative"
 ]
 
-# Classify robot idea using Gemini
+if USE_GEMINI:
+    import google.generativeai as genai
+    genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+    gemini_model = genai.GenerativeModel(model_name="models/gemini-1.5-pro-latest")
+
+elif USE_OPENAI:
+    from openai import OpenAI
+    openai_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+else:
+    raise EnvironmentError("No valid API key set. Please set GEMINI_API_KEY or OPENAI_API_KEY.")
 
 def classify_robot_idea(user_input: str) -> str:
     prompt = f"""
@@ -35,10 +38,22 @@ def classify_robot_idea(user_input: str) -> str:
     Category:
     """
 
-    response = model.generate_content(prompt)
-    return response.text.strip().lower()
+    if USE_GEMINI:
+        response = gemini_model.generate_content(prompt)
+        return response.text.strip().lower()
+
+    elif USE_OPENAI:
+        response = openai_client.chat.completions.create(
+            model="gpt-4o",
+            messages=[
+                {"role": "system", "content": "You are a classification AI for robotics ideas."},
+                {"role": "user", "content": prompt}
+            ],
+            temperature=0
+        )
+        return response.choices[0].message.content.strip().lower()
 
 # Example
 if __name__ == "__main__":
-    idea = "Build a robot that reminds elderly people to take medicine."
+    idea = "Build a robot that helps kids focus while doing homework."
     print("Predicted Intent:", classify_robot_idea(idea))