Create agents/topic_agent.py

agents/topic_agent.py

@@ -1,38 +1,34 @@
-# File: agents/topic_agent.py
+# agents/topic_agent.py
+import openai
 import json
 
 class TopicAgent:
+    def __init__(self, api_key=None):
+        if api_key:
+            openai.api_key = api_key
+    
     def generate_outline(self, topic, duration, difficulty):
-        # In production: Replace with actual LLM API call
+        if not openai.api_key:
+            return self._mock_outline(topic, duration, difficulty)
+            
+        response = openai.ChatCompletion.create(
+            model="gpt-4",
+            messages=[
+                {"role": "system", "content": "You're an expert corporate trainer creating AI workshop outlines"},
+                {"role": "user", "content": (
+                    f"Create a {duration}-hour {difficulty} workshop outline on {topic}. "
+                    "Include: 1) Key learning goals, 2) 4 modules with titles and durations, "
+                    "3) Hands-on exercises per module. Output as JSON."
+                )}
+            ]
+        )
+        return json.loads(response.choices[0].message.content)
+    
+    def _mock_outline(self, topic, duration, difficulty):
         return {
             "topic": topic,
             "duration": f"{duration} hours",
             "difficulty": difficulty,
-            "goals": [
-                f"Master advanced {topic} techniques",
-                "Develop industry-specific applications",
-                "Build and evaluate complex AI workflows",
-                "Implement best practices for production"
-            ],
-            "modules": [
-                {
-                    "title": f"Fundamentals of {topic}",
-                    "duration": "30 min",
-                    "learning_points": [
-                        "Core principles and terminology",
-                        "Patterns and anti-patterns",
-                        "Evaluation frameworks"
-                    ]
-                },
-                {
-                    "title": f"{topic} for Enterprise Applications",
-                    "duration": "45 min",
-                    "learning_points": [
-                        "Industry-specific use cases",
-                        "Integration with existing systems",
-                        "Scalability considerations"
-                    ]
-                }
-                # Additional modules would be generated here
-            ]
+            "goals": ["Goal 1", "Goal 2"],
+            "modules": [{"title": "Module 1", "duration": "30 min"}]
         }