Update agents/agents.py

agents/agents.py

@@ -11,12 +11,11 @@ load_dotenv()
 openai_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY")) if os.getenv("OPENAI_API_KEY") else None
 ELEVENLABS_API_KEY = os.getenv("ELEVENLABS_API_KEY")
 
-
 class TopicAgent:
     def generate_outline(self, topic, duration, difficulty):
         if not openai_client:
             return self._mock_outline(topic, duration, difficulty)
-
+            
         try:
             response = openai_client.chat.completions.create(
                 model="gpt-4-turbo",
@@ -45,9 +44,9 @@ class TopicAgent:
                 response_format={"type": "json_object"}
             )
             return json.loads(response.choices[0].message.content)
-        except Exception:
+        except Exception as e:
             return self._mock_outline(topic, duration, difficulty)
-
+    
     def _mock_outline(self, topic, duration, difficulty):
         return {
             "title": f"Mastering {topic} for Business Impact",
@@ -91,12 +90,11 @@ class TopicAgent:
             ]
         }
 
-
 class ContentAgent:
     def generate_content(self, outline):
         if not openai_client:
             return self._mock_content(outline)
-
+            
         try:
             response = openai_client.chat.completions.create(
                 model="gpt-4-turbo",
@@ -128,9 +126,9 @@ class ContentAgent:
                 response_format={"type": "json_object"}
             )
             return json.loads(response.choices[0].message.content)
-        except Exception:
+        except Exception as e:
             return self._mock_content(outline)
-
+    
     def _mock_content(self, outline):
         return {
             "workshop_title": outline.get("title", "Premium AI Workshop"),
@@ -172,12 +170,11 @@ class ContentAgent:
             ]
         }
 
-
 class SlideAgent:
     def generate_slides(self, content):
         if not openai_client:
             return self._professional_slides(content)
-
+            
         try:
             response = openai_client.chat.completions.create(
                 model="gpt-4-turbo",
@@ -204,9 +201,9 @@ class SlideAgent:
                 max_tokens=2500
             )
             return response.choices[0].message.content
-        except Exception:
+        except Exception as e:
             return self._professional_slides(content)
-
+    
     def _professional_slides(self, content):
         return f"""---
 marp: true
@@ -256,14 +253,13 @@ Calculate net ROI
 
 Q&A
 Let's discuss your specific challenges
-```"""
-
+"""
 
 class CodeAgent:
     def generate_code(self, content):
         if not openai_client:
             return self._professional_code(content)
-
+            
         try:
             response = openai_client.chat.completions.create(
                 model="gpt-4-turbo",
@@ -290,7 +286,7 @@ class CodeAgent:
                 max_tokens=2500
             )
             return response.choices[0].message.content
-        except Exception:
+        except Exception as e:
             return self._professional_code(content)
 
     def _professional_code(self, content):
@@ -332,12 +328,11 @@ def integrate_with_salesforce(prompt, salesforce_data):
     return call_ai_api(enriched_prompt)
 """
 
-
 class DesignAgent:
     def generate_design(self, slide_content):
         if not openai_client:
             return None
-
+            
         try:
             response = openai_client.images.generate(
                 model="dall-e-3",
@@ -350,23 +345,22 @@ class DesignAgent:
                 size="1024x1024"
             )
             return response.data[0].url
-        except Exception:
+        except Exception as e:
             return None
 
-
 class VoiceoverAgent:
     def __init__(self):
         self.api_key = ELEVENLABS_API_KEY
         self.voice_id = "21m00Tcm4TlvDq8ikWAM"  # Default voice ID
         self.model = "eleven_monolingual_v1"
-
+        
     def generate_voiceover(self, text, voice_id=None):
         if not self.api_key:
             return None
-
+            
         try:
             voice = voice_id if voice_id else self.voice_id
-
+            
             url = f"https://api.elevenlabs.io/v1/text-to-speech/{voice}"
             headers = {
                 "Accept": "audio/mpeg",
@@ -384,24 +378,24 @@ class VoiceoverAgent:
                 }
             }
             response = requests.post(url, json=data, headers=headers)
-
+            
             if response.status_code == 200:
                 return response.content
             return None
-        except Exception:
+        except Exception as e:
             return None
-
+    
     def get_voices(self):
         if not self.api_key:
             return []
-
+            
         try:
             url = "https://api.elevenlabs.io/v1/voices"
             headers = {"xi-api-key": self.api_key}
             response = requests.get(url, headers=headers)
-
+            
             if response.status_code == 200:
                 return response.json().get("voices", [])
             return []
-        except Exception:
-            return []
+        except Exception as e:
+            return []