Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import os
 from datetime import datetime
-import json
 from crewai import Agent, Task, Crew
 from langchain_openai import OpenAI
 from openai import OpenAI as OpenAIClient
@@ -12,13 +11,10 @@ import random
 from PIL import Image, ImageDraw, ImageFont
 from io import BytesIO
 
-# Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-
 def log_message(message, type="info"):
-    """Add a message to the session state logs with timestamp"""
     if 'process_logs' not in st.session_state:
         st.session_state.process_logs = []
     timestamp = datetime.now().strftime("%H:%M:%S")
@@ -28,44 +24,17 @@ def log_message(message, type="info"):
         "type": type
     })
 
-
 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"
-        ]
+        "Eastern": ["Buddhism", "Hinduism", "Taoism", "Zen", "Vedanta", "Yoga Philosophy"],
+        "Western": ["Christian Mysticism", "Sufi Wisdom", "Kabbalah", "Greek Philosophy"],
+        "Indigenous": ["Native American Wisdom", "Aboriginal Spirituality", "African Traditional"],
+        "Modern": ["Mindfulness", "Contemporary Philosophy", "Integrative Spirituality"]
     }
-
-    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"
-    ]
-
+    THEMES = ["Inner Peace", "Mindfulness", "Compassion", "Wisdom", "Self-Discovery", 
+              "Unity", "Balance", "Transformation", "Enlightenment", "Purpose"]
+    QUOTE_STYLES = ["contemplative", "inspirational", "philosophical", "practical wisdom", 
+                   "mystical insight", "ancient teaching", "modern interpretation"]
 
 class ContentGenerator:
     def __init__(self, openai_api_key, replicate_api_key):
@@ -74,7 +43,6 @@ class ContentGenerator:
         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)
@@ -84,58 +52,26 @@ class ContentGenerator:
         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"
-            }
+            {"text": "Be the change you wish to see in the world.",
+             "author": "Mahatma Gandhi", "tradition": "Modern Spirituality"},
+            {"text": "Peace comes from within. Do not seek it without.",
+             "author": "Buddha", "tradition": "Buddhism"},
+            {"text": "Love is the bridge between you and everything.",
+             "author": "Rumi", "tradition": "Sufi Wisdom"}
         ]
         quote = random.choice(fallback_quotes)
-        quote["generated_at"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        quote["source"] = "fallback"
+        quote.update({
+            "theme": random.choice(SpiritualThemes.THEMES),
+            "generated_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+            "source": "fallback"
+        })
         log_message("ℹ️ Using fallback quote system", "info")
         return quote
 
-    def _parse_quote_result(self, result):
-    
-        try:
-            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)}")
-            return self._get_fallback_quote()
-
     def generate_quote(self):
         try:
             log_message("🎯 Starting quote generation")
-            prompt_variation = self._generate_prompt_variation()
-            
             response = self.openai_client.chat.completions.create(
                 model="gpt-4",
                 messages=[{
@@ -143,49 +79,40 @@ class ContentGenerator:
                     "content": "Generate spiritual quotes with author, tradition and theme."
                 }, {
                     "role": "user",
-                    "content": prompt_variation + "\nProvide quote in this format only:\nQUOTE:\nAUTHOR:\nTRADITION:\nTHEME:"
+                    "content": self._generate_prompt_variation() + 
+                              "\nProvide quote in this format only:\nQUOTE:\nAUTHOR:\nTRADITION:\nTHEME:"
                 }],
                 temperature=0.7
             )
             
             text = response.choices[0].message.content
-            lines = text.strip().split('\n')
-            result = {}
-            for line in lines:
-                if ':' in line:
-                    key, value = line.split(':', 1)
-                    result[key.strip().lower()] = value.strip()
-            return self._parse_quote_result(result)
-            
+            result = dict(line.split(':', 1) for line in text.strip().split('\n') if ':' in line)
+            return {
+                "text": result.get('quote', '').strip(),
+                "author": result.get('author', '').strip(),
+                "tradition": result.get('tradition', '').strip(),
+                "theme": result.get('theme', '').strip(),
+                "generated_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            }
         except Exception as e:
             log_message(f"❌ Quote generation failed: {str(e)}")
             return self._get_fallback_quote()
-    
+
     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."""
+            Requirements: Suitable for Instagram, space for text overlay, balanced composition"""
+            
             response = self.openai_client.images.generate(
                 model="dall-e-3",
                 prompt=prompt,
@@ -203,40 +130,24 @@ class ContentGenerator:
         try:
             log_message("🎵 Starting audio generation")
             music_prompts = {
-                "Buddhism": {
-                    "Inner Peace": "Tibetan singing bowls and gentle bells with peaceful ambient drones",
-                    "Mindfulness": "Zen flute melodies with soft water sounds",
-                    "default": "Meditation bells with nature sounds"
-                },
-                "Hinduism": {
-                    "Devotion": "Indian bansuri flute with gentle tabla rhythms",
-                    "Wisdom": "Gentle sitar melodies with peaceful tanpura drone",
-                    "default": "Sanskrit chants with soft instrumentation"
-                },
-                "Modern Spirituality": {
-                    "Awareness": "Ambient synthesizer with crystal bowls",
-                    "Growth": "Uplifting piano melodies with gentle ambient pads",
-                    "default": "Modern meditation music with nature sounds"
-                },
-                "default": {
-                    "default": "Peaceful ambient music with gentle bells"
-                }
+                "Buddhism": "Tibetan singing bowls and gentle bells with peaceful ambient drones",
+                "Hinduism": "Indian bansuri flute with gentle tabla rhythms",
+                "Modern Spirituality": "Ambient synthesizer with crystal bowls",
+                "default": "Peaceful ambient music with gentle bells"
             }
-            tradition_prompts = music_prompts.get(tradition, music_prompts["default"])
-            prompt = tradition_prompts.get(theme, tradition_prompts["default"])
+            prompt = music_prompts.get(tradition, music_prompts["default"])
+            
             output = self.replicate_client.run(
                 "meta/musicgen:671ac645ce5e552cc63a54a2bbff63fcf798043055d2dac5fc9e36a837eedcfb",
                 input={
                     "prompt": prompt,
                     "model_version": "stereo-large",
                     "output_format": "mp3",
-                    "normalization_strategy": "peak",
                     "duration": 15
                 }
             )
-            audio_url = output if isinstance(output, str) else str(output)
             log_message("✨ Audio generated successfully")
-            return audio_url
+            return output if isinstance(output, str) else str(output)
         except Exception as e:
             log_message(f"❌ Audio generation failed: {str(e)}", "error")
             raise
@@ -246,97 +157,38 @@ class ContentGenerator:
             response = requests.get(image_url)
             img = Image.open(BytesIO(response.content))
             width, height = img.size
-
-            # Create blank image with text
             txt = Image.new('RGBA', (width, height), (255, 255, 255, 0))
             d = ImageDraw.Draw(txt)
 
-            # Semi-transparent background for text
-            d.rectangle([0, height / 2 - 100, width, height / 2 + 100], 
-                        fill=(0, 0, 0, 127))
-
+            d.rectangle([0, height/2 - 100, width, height/2 + 100], fill=(0, 0, 0, 127))
+            
             try:
                 quote_font = ImageFont.truetype("DejaVuSans-Bold", 40)
                 author_font = ImageFont.truetype("DejaVuSans", 30)
             except:
-                quote_font = ImageFont.load_default()
-                author_font = ImageFont.load_default()
+                quote_font = author_font = ImageFont.load_default()
 
-            # Add text
-            d.text((width / 2, height / 2), quote_text, 
-                   font=quote_font, fill=(255, 255, 255, 255), anchor="mm")
-            d.text((width / 2, height / 2 + 60), f"- {author}", 
-                   font=author_font, fill=(255, 255, 255, 255), anchor="mm")
+            d.text((width/2, height/2), quote_text, font=quote_font, 
+                   fill=(255, 255, 255, 255), anchor="mm")
+            d.text((width/2, height/2 + 60), f"- {author}", font=author_font,
+                   fill=(255, 255, 255, 255), anchor="mm")
 
-            # Combine images
             out = Image.alpha_composite(img.convert('RGBA'), txt)
-
-            # Convert to bytes
             buffer = BytesIO()
             out.convert('RGB').save(buffer, format='JPEG')
             return buffer.getvalue()
-
         except Exception as e:
             log_message(f"Error creating overlay: {str(e)}")
             return None
 
-
-    def display_debug_logs():
-        """Display debug logs with enhanced visualization"""
-        st.markdown("### 🔍 Process Logs")
-        # Create tabs for different log views
-        log_tabs = st.tabs(["All Logs", "Errors Only", "Timeline"])
-        with log_tabs[0]:
-            for log in st.session_state.process_logs:
-                if log["type"] == "error":
-                    st.error(f"{log['timestamp']} - {log['message']}")
-                else:
-                    st.info(f"{log['timestamp']} - {log['message']}")
-        with log_tabs[1]:
-            errors = [log for log in st.session_state.process_logs if log["type"] == "error"]
-            if errors:
-                for error in errors:
-                    st.error(f"{error['timestamp']} - {error['message']}")
-            else:
-                st.success("No errors found! 🎉")
-        with log_tabs[2]:
-            st.markdown("#### Generation Timeline")
-            for log in st.session_state.process_logs:
-                with st.expander(f"{log['timestamp']} - {log['message'][:50]}...", expanded=False):
-                    st.text(log['message'])
-
-
-    def generate_hashtags(tradition, theme):
-        """Generate relevant hashtags"""
-        base_tags = ["#spirituality", "#mindfulness", "#wisdom", "#inspiration", "#meditation"]
-        tradition_tags = {
-            "Buddhism": ["#buddhism", "#zen", "#buddhist", "#enlightenment", "#dharma"],
-            "Hinduism": ["#hinduism", "#vedanta", "#yoga", "#sanskrit", "#consciousness"],
-            "Taoism": ["#taoism", "#tao", "#balance", "#harmony", "#philosophy"],
-            "Sufi Wisdom": ["#sufi", "#rumi", "#mysticism", "#spiritualpath", "#divine"],
-            "Modern Spirituality": ["#spiritual", "#awakening", "#higherconsciousness", "#growth", "#innerpeace"]
-        }
-        theme_tags = {
-            "Inner Peace": ["#peace", "#calm", "#serenity", "#tranquility"],
-            "Mindfulness": ["#mindful", "#present", "#awareness", "#conscious"],
-            "Wisdom": ["#wisdom", "#truth", "#knowledge", "#insight"],
-            "Transformation": ["#transform", "#change", "#evolution", "#journey"]
-        }
-        custom_tags = tradition_tags.get(tradition, []) + theme_tags.get(theme, [])
-        return list(set(base_tags + custom_tags))[:15]
-
-
-    def display_debug_logs():
-        """Display debug logs with enhanced visualization"""
+    def display_debug_logs(self):
         st.markdown("### 🔍 Process Logs")
         log_tabs = st.tabs(["All Logs", "Errors Only", "Timeline"])
         
         with log_tabs[0]:
             for log in st.session_state.process_logs:
-                if log["type"] == "error":
-                    st.error(f"{log['timestamp']} - {log['message']}")
-                else:
-                    st.info(f"{log['timestamp']} - {log['message']}")
+                st.info(f"{log['timestamp']} - {log['message']}" if log["type"] != "error"
+                       else st.error(f"{log['timestamp']} - {log['message']}"))
         
         with log_tabs[1]:
             errors = [log for log in st.session_state.process_logs if log["type"] == "error"]
@@ -352,81 +204,65 @@ class ContentGenerator:
                 with st.expander(f"{log['timestamp']} - {log['message'][:50]}...", expanded=False):
                     st.text(log['message'])
 
+def generate_hashtags(tradition, theme):
+    base_tags = ["#spirituality", "#mindfulness", "#wisdom", "#inspiration"]
+    tradition_tags = {
+        "Buddhism": ["#buddhism", "#zen", "#buddhist"],
+        "Hinduism": ["#hinduism", "#vedanta", "#yoga"],
+        "Taoism": ["#taoism", "#tao", "#balance"],
+        "Modern Spirituality": ["#spiritual", "#awakening", "#growth"]
+    }
+    theme_tags = {
+        "Inner Peace": ["#peace", "#calm", "#serenity"],
+        "Mindfulness": ["#mindful", "#present", "#awareness"],
+        "Wisdom": ["#wisdom", "#truth", "#knowledge"]
+    }
+    custom_tags = tradition_tags.get(tradition, []) + theme_tags.get(theme, [])
+    return list(set(base_tags + custom_tags))[:10]
+
 def main():
-    st.set_page_config(
-        page_title="Spiritual Content Generator",
-        page_icon="🕉️",
-        layout="wide"
-    )
+    st.set_page_config(page_title="Spiritual Content Generator", page_icon="🕉️", layout="wide")
 
-    # Initialize session state
-    if 'process_logs' not in st.session_state:
-        st.session_state.process_logs = []
-    if 'generated_content' not in st.session_state:
-        st.session_state.generated_content = None
-    if 'generated_image' not in st.session_state:
-        st.session_state.generated_image = None
-    if 'generated_audio' not in st.session_state:
-        st.session_state.generated_audio = None
+    for state_var in ['process_logs', 'generated_content', 'generated_image', 'generated_audio']:
+        if state_var not in st.session_state:
+            st.session_state[state_var] = None
 
-    # Check for API keys
-    if 'OPENAI_API_KEY' not in st.secrets:
-        st.error("OpenAI API key not found in secrets!")
-        st.stop()
-    if 'REPLICATE_API_TOKEN' not in st.secrets:
-        st.error("Replicate API token not found in secrets!")
+    if not all(key in st.secrets for key in ['OPENAI_API_KEY', 'REPLICATE_API_TOKEN']):
+        st.error("Missing API keys in secrets!")
         st.stop()
 
-    # Tabs
     tab1, tab2, tab3 = st.tabs(["Generate Content", "Instagram Preview", "Debug Log"])
 
     with tab1:
         st.title("🕉️ Spiritual Content Generator")
-
-        if st.button("Generate New Content", key="generate_button"):
+        
+        if st.button("Generate New Content"):
             try:
-                with st.spinner("Initializing content generation..."):
-                    generator = ContentGenerator(
-                        st.secrets["OPENAI_API_KEY"],
-                        st.secrets["REPLICATE_API_TOKEN"]
-                    )
-
-                    # Generate quote
-                    with st.status("Generating spiritual quote...") as status:
-                        quote_data = generator.generate_quote()
-                        st.session_state.generated_content = quote_data
-                        status.update(label="Quote generated successfully!", state="complete")
-
-                    # Generate image
-                    with st.status("Creating visual representation...") as status:
-                        image_url = generator.generate_image(
-                            quote_data["text"],
-                            quote_data["tradition"],
-                            quote_data["theme"]
-                        )
-                        st.session_state.generated_image = image_url
-                        status.update(label="Image generated successfully!", state="complete")
-
-                    # Generate audio
-                    with st.status("Composing background music...") as status:
-                        audio_url = generator.generate_audio(
-                            quote_data["tradition"],
-                            quote_data["theme"]
-                        )
-                        st.session_state.generated_audio = audio_url
-                        status.update(label="Audio generated successfully!", state="complete")
-
-                    st.success("✨ All content generated successfully!")
-
+                generator = ContentGenerator(st.secrets["OPENAI_API_KEY"], 
+                                          st.secrets["REPLICATE_API_TOKEN"])
+
+                with st.spinner("Generating content..."):
+                    quote_data = generator.generate_quote()
+                    st.session_state.generated_content = quote_data
+                    
+                    image_url = generator.generate_image(quote_data["text"], 
+                                                       quote_data["tradition"],
+                                                       quote_data["theme"])
+                    st.session_state.generated_image = image_url
+                    
+                    audio_url = generator.generate_audio(quote_data["tradition"],
+                                                       quote_data["theme"])
+                    st.session_state.generated_audio = audio_url
+
+                st.success("✨ Content generated successfully!")
+                
             except Exception as e:
                 st.error(f"Error during generation: {str(e)}")
 
-        # Display generated content
         if st.session_state.generated_content:
-            st.subheader("Generated Content Details")
+            st.subheader("Generated Content")
             st.json(st.session_state.generated_content)
 
-    # Replace the tab2 section with:
     with tab2:
         st.title("Instagram Preview")
         if all([st.session_state.generated_content, 
@@ -434,14 +270,11 @@ def main():
                 st.session_state.generated_audio]):
 
             col1, col2 = st.columns([2, 1])
-
+            
             with col1:
-                # Image with quote overlay
+                generator = ContentGenerator(st.secrets["OPENAI_API_KEY"],
+                                          st.secrets["REPLICATE_API_TOKEN"])
                 quote_data = st.session_state.generated_content
-                generator = ContentGenerator(
-                    st.secrets["OPENAI_API_KEY"],
-                    st.secrets["REPLICATE_API_TOKEN"]
-                )
                 final_image = generator.create_image_with_quote(
                     st.session_state.generated_image,
                     quote_data['text'],
@@ -449,12 +282,8 @@ def main():
                 )
                 if final_image:
                     st.image(final_image)
-                    st.download_button(
-                        "📥 Download Image",
-                        data=final_image,
-                        file_name="spiritual_quote.jpg",
-                        mime="image/jpeg"
-                    )
+                    st.download_button("📥 Download Image", final_image,
+                                     "spiritual_quote.jpg", "image/jpeg")
 
             with col2:
                 st.subheader("🎵 Background Music")
@@ -463,17 +292,12 @@ def main():
                         audio_response = requests.get(st.session_state.generated_audio)
                         if audio_response.status_code == 200:
                             st.audio(audio_response.content, format='audio/mp3')
-                            st.download_button(
-                                "📥 Download Audio",
-                                data=audio_response.content,
-                                file_name="background_music.mp3",
-                                mime="audio/mp3"
-                            )
+                            st.download_button("📥 Download Audio", audio_response.content,
+                                             "background_music.mp3", "audio/mp3")
+                            
                             st.subheader("📝 Caption")
-                            hashtags = generate_hashtags(
-                                quote_data['tradition'],
-                                quote_data.get('theme', 'Wisdom')
-                            )
+                            hashtags = generate_hashtags(quote_data['tradition'],
+                                                      quote_data.get('theme', 'Wisdom'))
                             caption = f"""✨ Wisdom from {quote_data['tradition']} ✨
 
 "{quote_data['text']}"
@@ -481,16 +305,15 @@ def main():
 
 {' '.join(hashtags)}"""
                             st.text_area("Instagram Caption", caption, height=200)
-                        else:
-                            st.error("Failed to load audio preview")
                     except Exception as e:
                         st.error(f"Error playing audio: {str(e)}")
         else:
             st.info("Generate content first to see the Instagram preview")
             
     with tab3:
-        display_debug_logs()
-
+        generator = ContentGenerator(st.secrets["OPENAI_API_KEY"],
+                                  st.secrets["REPLICATE_API_TOKEN"])
+        generator.display_debug_logs()
 
 if __name__ == "__main__":
-    main()
+    main()