Update app.py

app.py

@@ -14,8 +14,9 @@ from dotenv import load_dotenv
 # Load environment variables
 load_dotenv()
 
-# Initialize OpenAI API key
+# Initialize API keys
 openai.api_key = os.getenv("OPENAI_API_KEY")
+ELEVENLABS_API_KEY = os.getenv("ELEVENLABS_API_KEY")
 
 # =============================
 # ENHANCED AGENT IMPLEMENTATION
@@ -292,12 +293,69 @@ class DesignAgent:
             st.error(f"Design generation error: {str(e)}")
             return None
 
+class VoiceoverAgent:
+    def __init__(self):
+        self.api_key = ELEVENLABS_API_KEY
+        self.voice_id = "21m00Tcm4TlvDq8ikWAM"  # Default voice ID (Bella)
+        self.model = "eleven_monolingual_v1"
+        
+    def generate_voiceover(self, text, voice_id=None):
+        if not self.api_key:
+            return None
+            
+        try:
+            # Use custom voice if provided, otherwise use default
+            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",
+                "Content-Type": "application/json",
+                "xi-api-key": self.api_key
+            }
+            data = {
+                "text": text,
+                "model_id": self.model,
+                "voice_settings": {
+                    "stability": 0.7,
+                    "similarity_boost": 0.8,
+                    "style": 0.5,
+                    "use_speaker_boost": True
+                }
+            }
+            response = requests.post(url, json=data, headers=headers)
+            
+            if response.status_code == 200:
+                return response.content
+            else:
+                st.error(f"Voiceover API error: {response.status_code} - {response.text}")
+                return None
+        except Exception as e:
+            st.error(f"Voiceover generation error: {str(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:
+            return []
+
 # Initialize agents
 topic_agent = TopicAgent()
 content_agent = ContentAgent()
 slide_agent = SlideAgent()
 code_agent = CodeAgent()
 design_agent = DesignAgent()
+voiceover_agent = VoiceoverAgent()
 
 # =====================
 # STREAMLIT APPLICATION
@@ -337,6 +395,35 @@ st.markdown("""
         border-radius: 10px;
         padding: 15px;
     }
+    .audio-player {
+        margin: 15px 0;
+        border-radius: 10px;
+        background: rgba(255,255,255,0.1);
+        padding: 15px;
+    }
+    .voice-option {
+        display: flex;
+        align-items: center;
+        margin: 5px 0;
+        padding: 8px;
+        border-radius: 8px;
+        cursor: pointer;
+        transition: background 0.3s;
+    }
+    .voice-option:hover {
+        background: rgba(255,255,255,0.2);
+    }
+    .voice-option.selected {
+        background: rgba(0,180,155,0.3);
+        border: 2px solid #00b09b;
+    }
+    .voice-thumb {
+        width: 40px;
+        height: 40px;
+        border-radius: 50%;
+        margin-right: 10px;
+        object-fit: cover;
+    }
 </style>
 """, unsafe_allow_html=True)
 
@@ -346,33 +433,108 @@ with col1:
     st.image("https://cdn-icons-png.flaticon.com/512/1995/1995485.png", width=100)
 with col2:
     st.title("🤖 Workshop in a Box Pro")
-    st.caption("Generate Premium Corporate AI Training Workshops in Minutes")
+    st.caption("Generate Premium Corporate AI Training Workshops with Voiceovers")
+
+# Initialize session state
+if 'workshop_topic' not in st.session_state:
+    st.session_state.workshop_topic = "Advanced Prompt Engineering"
+if 'generated' not in st.session_state:
+    st.session_state.generated = False
+if 'generating' not in st.session_state:
+    st.session_state.generating = False
+if 'voiceovers' not in st.session_state:
+    st.session_state.voiceovers = {}
+if 'selected_voice' not in st.session_state:
+    st.session_state.selected_voice = "21m00Tcm4TlvDq8ikWAM"  # Default voice ID
 
 # Sidebar configuration
 with st.sidebar:
     st.header("⚙️ Workshop Configuration")
-    workshop_topic = st.text_input("Workshop Topic", "Advanced Prompt Engineering")
+    
+    # Workshop topic input with session state
+    st.session_state.workshop_topic = st.text_input(
+        "Workshop Topic", 
+        st.session_state.workshop_topic,
+        key="topic_input",
+        help="Enter the main topic for your workshop"
+    )
+    
+    # Validate topic input
+    if st.session_state.workshop_topic.strip() == "":
+        st.warning("Please enter a workshop topic")
+        st.stop()
+    
     duration = st.slider("Duration (hours)", 1.0, 8.0, 3.0, 0.5)
     difficulty = st.selectbox("Difficulty Level", 
                             ["Beginner", "Intermediate", "Advanced", "Expert"])
     include_code = st.checkbox("Include Code Labs", True)
     include_design = st.checkbox("Generate Visual Designs", True)
+    include_voiceover = st.checkbox("Generate Voiceovers", True)
+    
+    # Voice selection
+    if include_voiceover:
+        st.subheader("🎙️ Voice Selection")
+        
+        # Get available voices
+        voices = voiceover_agent.get_voices()
+        
+        # If we have voices, let the user select one
+        if voices:
+            # Create 2 columns for voice selection
+            cols = st.columns(2)
+            for i, voice in enumerate(voices[:4]):  # Show first 4 voices
+                with cols[i % 2]:
+                    # Create a unique key for each voice button
+                    voice_key = f"voice_{voice['voice_id']}"
+                    
+                    # Display voice option
+                    if st.button(
+                        f"🗣️ {voice['name']}",
+                        key=voice_key,
+                        use_container_width=True,
+                        help=f"Select {voice['name']} voice"
+                    ):
+                        st.session_state.selected_voice = voice['voice_id']
+            
+            # Show which voice is currently selected
+            selected_voice_name = next((v['name'] for v in voices if v['voice_id'] == st.session_state.selected_voice), "Default")
+            st.info(f"Selected Voice: **{selected_voice_name}**")
+        else:
+            st.warning("Couldn't load voices. Using default voice.")
     
     if st.button("✨ Generate Workshop", type="primary", use_container_width=True):
         st.session_state.generating = True
+        st.session_state.voiceovers = {}  # Reset previous voiceovers
 
 # Generation pipeline
-if hasattr(st.session_state, 'generating'):
-    with st.spinner("🚀 Creating your premium workshop materials..."):
+if st.session_state.generating:
+    with st.spinner(f"🚀 Creating your {st.session_state.workshop_topic} workshop..."):
         start_time = time.time()
         
         # Agent pipeline
-        outline = topic_agent.generate_outline(workshop_topic, duration, difficulty)
+        outline = topic_agent.generate_outline(st.session_state.workshop_topic, duration, difficulty)
         content = content_agent.generate_content(outline)
         slides = slide_agent.generate_slides(content)
         code_labs = code_agent.generate_code(content) if include_code else None
         design_url = design_agent.generate_design(slides) if include_design else None
         
+        # Generate voiceovers if enabled
+        voiceovers = {}
+        if include_voiceover and ELEVENLABS_API_KEY:
+            for i, module in enumerate(content.get("modules", [])):
+                # Create a short intro for each module (limit to 500 characters)
+                intro_text = f"Welcome to Module {i+1}: {module['title']}. " + \
+                             f"In this module, we'll cover: {', '.join(module.get('speaker_notes', []))[:300]}"
+                
+                # Generate voiceover
+                audio_data = voiceover_agent.generate_voiceover(
+                    intro_text,
+                    st.session_state.selected_voice
+                )
+                
+                if audio_data:
+                    voiceovers[f"module_{i+1}_intro.mp3"] = audio_data
+        
         # Prepare download package
         zip_buffer = io.BytesIO()
         with zipfile.ZipFile(zip_buffer, "a") as zip_file:
@@ -387,6 +549,9 @@ if hasattr(st.session_state, 'generating'):
                     zip_file.writestr("slide_design.png", img_data)
                 except:
                     pass
+            # Add voiceovers to ZIP
+            for filename, audio_data in voiceovers.items():
+                zip_file.writestr(f"voiceovers/{filename}", audio_data)
         
         # Store results
         st.session_state.outline = outline
@@ -394,20 +559,21 @@ if hasattr(st.session_state, 'generating'):
         st.session_state.slides = slides
         st.session_state.code_labs = code_labs
         st.session_state.design_url = design_url
+        st.session_state.voiceovers = voiceovers
         st.session_state.zip_buffer = zip_buffer
         st.session_state.gen_time = round(time.time() - start_time, 2)
         st.session_state.generated = True
         st.session_state.generating = False
 
 # Results display
-if hasattr(st.session_state, 'generated'):
-    st.success(f"✅ Premium workshop materials generated in {st.session_state.gen_time} seconds!")
+if st.session_state.generated:
+    st.success(f"✅ {st.session_state.workshop_topic} workshop generated in {st.session_state.gen_time} seconds!")
     
     # Download button
     st.download_button(
         label="📥 Download Workshop Package",
         data=st.session_state.zip_buffer.getvalue(),
-        file_name=f"{workshop_topic.replace(' ', '_')}_workshop.zip",
+        file_name=f"{st.session_state.workshop_topic.replace(' ', '_')}_workshop.zip",
         mime="application/zip",
         use_container_width=True
     )
@@ -429,15 +595,35 @@ if hasattr(st.session_state, 'generated'):
     if st.session_state.design_url:
         with st.expander("🎨 Generated Design"):
             st.image(st.session_state.design_url, caption="Custom Slide Design")
+    
+    # Voiceover player
+    if st.session_state.voiceovers:
+        with st.expander("🔊 Voiceover Previews"):
+            for i, (filename, audio_bytes) in enumerate(st.session_state.voiceovers.items()):
+                module_num = filename.split("_")[1]
+                st.subheader(f"Module {module_num} Introduction")
+                
+                # Create an audio player for each voiceover
+                st.audio(audio_bytes, format="audio/mp3")
+                
+                # Add download button for individual voiceover
+                st.download_button(
+                    label=f"Download Module {module_num} Voiceover",
+                    data=audio_bytes,
+                    file_name=filename,
+                    mime="audio/mpeg",
+                    key=f"voiceover_dl_{i}"
+                )
 
 # Sales and booking section
 st.divider()
 st.subheader("🚀 Ready to Deliver This Workshop?")
-st.markdown("""
-### Premium Corporate Training Package
+st.markdown(f"""
+### Premium {st.session_state.workshop_topic} Training Package
 - **Live Workshop Delivery**: $10,000 per session
 - **On-Demand Course**: $5,000 (unlimited access)
 - **Pilot Program**: $1,000 refundable deposit
+- **Voiceover Add-on**: $500 per module
 
 ✨ **All inclusive**: Customization, materials, and follow-up support
 """)
@@ -458,12 +644,20 @@ with st.sidebar:
     else:
         st.warning("OpenAI API not set - using enhanced mock data")
     
-    st.info("""
+    if ELEVENLABS_API_KEY:
+        st.success("ElevenLabs API Connected")
+    elif st.session_state.get('voiceovers'):
+        st.warning("ElevenLabs API not set - using mock voiceovers")
+    
+    st.info(f"""
+    **Current Workshop:**
+    {st.session_state.workshop_topic}
+    
     **Premium Features:**
-    - AI-generated slide designs
+    - AI-generated voiceovers
+    - Professional slide designs
     - Real-world case studies
     - Practical code labs
-    - Professional templates
     """)
 
 # How it works section
@@ -471,10 +665,10 @@ st.divider()
 st.subheader("💡 How It Works")
 st.markdown("""
 1. **Configure** your workshop topic and parameters
-2. **Generate** premium training materials in seconds
+2. **Generate** premium training materials with voiceovers
 3. **Customize** the content to your specific needs
 4. **Deliver** high-value corporate training at $10K/session
 5. **Reuse** the materials for unlimited revenue
 
-*"Created 3 workshops in 15 minutes and booked $30K in contracts"* - Sarah T., AI Training Consultant
+*"The voiceover feature helped me create on-demand courses that sold for $5K each"* - Michael L., AI Consultant
 """)