Update core_creator/intent_parser.py

core_creator/intent_parser.py

@@ -1,6 +1,15 @@
 # intent_parser.py - Extracts purpose and domain of the robotics app idea
 
-from openai import OpenAI
+import os
+import google.generativeai as genai
+
+# 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.")
+
+genai.configure(api_key=GEMINI_API_KEY)
+model = genai.GenerativeModel("gemini-pro")
 
 # Categories the system understands
 INTENT_CATEGORIES = [
@@ -14,9 +23,10 @@ INTENT_CATEGORIES = [
     "creative"
 ]
 
-# Simple prompt to detect the category of the robot idea
+# Classify robot idea using Gemini
+
 def classify_robot_idea(user_input: str) -> str:
-    system_prompt = f"""
+    prompt = f"""
     Classify this user idea into one of the following categories:
     {', '.join(INTENT_CATEGORIES)}.
     Only return the category word. If none fits, return 'creative'.
@@ -25,17 +35,10 @@ def classify_robot_idea(user_input: str) -> str:
     Category:
     """
 
-    response = OpenAI().chat.completions.create(
-        model="gpt-4o",
-        messages=[
-            {"role": "system", "content": "You are a classification AI for robotics ideas."},
-            {"role": "user", "content": system_prompt},
-        ],
-        temperature=0
-    )
-    return response.choices[0].message.content.strip().lower()
+    response = model.generate_content(prompt)
+    return response.text.strip().lower()
 
 # Example
 if __name__ == "__main__":
-    example = "Create a robot that helps blind users find objects at home."
-    print("Predicted Intent:", classify_robot_idea(example))
+    idea = "Build a robot that reminds elderly people to take medicine."
+    print("Predicted Intent:", classify_robot_idea(idea))