Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import os
-import json  # Fix: Import json to use json functionalities
+import json
 from datetime import datetime
 from crewai import Agent, Task, Crew
 from langchain_openai import OpenAI
@@ -22,11 +22,17 @@ BASE_DIR = "data"
 QUOTES_DIR = os.path.join(BASE_DIR, "quotes")
 IMAGES_DIR = os.path.join(BASE_DIR, "images")
 FINAL_IMAGES_DIR = os.path.join(BASE_DIR, "final_images")
+MUSIC_DIR = os.path.join(BASE_DIR, "music")  # New directory for storing music files
 
 # Ensure the directories exist
-os.makedirs(QUOTES_DIR, exist_ok=True)
-os.makedirs(IMAGES_DIR, exist_ok=True)
-os.makedirs(FINAL_IMAGES_DIR, exist_ok=True)
+try:
+    os.makedirs(QUOTES_DIR, exist_ok=True)
+    os.makedirs(IMAGES_DIR, exist_ok=True)
+    os.makedirs(FINAL_IMAGES_DIR, exist_ok=True)
+    os.makedirs(MUSIC_DIR, exist_ok=True)  # Create music directory
+    logger.info(f"Directories successfully created or verified: {BASE_DIR}, {QUOTES_DIR}, {IMAGES_DIR}, {FINAL_IMAGES_DIR}, {MUSIC_DIR}")
+except Exception as e:
+    logger.error(f"Error creating directories: {str(e)}")
 
 def log_message(message, type="info"):
     if 'process_logs' not in st.session_state:
@@ -115,10 +121,15 @@ class ContentGenerator:
             # Save the quote to a file
             filename = f"quote_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
             file_path = os.path.join(QUOTES_DIR, filename)
-            with open(file_path, 'w') as f:
-                json.dump(result, f, indent=4)
+            try:
+                with open(file_path, 'w') as f:
+                    json.dump(result, f, indent=4)
+                log_message(f"✨ Generated quote saved at {file_path}")
+                logger.info(f"Quote saved successfully at: {file_path}")
+            except Exception as e:
+                log_message(f"❌ Failed to save quote to file: {str(e)}", "error")
+                logger.error(f"Error saving quote: {str(e)}")
             
-            log_message(f"✨ Generated quote saved at {file_path}")
             return result
             
         except Exception as e:
@@ -154,15 +165,65 @@ class ContentGenerator:
             if image_response.status_code == 200:
                 filename = f"image_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
                 file_path = os.path.join(IMAGES_DIR, filename)
-                with open(file_path, 'wb') as f:
-                    f.write(image_response.content)
-                log_message(f"✨ Image generated and saved at {file_path}")
+                try:
+                    with open(file_path, 'wb') as f:
+                        f.write(image_response.content)
+                    log_message(f"✨ Image generated and saved at {file_path}")
+                    logger.info(f"Image saved successfully at: {file_path}")
+                except Exception as e:
+                    log_message(f"❌ Failed to save image to file: {str(e)}", "error")
+                    logger.error(f"Error saving image: {str(e)}")
             else:
                 raise Exception("Failed to download image")
 
             return image_url
         except Exception as e:
             log_message(f"❌ Image generation failed: {str(e)}", "error")
+            logger.error(f"Image generation failed: {str(e)}")
+            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
+                }
+            )
+            audio_url = output if isinstance(output, str) else str(output)
+            
+            # Download and save the generated music
+            music_response = requests.get(audio_url)
+            if music_response.status_code == 200:
+                filename = f"music_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp3"
+                file_path = os.path.join(MUSIC_DIR, filename)
+                try:
+                    with open(file_path, 'wb') as f:
+                        f.write(music_response.content)
+                    log_message(f"✨ Music generated and saved at {file_path}")
+                    logger.info(f"Music saved successfully at: {file_path}")
+                except Exception as e:
+                    log_message(f"❌ Failed to save music to file: {str(e)}", "error")
+                    logger.error(f"Error saving music: {str(e)}")
+            else:
+                raise Exception("Failed to download audio")
+
+            return audio_url
+        except Exception as e:
+            log_message(f"❌ Music generation failed: {str(e)}", "error")
+            logger.error(f"Music generation failed: {str(e)}")
             raise
 
     def create_image_with_quote(self, image_url, quote_text, author):
@@ -218,14 +279,20 @@ class ContentGenerator:
             
             filename = f"final_image_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
             file_path = os.path.join(FINAL_IMAGES_DIR, filename)
-            with open(file_path, 'wb') as f:
-                f.write(final_image_bytes)
+            try:
+                with open(file_path, 'wb') as f:
+                    f.write(final_image_bytes)
+                log_message(f"✨ Final image saved at {file_path}")
+                logger.info(f"Final image saved successfully at: {file_path}")
+            except Exception as e:
+                log_message(f"❌ Failed to save final image to file: {str(e)}", "error")
+                logger.error(f"Error saving final image: {str(e)}")
             
-            log_message(f"✨ Final image saved at {file_path}")
             return final_image_bytes
     
         except Exception as e:
-            log_message(f"Error creating overlay: {str(e)}")
+            log_message(f"Error creating overlay: {str(e)}", "error")
+            logger.error(f"Error creating image with quote: {str(e)}")
             return None
 
 def generate_hashtags(tradition, theme):
@@ -290,7 +357,22 @@ def main():
             except Exception as e:
                 st.error(f"Error generating image: {str(e)}")
 
-        # Step 3: Preview Quote on Image
+        # Step 3: Generate Music
+        if st.session_state.generated_content 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)}")
+
+        # Step 4: 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(