Update app.py

app.py

@@ -384,94 +384,71 @@ def main():
             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:
-           # Image with quote overlay
-           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'],
-               quote_data['author']
-           )
-           if final_image:
-               st.image(final_image)
-               st.download_button(
-                   "📥 Download Image",
-                   data=final_image,
-                   file_name="spiritual_quote.jpg",
-                   mime="image/jpeg"
-               )
-
-       with col2:
-           # Audio preview
-           st.subheader("🎵 Background Music")
-           if st.session_state.generated_audio:
-               try:
-                   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"
-                       )
-                       
-                       # Add caption and hashtags
-                       st.subheader("📝 Caption")
-                       hashtags = generate_hashtags(
-                           quote_data['tradition'],
-                           quote_data.get('theme', 'Wisdom')
-                       )
-                       caption = f"""✨ Wisdom from {quote_data['tradition']} ✨
-
-"{quote_data['text']}"
-- {quote_data['author']}
-
-{' '.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")
-    if final_image:
-        st.image(final_image)
-
-        # 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.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:
+                # Image with quote overlay
+                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'],
+                    quote_data['author']
+                )
+                if final_image:
+                    st.image(final_image)
                     st.download_button(
-                        "📥 Download Audio",
-                        data=audio_response.content,
-                        file_name="background_music.mp3",
-                        mime="audio/mp3"
+                        "📥 Download Image",
+                        data=final_image,
+                        file_name="spiritual_quote.jpg",
+                        mime="image/jpeg"
                     )
-                else:
-                    st.error("Failed to load audio preview")
-            except Exception as e:
-                st.error(f"Error playing audio: {str(e)}")
 
+            with col2:
+                # Audio preview
+                st.subheader("🎵 Background Music")
+                if st.session_state.generated_audio:
+                    try:
+                        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"
+                            )
+
+                        # Add caption and hashtags
+                        st.subheader("📝 Caption")
+                        hashtags = generate_hashtags(
+                            quote_data['tradition'],
+                            quote_data.get('theme', 'Wisdom')
+                        )
+                        caption = f"""✨ Wisdom from {quote_data['tradition']} ✨
+
+    "{quote_data['text']}"
+    - {quote_data['author']}
+
+    {' '.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()