Update app.py

app.py

@@ -62,7 +62,217 @@ class SpiritualThemes:
         "meditation guidance"
     ]
 
-[Previous ContentGenerator class code goes here]
+
+class ContentGenerator:
+    def __init__(self, openai_api_key, replicate_api_key):
+        self.llm = OpenAI(api_key=openai_api_key, temperature=0.7)
+        self.openai_client = OpenAIClient(api_key=openai_api_key)
+        self.replicate_client = replicate.Client(api_token=replicate_api_key)
+
+    def _generate_prompt_variation(self):
+        """Generate varied prompts for quote generation"""
+        tradition_category = random.choice(list(SpiritualThemes.TRADITIONS.keys()))
+        traditions = SpiritualThemes.TRADITIONS[tradition_category]
+        themes = random.sample(SpiritualThemes.THEMES, 2)
+        style = random.choice(SpiritualThemes.QUOTE_STYLES)
+        
+        return f"""Focus on {tradition_category} wisdom, particularly from {', '.join(traditions)}.
+        Explore themes of {themes[0]} and {themes[1]} in a {style} style.
+        Ensure the quote is profound yet accessible for social media."""
+
+    def _get_fallback_quote(self):
+        """Provide a fallback quote when generation fails"""
+        fallback_quotes = [
+            {
+                "text": "Be the change you wish to see in the world.",
+                "author": "Mahatma Gandhi",
+                "tradition": "Modern Spirituality",
+                "theme": "Transformation"
+            },
+            {
+                "text": "Peace comes from within. Do not seek it without.",
+                "author": "Buddha",
+                "tradition": "Buddhism",
+                "theme": "Inner Peace"
+            },
+            {
+                "text": "The journey of a thousand miles begins with one step.",
+                "author": "Lao Tzu",
+                "tradition": "Taoism",
+                "theme": "Growth"
+            },
+            {
+                "text": "Love is the bridge between you and everything.",
+                "author": "Rumi",
+                "tradition": "Sufi Wisdom",
+                "theme": "Love"
+            }
+        ]
+        quote = random.choice(fallback_quotes)
+        quote["generated_at"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        quote["source"] = "fallback"
+        log_message("ℹ️ Using fallback quote system", "info")
+        return quote
+
+    def generate_quote(self):
+        """Generate a new spiritual quote with enhanced variety"""
+        try:
+            log_message("🎯 Starting quote generation process")
+            
+            prompt_variation = self._generate_prompt_variation()
+            
+            quote_curator = Agent(
+                role='Quote Curator',
+                goal='Generate diverse and profound spiritual quotes',
+                backstory="""You are a wise curator of spiritual wisdom with extensive knowledge 
+                of both ancient traditions and modern spiritual thought. You understand the nuances 
+                of different spiritual and philosophical traditions, and can communicate their wisdom 
+                in an engaging, social media-friendly way.""",
+                llm=self.llm,
+                verbose=True
+            )
+
+            task = Task(
+                description=f"""Generate a spiritual quote following these criteria:
+                {prompt_variation}
+                
+                Requirements:
+                1. Ensure quote authenticity and proper attribution
+                2. Balance depth with accessibility
+                3. Keep length appropriate for Instagram
+                4. Include cultural context if relevant
+                
+                Return in JSON format with keys:
+                - quote: the spiritual quote text
+                - author: name of the source
+                - tradition: specific tradition or philosophy
+                - theme: primary theme of the quote""",
+                expected_output="""A JSON object containing the quote details:
+                {
+                    "quote": "The quote text",
+                    "author": "Author name",
+                    "tradition": "Spiritual tradition",
+                    "theme": "Primary theme"
+                }""",
+                agent=quote_curator
+            )
+
+            crew = Crew(
+                agents=[quote_curator],
+                tasks=[task],
+                verbose=True
+            )
+
+            result = crew.kickoff()
+            log_message("✨ Quote generation completed successfully")
+            return self._parse_quote_result(result)
+            
+        except Exception as e:
+            log_message(f"❌ Quote generation failed: {str(e)}", "error")
+            return self._get_fallback_quote()
+
+    def _parse_quote_result(self, result):
+        """Parse the generated quote result"""
+        try:
+            if isinstance(result, str):
+                result = json.loads(result)
+            return {
+                "text": result.get('quote', ''),
+                "author": result.get('author', ''),
+                "tradition": result.get('tradition', ''),
+                "theme": result.get('theme', ''),
+                "generated_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            }
+        except Exception as e:
+            log_message(f"❌ Error parsing quote result: {str(e)}", "error")
+            raise
+
+    def generate_image(self, quote, tradition, theme):
+        """Generate an image using OpenAI DALL-E with enhanced prompts"""
+        try:
+            log_message("🎨 Starting image generation")
+            
+            style_prompts = {
+                "Buddhism": "zen minimalism, misty mountains, lotus flowers",
+                "Hinduism": "mandala patterns, sacred geometry, vibrant cosmos",
+                "Taoism": "flowing water, yin-yang harmony, natural elements",
+                "Modern Spirituality": "abstract sacred geometry, cosmic patterns",
+                "Christian Mysticism": "ethereal light, cathedral aesthetics",
+                "Sufi Wisdom": "whirling patterns, desert wisdom, geometric art",
+                "Indigenous": "natural elements, earth tones, tribal patterns",
+                "default": "serene minimalism, sacred geometry, natural elements"
+            }
+            
+            style = style_prompts.get(tradition, style_prompts["default"])
+            
+            prompt = f"""Create a spiritual and contemplative image inspired by this theme: '{theme}'
+            Style inspiration: {style}
+            Mood: serene, inspiring, contemplative
+            Requirements: 
+            - Suitable for Instagram
+            - Leave space for text overlay
+            - Balanced composition
+            - No text or symbols
+            - Peaceful and harmonious colors
+            Make it visually striking yet subtle enough for quote overlay."""
+
+            response = self.openai_client.images.generate(
+                model="dall-e-3",
+                prompt=prompt,
+                size="1024x1024",
+                quality="standard",
+                n=1
+            )
+            log_message("✨ Image generated successfully")
+            return response.data[0].url
+        except Exception as e:
+            log_message(f"❌ Image generation failed: {str(e)}", "error")
+            raise
+
+    def generate_audio(self, tradition, theme):
+        """Generate background music using MusicGen with enhanced variety"""
+        try:
+            log_message("🎵 Starting audio generation")
+            
+            music_prompts = {
+                "Buddhism": {
+                    "Inner Peace": "Tibetan singing bowls with gentle bells, very peaceful",
+                    "Mindfulness": "Zen flute with soft water sounds, mindful atmosphere",
+                    "default": "Meditation bells with nature sounds"
+                },
+                "Hinduism": {
+                    "Devotion": "Indian flute with soft tabla, devotional mood",
+                    "Wisdom": "Gentle sitar with peaceful drone",
+                    "default": "Sanskrit chants with soft instrumentation"
+                },
+                "Modern Spirituality": {
+                    "Awareness": "Ambient synthesizer with crystal bowls",
+                    "Growth": "Uplifting piano with gentle pads",
+                    "default": "Modern meditation music with nature sounds"
+                },
+                "default": {
+                    "default": "Peaceful ambient music with gentle bells"
+                }
+            }
+            
+            tradition_prompts = music_prompts.get(tradition, music_prompts["default"])
+            prompt = tradition_prompts.get(theme, tradition_prompts["default"])
+            
+            output = self.replicate_client.run(
+                "meta/musicgen:7be0f12c54a8d0d0d720305c1e6ea10c48d5f0a3afce48478341a0fe682a8787",
+                input={
+                    "prompt": prompt,
+                    "duration": 30,
+                    "model_version": "melody",
+                    "output_format": "mp3"
+                }
+            )
+            
+            log_message("✨ Audio generated successfully")
+            return output
+        except Exception as e:
+            log_message(f"❌ Audio generation failed: {str(e)}", "error")
+            raise
 
 def display_debug_logs():
     """Display debug logs with enhanced visualization"""