Update app.py

app.py

@@ -249,48 +249,36 @@ class ContentGenerator:
             log_message(f"❌ Image generation failed: {str(e)}", "error")
             raise
 
-    def generate_audio(self, tradition, theme):
+  def generate_audio(self, tradition, theme):
         """Generate background music using MusicGen"""
         try:
             log_message("🎵 Starting audio generation")
             
-            music_prompts = {
-                "Buddhism": {
-                    "Inner Peace": "Tibetan singing bowls and gentle bells with peaceful ambient drones, meditative and serene",
-                    "Mindfulness": "Zen flute melodies with soft water sounds, creating a mindful atmosphere",
-                    "default": "Meditation bells with nature sounds, peaceful and calming"
-                },
-                "Hinduism": {
-                    "Devotion": "Indian bansuri flute with gentle tabla rhythms, creating a devotional atmosphere",
-                    "Wisdom": "Gentle sitar melodies with peaceful tanpura drone",
-                    "default": "Sanskrit chants with soft instrumentation, peaceful and meditative"
-                },
-                "Modern Spirituality": {
-                    "Awareness": "Ambient synthesizer with crystal bowls, creating a modern meditative atmosphere",
-                    "Growth": "Uplifting piano melodies with gentle ambient pads",
-                    "default": "Modern meditation music with nature sounds and gentle electronics"
-                },
-                "default": {
-                    "default": "Peaceful ambient music with gentle bells and subtle harmonies"
-                }
-            }
-            
             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",  # Updated model version
+                "meta/musicgen:671ac645ce5e552cc63a54a2bbff63fcf798043055d2dac5fc9e36a837eedcfb",
                 input={
                     "prompt": prompt,
-                    "model_version": "stereo-large",  # Using stereo large model
+                    "model_version": "stereo-large",
                     "output_format": "mp3",
                     "normalization_strategy": "peak",
-                    "duration": 15  # Length in seconds
+                    "duration": 15
                 }
             )
             
+            # Convert FileOutput to URL if needed
+            if hasattr(output, 'url'):
+                audio_url = output.url
+            elif isinstance(output, str):
+                audio_url = output
+            else:
+                audio_url = str(output)
+            
             log_message("✨ Audio generated successfully")
-            return output
+            return audio_url
         except Exception as e:
             log_message(f"❌ Audio generation failed: {str(e)}", "error")
             raise
@@ -417,93 +405,28 @@ def main():
             st.subheader("Generated Content Details")
             st.json(st.session_state.generated_content)
 
-    with tab2:
-        st.title("Instagram Preview")
-        
-        if all([st.session_state.generated_content, 
-                st.session_state.generated_image,
-                st.session_state.generated_audio]):
-            
-            col1, col2 = st.columns([2, 1])
-            
-            with col1:
-                # Display generated image with quote overlay
-                st.image(st.session_state.generated_image, use_column_width=True)
-                
-                # Quote overlay
-                quote = st.session_state.generated_content
-                st.markdown(f"""
-                <div style='
-                    background-color: rgba(255,255,255,0.8);
-                    padding: 20px;
-                    border-radius: 10px;
-                    margin: 10px 0;
-                    text-align: center;
-                    box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-                '>
-                    <h2 style='font-size: 24px; margin-bottom: 10px;'>{quote['text']}</h2>
-                    <p style='font-style: italic;'>- {quote['author']}</p>
-                </div>
-                """, unsafe_allow_html=True)
-
-            with col2:
-                # Audio preview
-                st.subheader("🎵 Background Music")
-                if st.session_state.generated_audio:
-                    st.audio(st.session_state.generated_audio, format='audio/mp3')
-                
-                # Caption generation
-                st.subheader("📝 Caption")
-                hashtags = generate_hashtags(
-                    quote['tradition'],
-                    quote.get('theme', 'Wisdom')
-                )
-                caption = f"""✨ Wisdom from {quote['tradition']} ✨
-
-"{quote['text']}"
-- {quote['author']}
-
-Theme: {quote.get('theme', 'Wisdom')}
-
-{' '.join(hashtags)}"""
-                
-                st.text_area("Instagram Caption", caption, height=200)
-                
-                # Download buttons
-                col1, col2 = st.columns(2)
-                with col1:
-                    if st.session_state.generated_image:
-                        image_data = requests.get(st.session_state.generated_image).content
-                        st.download_button(
-                            "📥 Download Image",
-                            data=image_data,
-                            file_name="spiritual_quote.jpg",
-                            mime="image/jpeg"
-                        )
-                
-                with col2:
-                    if st.session_state.generated_audio:
-                        audio_data = requests.get(st.session_state.generated_audio).content
-                        st.download_button(
-                            "📥 Download Audio",
-                            data=audio_data,
-                            file_name="background_music.mp3",
-                            mime="audio/mp3"
-                        )
-                
-                # Posting tips
-                st.markdown("""
-                ### 📱 Posting Tips
-                1. Download the image and audio
-                2. Combine them using any video editor
-                3. Copy the generated caption
-                4. Best posting times: 8-10 AM or 6-8 PM
-                5. Engage with your audience about the quote's meaning
-                6. Use Instagram's Reels for better reach
-                """)
-
-        else:
-            st.info("Generate content first to see the Instagram preview")
+    with col2:
+        # Audio preview
+        st.subheader("🎵 Background Music")
+        if st.session_state.generated_audio:
+            try:
+                # Get audio data from URL
+                audio_response = requests.get(st.session_state.generated_audio)
+                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",
+                        data=audio_response.content,
+                        file_name="background_music.mp3",
+                        mime="audio/mp3"
+                    )
+                else:
+                    st.error("Failed to load audio preview")
+            except Exception as e:
+                st.error(f"Error playing audio: {str(e)}")
 
     with tab3:
         display_debug_logs()