Update app.py

app.py

@@ -19,6 +19,11 @@ logger = logging.getLogger(__name__)
 
 class EnhancedContentGenerator:
     def __init__(self):
+        # Check for API key
+        self.api_key = os.getenv("GROQ_API_KEY")
+        if not self.api_key:
+            raise ValueError("GROQ_API_KEY not found in environment variables")
+            
         self.output_dir = Path("generated_content")
         self.output_dir.mkdir(exist_ok=True)
         
@@ -35,9 +40,13 @@ class EnhancedContentGenerator:
         self.pipe.enable_attention_slicing()
         
         # Initialize Groq client
-        if not (self.api_key := os.getenv("GROQ_API_KEY")):
-            raise ValueError("GROQ_API_KEY not found in environment variables")
         self.groq_client = groq.Groq(api_key=self.api_key)
+        
+        # Create output directories if they don't exist
+        self.audio_dir = self.output_dir / "audio"
+        self.video_dir = self.output_dir / "video"
+        self.audio_dir.mkdir(exist_ok=True)
+        self.video_dir.mkdir(exist_ok=True)
 
     def generate_cartoon_frame(self, prompt, style="cartoon"):
         """Generate a single cartoon frame with specified style"""
@@ -101,7 +110,7 @@ class EnhancedContentGenerator:
     def enhance_audio(self, audio_path, style="cartoon"):
         """Add effects to the audio based on style"""
         try:
-            audio = AudioFileClip(audio_path)
+            audio = AudioFileClip(str(audio_path))
             
             if style == "cartoon":
                 # Speed up slightly for cartoon effect
@@ -111,12 +120,12 @@ class EnhancedContentGenerator:
                 echo = audio.set_start(0.1)
                 audio = CompositeVideoClip([audio, echo.volumex(0.3)])
             
-            enhanced_path = audio_path.replace('.wav', '_enhanced.wav')
+            enhanced_path = str(audio_path).replace('.wav', '_enhanced.wav')
             audio.write_audiofile(enhanced_path)
             return enhanced_path
         except Exception as e:
             logger.error(f"Error enhancing audio: {str(e)}")
-            return audio_path
+            return str(audio_path)
 
     def generate_comedy_animation(self, prompt):
         """Generate enhanced comedy animation"""
@@ -130,29 +139,30 @@ class EnhancedContentGenerator:
             - Scene descriptions in (parentheses)
             Keep it family-friendly and around 3-4 scenes."""
             
-            script = self.groq_client.chat.completions.create(
+            completion = self.groq_client.chat.completions.create(
                 messages=[
                     {"role": "system", "content": "You are a professional cartoon comedy writer."},
                     {"role": "user", "content": script_prompt}
                 ],
                 model="mixtral-8x7b-32768",
                 temperature=0.7
-            ).choices[0].message.content
+            )
+            script = completion.choices[0].message.content
             
             # Generate frames with cartoon style
             frames = self.generate_video_sequence(script, style="cartoon")
             
             # Generate and enhance audio
-            speech_path = str(self.output_dir / f"speech_{hash(script)}.wav")
-            self.tts.tts_to_file(text=script, file_path=speech_path)
+            speech_path = self.audio_dir / f"speech_{hash(script)}.wav"
+            self.tts.tts_to_file(text=script, file_path=str(speech_path))
             enhanced_speech = self.enhance_audio(speech_path, "cartoon")
             
             # Create video with enhanced frames
-            video_path = str(self.output_dir / f"video_{hash(prompt)}.mp4")
+            video_path = self.video_dir / f"video_{hash(prompt)}.mp4"
             frames_tensor = torch.from_numpy(np.array(frames)).permute(0, 3, 1, 2)
-            write_video(video_path, frames_tensor, fps=12)  # Higher FPS for smoother animation
+            write_video(str(video_path), frames_tensor, fps=12)  # Higher FPS for smoother animation
             
-            return script, video_path, enhanced_speech
+            return script, str(video_path), enhanced_speech
             
         except Exception as e:
             logger.error(f"Error in comedy animation generation: {str(e)}")
@@ -170,29 +180,30 @@ class EnhancedContentGenerator:
             - (Action descriptions) for animation
             Make it upbeat and memorable!"""
             
-            lyrics = self.groq_client.chat.completions.create(
+            completion = self.groq_client.chat.completions.create(
                 messages=[
                     {"role": "system", "content": "You are a children's music composer."},
                     {"role": "user", "content": lyrics_prompt}
                 ],
                 model="mixtral-8x7b-32768",
                 temperature=0.7
-            ).choices[0].message.content
+            )
+            lyrics = completion.choices[0].message.content
             
             # Generate frames with kids' style
             frames = self.generate_video_sequence(lyrics, style="kids", num_frames=36)
             
             # Generate and enhance audio
-            speech_path = str(self.output_dir / f"music_{hash(lyrics)}.wav")
-            self.tts.tts_to_file(text=lyrics, file_path=speech_path)
+            speech_path = self.audio_dir / f"music_{hash(lyrics)}.wav"
+            self.tts.tts_to_file(text=lyrics, file_path=str(speech_path))
             enhanced_speech = self.enhance_audio(speech_path, "kids")
             
             # Create video with enhanced frames
-            video_path = str(self.output_dir / f"video_{hash(theme)}.mp4")
+            video_path = self.video_dir / f"video_{hash(theme)}.mp4"
             frames_tensor = torch.from_numpy(np.array(frames)).permute(0, 3, 1, 2)
-            write_video(video_path, frames_tensor, fps=15)  # Smooth animation for kids
+            write_video(str(video_path), frames_tensor, fps=15)  # Smooth animation for kids
             
-            return lyrics, video_path, enhanced_speech
+            return lyrics, str(video_path), enhanced_speech
             
         except Exception as e:
             logger.error(f"Error in kids music animation generation: {str(e)}")
@@ -243,4 +254,4 @@ def create_interface():
 
 if __name__ == "__main__":
     app = create_interface()
-    app.launch()
+    app.launch()