Update app.py

app.py

@@ -3,14 +3,12 @@ import os
 from datetime import datetime
 import json
 from crewai import Agent, Task, Crew
-from langchain_community.tools import DuckDuckGoSearchRun
-from langchain_community.utilities import DuckDuckGoSearchAPIWrapper
 from langchain_openai import OpenAI
 from openai import OpenAI as OpenAIClient
 import logging
 import requests
 import replicate
-import time
+import random
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -27,58 +25,99 @@ def log_message(message, type="info"):
         "type": type
     })
 
-class SearchTool:
-    def __init__(self):
-        self.name = "web_search"
-        self.description = "Search the internet for information and verification"
-        self.search = DuckDuckGoSearchRun()
-
-    def func(self, query: str) -> str:
-        """Execute the search"""
-        try:
-            log_message(f"🔍 Searching for: {query}")
-            result = self.search.run(query)
-            log_message(f"✓ Search completed")
-            return result
-        except Exception as e:
-            log_message(f"❌ Search error: {str(e)}", "error")
-            return f"Error performing search: {str(e)}"
+class SpiritualThemes:
+    """Diverse spiritual themes for content generation"""
+    
+    TRADITIONS = {
+        "Eastern": [
+            "Buddhism", "Hinduism", "Taoism", "Zen", "Vedanta",
+            "Yoga Philosophy", "Jainism", "Sikhism", "Confucianism"
+        ],
+        "Western": [
+            "Christian Mysticism", "Sufi Wisdom", "Kabbalah",
+            "Greek Philosophy", "Modern Spirituality", "Stoicism"
+        ],
+        "Indigenous": [
+            "Native American Wisdom", "Aboriginal Spirituality",
+            "African Traditional", "Maori Wisdom"
+        ],
+        "Modern": [
+            "Mindfulness", "Contemporary Philosophy",
+            "Integrative Spirituality", "Secular Wisdom"
+        ]
+    }
+    
+    THEMES = [
+        "Inner Peace", "Mindfulness", "Compassion", "Wisdom",
+        "Self-Discovery", "Unity", "Balance", "Transformation",
+        "Enlightenment", "Purpose", "Gratitude", "Presence",
+        "Love", "Harmony", "Truth", "Liberation", "Service",
+        "Connection", "Growth", "Awareness"
+    ]
+    
+    QUOTE_STYLES = [
+        "contemplative", "inspirational", "philosophical",
+        "practical wisdom", "mystical insight", "ancient teaching",
+        "modern interpretation", "timeless truth", "daily practice",
+        "meditation guidance"
+    ]
 
 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)
-        self.search_tool = SearchTool()
+
+    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 generate_quote(self):
-        """Generate a new spiritual quote using CrewAI"""
+        """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 and validate spiritual quotes',
-                backstory="You are an expert in spiritual traditions and wisdom, focusing on authentic and meaningful quotes.",
-                tools=[self.search_tool],
+                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,
-                max_iterations=3
+                verbose=True
             )
 
             task = Task(
-                description="""Generate a profound spiritual quote following these criteria:
-                1. Should be from any major spiritual tradition (Buddhism, Hinduism, Taoism, stoicism, sufism etc.)
-                2. Must be meaningful and inspiring
-                3. Include accurate author attribution and tradition
-                4. Ensure it's authentic and verifiable
-                5. Keep the quote concise and impactful
-                Return in JSON format with keys: quote, author, tradition""",
-                expected_output="""A JSON object containing:
+                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 spiritual quote text",
-                    "author": "Name of the author or source",
-                    "tradition": "The spiritual tradition (e.g., Buddhism, Hinduism, Taoism)"
+                    "quote": "The quote text",
+                    "author": "Author name",
+                    "tradition": "Spiritual tradition",
+                    "theme": "Primary theme"
                 }""",
                 agent=quote_curator
             )
@@ -89,45 +128,60 @@ class ContentGenerator:
                 verbose=True
             )
 
-            log_message("🤖 Agent starting task...")
             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")
-            raise
+            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)
-            parsed_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")
             }
-            log_message(f"📝 Generated quote: '{parsed_result['text']}' - {parsed_result['author']}")
-            return parsed_result
         except Exception as e:
             log_message(f"❌ Error parsing quote result: {str(e)}", "error")
             raise
 
-    def generate_image(self, quote, tradition):
-        """Generate an image using OpenAI DALL-E"""
+    def generate_image(self, quote, tradition, theme):
+        """Generate an image using OpenAI DALL-E with enhanced prompts"""
         try:
             log_message("🎨 Starting image generation")
-            prompt = f"""Create a serene and spiritual image that represents this quote: '{quote}'
-            From {tradition} tradition.
-            Style: Minimalistic, inspirational, Instagram-worthy.
-            Requirements: Leave space for text overlay, balanced composition.
-            Colors: Peaceful and harmonious tones.
-            The image should be suitable for social media and should not include any text."""
+            
+            # Generate varied image styles based on tradition and theme
+            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."""
 
-            log_message(f"📤 Sending prompt to DALL-E")
             response = self.openai_client.images.generate(
                 model="dall-e-3",
                 prompt=prompt,
@@ -141,21 +195,36 @@ class ContentGenerator:
             log_message(f"❌ Image generation failed: {str(e)}", "error")
             raise
 
-    def generate_audio(self, tradition):
-        """Generate background music using MusicGen"""
+    def generate_audio(self, tradition, theme):
+        """Generate background music using MusicGen with enhanced variety"""
         try:
             log_message("🎵 Starting audio generation")
-            prompts = {
-                "Hinduism": "Indian meditation music with soft flute and gentle tabla, peaceful and calming",
-                "Buddhism": "Tibetan singing bowls with gentle bells and deep resonance, meditative atmosphere",
-                "Taoism": "Chinese zen music with flowing water sounds and bamboo flute, tranquil and balanced",
-                "default": "Peaceful ambient meditation music with soft drones and gentle bells"
+            
+            # Enhanced music prompts based on tradition and theme
+            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"
+                }
             }
             
-            prompt = prompts.get(tradition, prompts["default"])
-            log_message(f"🎼 Selected music style: {tradition}")
+            tradition_prompts = music_prompts.get(tradition, music_prompts["default"])
+            prompt = tradition_prompts.get(theme, tradition_prompts["default"])
             
-            log_message("📤 Sending prompt to MusicGen")
             output = self.replicate_client.run(
                 "meta/musicgen:7be0f12c54a8d0d0d720305c1e6ea10c48d5f0a3afce48478341a0fe682a8787",
                 input={
@@ -166,15 +235,13 @@ class ContentGenerator:
                 }
             )
             
-            if output:
-                log_message("✨ Audio generated successfully")
-                return output
-            else:
-                raise Exception("No output received from MusicGen")
+            log_message("✨ Audio generated successfully")
+            return output
         except Exception as e:
             log_message(f"❌ Audio generation failed: {str(e)}", "error")
             raise
 
+# ... [Rest of the code including main() function remains the same] ...
 def generate_hashtags(tradition):
     """Generate relevant hashtags"""
     base_tags = ["#spirituality", "#mindfulness", "#wisdom", "#inspiration", "#meditation"]