Update app.py

app.py

@@ -136,32 +136,6 @@ class ContentGenerator:
             log_message(f"❌ Image generation failed: {str(e)}", "error")
             raise
 
-    def generate_audio(self, tradition, theme):
-        try:
-            log_message("🎵 Starting audio generation")
-            music_prompts = {
-                "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"
-            }
-            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",
-                    "duration": 15
-                }
-            )
-            log_message("✨ Audio generated successfully")
-            return output if isinstance(output, str) else str(output)
-        except Exception as e:
-            log_message(f"❌ Audio generation failed: {str(e)}", "error")
-            raise
-
     def create_image_with_quote(self, image_url, quote_text, author):
         try:
             response = requests.get(image_url)
@@ -188,7 +162,11 @@ class ContentGenerator:
             # Draw each line of the wrapped text
             y_offset = height / 2 - 100  # Starting Y position for the quote
             for line in lines:
-                line_width, line_height = d.textsize(line, font=quote_font)
+                # Calculate the text size using textbbox
+                bbox = d.textbbox((0, 0), line, font=quote_font)
+                line_width = bbox[2] - bbox[0]
+                line_height = bbox[3] - bbox[1]
+    
                 x_position = (width - line_width) / 2  # Center the line horizontally
                 d.text((x_position, y_offset), line, font=quote_font, fill=(255, 255, 255, 255))
                 y_offset += line_height + 5  # Move to the next line, with a small line gap
@@ -196,7 +174,8 @@ class ContentGenerator:
             # Draw the author text below the quote
             author_y_position = y_offset + 20  # Leave some space below the quote for the author
             author_text = f"- {author}"
-            author_width, author_height = d.textsize(author_text, font=author_font)
+            author_bbox = d.textbbox((0, 0), author_text, font=author_font)
+            author_width = author_bbox[2] - author_bbox[0]
             author_x_position = (width - author_width) / 2
             d.text((author_x_position, author_y_position), author_text, font=author_font, fill=(255, 255, 255, 255))
     
@@ -212,29 +191,6 @@ class ContentGenerator:
             log_message(f"Error creating overlay: {str(e)}")
             return None
 
-    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:
-                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"]
-            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):
     base_tags = ["#spirituality", "#mindfulness", "#wisdom", "#inspiration"]
     tradition_tags = {
@@ -271,10 +227,9 @@ def main():
     with col1:
         st.title("🕉️ Spiritual Content Generator")
         
-        generator = ContentGenerator(st.secrets["OPENAI_API_KEY"], 
-                                  st.secrets["REPLICATE_API_TOKEN"])
+        generator = ContentGenerator(st.secrets["OPENAI_API_KEY"], st.secrets["REPLICATE_API_TOKEN"])
 
-        # Stepped generation buttons
+        # Step 1: Generate Quote
         if st.button("Generate Quote"):
             try:
                 quote_data = generator.generate_quote()
@@ -284,6 +239,7 @@ def main():
             except Exception as e:
                 st.error(f"Error generating quote: {str(e)}")
 
+        # Step 2: Create Image with Quote
         if st.session_state.generated_content and st.button("Create Image"):
             try:
                 image_url = generator.generate_image(
@@ -297,53 +253,48 @@ def main():
             except Exception as e:
                 st.error(f"Error generating image: {str(e)}")
 
-        if st.session_state.generated_image and st.button("Generate Music"):
-            try:
-                audio_url = generator.generate_audio(
-                    st.session_state.generated_content["tradition"],
-                    st.session_state.generated_content["theme"]
-                )
-                st.session_state.generated_audio = audio_url
-                st.success("✨ Music generated!")
-                audio_response = requests.get(audio_url)
-                if audio_response.status_code == 200:
-                    st.audio(audio_response.content, format='audio/mp3')
-            except Exception as e:
-                st.error(f"Error generating audio: {str(e)}")
-
-        # Preview tab content
-        if all([st.session_state.generated_content, 
-                st.session_state.generated_image,
-                st.session_state.generated_audio]):
+        # Step 3: Preview Quote on Image
+        if st.session_state.generated_content and st.session_state.generated_image:
             st.subheader("Final Preview")
-            
             final_image = generator.create_image_with_quote(
                 st.session_state.generated_image,
                 st.session_state.generated_content['text'],
                 st.session_state.generated_content['author']
             )
-            
             if final_image:
                 st.image(final_image)
                 st.download_button("📥 Download Image", final_image,
-                                 "spiritual_quote.jpg", "image/jpeg")
+                                   "spiritual_quote.jpg", "image/jpeg")
 
-            audio_response = requests.get(st.session_state.generated_audio)
-            if audio_response.status_code == 200:
-                st.download_button("📥 Download Audio", audio_response.content,
-                                 "background_music.mp3", "audio/mp3")
-                
-                hashtags = generate_hashtags(
-                    st.session_state.generated_content['tradition'],
-                    st.session_state.generated_content.get('theme', 'Wisdom')
-                )
-                caption = f"""✨ Wisdom from {st.session_state.generated_content['tradition']} ✨
+            # Generate Instagram Caption
+            hashtags = generate_hashtags(
+                st.session_state.generated_content['tradition'],
+                st.session_state.generated_content.get('theme', 'Wisdom')
+            )
+            caption = f"""✨ Wisdom from {st.session_state.generated_content['tradition']} ✨
 
 "{st.session_state.generated_content['text']}"
 - {st.session_state.generated_content['author']}
 
 {' '.join(hashtags)}"""
-                st.text_area("Instagram Caption", caption, height=200)
+            st.text_area("Instagram Caption", caption, height=200)
+
+        # Optional Step: Generate Music
+        if st.session_state.generated_image and st.button("Generate Music"):
+            try:
+                audio_url = generator.generate_audio(
+                    st.session_state.generated_content["tradition"],
+                    st.session_state.generated_content["theme"]
+                )
+                st.session_state.generated_audio = audio_url
+                st.success("✨ Music generated!")
+                audio_response = requests.get(audio_url)
+                if audio_response.status_code == 200:
+                    st.audio(audio_response.content, format='audio/mp3')
+                    st.download_button("📥 Download Audio", audio_response.content,
+                                       "background_music.mp3", "audio/mp3")
+            except Exception as e:
+                st.error(f"Error generating audio: {str(e)}")
 
     # Live debug log
     with col2:
@@ -361,4 +312,4 @@ def main():
         update_logs()
 
 if __name__ == "__main__":
-    main()
+    main()