Update app.py

app.py

@@ -14,6 +14,7 @@ import random
 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:
@@ -25,9 +26,10 @@ def log_message(message, type="info"):
         "type": type
     })
 
+
 class SpiritualThemes:
     """Diverse spiritual themes for content generation"""
-    
+
     TRADITIONS = {
         "Eastern": [
             "Buddhism", "Hinduism", "Taoism", "Zen", "Vedanta",
@@ -46,7 +48,7 @@ class SpiritualThemes:
             "Integrative Spirituality", "Secular Wisdom"
         ]
     }
-    
+
     THEMES = [
         "Inner Peace", "Mindfulness", "Compassion", "Wisdom",
         "Self-Discovery", "Unity", "Balance", "Transformation",
@@ -54,7 +56,7 @@ class SpiritualThemes:
         "Love", "Harmony", "Truth", "Liberation", "Service",
         "Connection", "Growth", "Awareness"
     ]
-    
+
     QUOTE_STYLES = [
         "contemplative", "inspirational", "philosophical",
         "practical wisdom", "mystical insight", "ancient teaching",
@@ -62,6 +64,7 @@ class SpiritualThemes:
         "meditation guidance"
     ]
 
+
 class ContentGenerator:
     def __init__(self, openai_api_key, replicate_api_key):
         self.llm = OpenAI(api_key=openai_api_key, temperature=0.7)
@@ -74,7 +77,7 @@ class ContentGenerator:
         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."""
@@ -117,9 +120,9 @@ class ContentGenerator:
         """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',
@@ -134,13 +137,13 @@ class ContentGenerator:
             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 ONLY a JSON object in this exact format, with no additional text:
                 {{
                     "quote": "The actual quote text",
@@ -149,12 +152,12 @@ class ContentGenerator:
                     "theme": "The main theme"
                 }}""",
                 expected_output="""Plain JSON object with no markdown formatting:
-                {
+                {{
                     "quote": "The quote text",
                     "author": "Author name",
                     "tradition": "Spiritual tradition",
                     "theme": "Main theme"
-                }""",
+                }}""",
                 agent=quote_curator
             )
 
@@ -167,7 +170,7 @@ class ContentGenerator:
             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()
@@ -177,24 +180,24 @@ class ContentGenerator:
         try:
             log_message(f"Raw result type: {type(result)}")
             log_message(f"Raw result content: {result}")
-            
+
             # Handle CrewOutput object
             if hasattr(result, 'result'):
                 result_str = str(result.result)
             else:
                 result_str = str(result)
-            
+
             log_message(f"Processing result string: {result_str}")
-            
+
             # Clean the string and extract JSON
             result_str = result_str.strip()
             if result_str.startswith("```json"):
                 result_str = result_str[7:-3]  # Remove ```json and ``` markers
-            
+
             # Parse the JSON
             parsed_json = json.loads(result_str)
             log_message(f"Parsed JSON: {parsed_json}")
-            
+
             return {
                 "text": parsed_json.get('quote', ''),
                 "author": parsed_json.get('author', ''),
@@ -211,7 +214,7 @@ class ContentGenerator:
         """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",
@@ -222,9 +225,9 @@ class ContentGenerator:
                 "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
@@ -253,10 +256,10 @@ class ContentGenerator:
         """Generate background music using MusicGen"""
         try:
             log_message("🎵 Starting audio generation")
-            
+
             tradition_prompts = music_prompts.get(tradition, music_prompts["default"])
             prompt = tradition_prompts.get(theme, tradition_prompts["default"])
-            
+
             # Get the audio file URL
             output = self.replicate_client.run(
                 "meta/musicgen:671ac645ce5e552cc63a54a2bbff63fcf798043055d2dac5fc9e36a837eedcfb",
@@ -268,7 +271,7 @@ class ContentGenerator:
                     "duration": 15
                 }
             )
-            
+
             # Convert FileOutput to URL if needed
             if hasattr(output, 'url'):
                 audio_url = output.url
@@ -276,27 +279,28 @@ class ContentGenerator:
                 audio_url = output
             else:
                 audio_url = str(output)
-            
+
             log_message("✨ Audio generated successfully")
             return audio_url
         except Exception as e:
             log_message(f"❌ Audio generation failed: {str(e)}", "error")
             raise
 
-    def display_debug_logs():
+
+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:
@@ -304,14 +308,15 @@ class ContentGenerator:
                 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):
+
+def generate_hashtags(tradition, theme):
     """Generate relevant hashtags"""
     base_tags = ["#spirituality", "#mindfulness", "#wisdom", "#inspiration", "#meditation"]
     tradition_tags = {
@@ -327,10 +332,11 @@ class ContentGenerator:
         "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 main():
     st.set_page_config(
         page_title="Spiritual Content Generator",
@@ -361,7 +367,7 @@ def main():
 
     with tab1:
         st.title("🕉️ Spiritual Content Generator")
-        
+
         if st.button("Generate New Content", key="generate_button"):
             try:
                 with st.spinner("Initializing content generation..."):
@@ -369,13 +375,13 @@ def main():
                         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(
@@ -385,7 +391,7 @@ def main():
                         )
                         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(
@@ -394,7 +400,7 @@ def main():
                         )
                         st.session_state.generated_audio = audio_url
                         status.update(label="Audio generated successfully!", state="complete")
-                    
+
                     st.success("✨ All content generated successfully!")
 
             except Exception as e:
@@ -405,7 +411,12 @@ def main():
             st.subheader("Generated Content Details")
             st.json(st.session_state.generated_content)
 
-    with col2:
+    with tab2:  # This was corrected from 'col2'
+        # Image preview (this was missing)
+        st.subheader("🖼️ Generated Image")
+        if st.session_state.generated_image:
+            st.image(st.session_state.generated_image, caption="Generated Image")
+
         # Audio preview
         st.subheader("🎵 Background Music")
         if st.session_state.generated_audio:
@@ -415,7 +426,7 @@ def main():
                 if audio_response.status_code == 200:
                     # Create audio player
                     st.audio(audio_response.content, format='audio/mp3')
-                    
+
                     # Download button for audio
                     st.download_button(
                         "📥 Download Audio",