Update app.py

app.py

@@ -142,8 +142,6 @@ class ContentGenerator:
             log_message(f"❌ Quote generation failed: {str(e)}")
             return self._get_fallback_quote()
 
-    
-
     def generate_image(self, quote, tradition, theme):
         """Generate an image using OpenAI DALL-E with enhanced prompts"""
         try:
@@ -224,44 +222,44 @@ class ContentGenerator:
             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)
-        img = Image.open(BytesIO(response.content))
-        width, height = img.size
-
-        # Create blank image with text
-        txt = Image.new('RGBA', (width, height), (255, 255, 255, 0))
-        d = ImageDraw.Draw(txt)
-
-        # Semi-transparent background for text
-        d.rectangle([0, height/2-100, width, height/2+100], 
-                    fill=(0, 0, 0, 127))
-
+    def create_image_with_quote(self, image_url, quote_text, author):
         try:
-            quote_font = ImageFont.truetype("DejaVuSans-Bold", 40)
-            author_font = ImageFont.truetype("DejaVuSans", 30)
-        except:
-            quote_font = ImageFont.load_default()
-            author_font = ImageFont.load_default()
+            response = requests.get(image_url)
+            img = Image.open(BytesIO(response.content))
+            width, height = img.size
 
-        # Add text
-        d.text((width/2, height/2), quote_text, 
-                font=quote_font, fill=(255,255,255,255), anchor="mm")
-        d.text((width/2, height/2+60), f"- {author}", 
-                font=author_font, fill=(255,255,255,255), anchor="mm")
+            # Create blank image with text
+            txt = Image.new('RGBA', (width, height), (255, 255, 255, 0))
+            d = ImageDraw.Draw(txt)
 
-        # Combine images
-        out = Image.alpha_composite(img.convert('RGBA'), txt)
+            # Semi-transparent background for text
+            d.rectangle([0, height / 2 - 100, width, height / 2 + 100], 
+                        fill=(0, 0, 0, 127))
 
-        # Convert to bytes
-        buffer = BytesIO()
-        out.convert('RGB').save(buffer, format='JPEG')
-        return buffer.getvalue()
+            try:
+                quote_font = ImageFont.truetype("DejaVuSans-Bold", 40)
+                author_font = ImageFont.truetype("DejaVuSans", 30)
+            except:
+                quote_font = ImageFont.load_default()
+                author_font = ImageFont.load_default()
+
+            # Add text
+            d.text((width / 2, height / 2), quote_text, 
+                   font=quote_font, fill=(255, 255, 255, 255), anchor="mm")
+            d.text((width / 2, height / 2 + 60), f"- {author}", 
+                   font=author_font, fill=(255, 255, 255, 255), anchor="mm")
+
+            # Combine images
+            out = Image.alpha_composite(img.convert('RGBA'), txt)
+
+            # Convert to bytes
+            buffer = BytesIO()
+            out.convert('RGB').save(buffer, format='JPEG')
+            return buffer.getvalue()
 
-    except Exception as e:
-        log_message(f"Error creating overlay: {str(e)}")
-        return None    
+        except Exception as e:
+            log_message(f"Error creating overlay: {str(e)}")
+            return None
 
 
 def display_debug_logs():
@@ -384,73 +382,73 @@ 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.title("Instagram Preview")
+        if all([st.session_state.generated_content, 
                 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"
-                )
+                st.session_state.generated_audio]):
 
-        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"
-                        )
+            col1, col2 = st.columns([2, 1])
 
-                    # Add caption and hashtags
-                    st.subheader("📝 Caption")
-                    hashtags = generate_hashtags(
-                        quote_data['tradition'],
-                        quote_data.get('theme', 'Wisdom')
+            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"
                     )
-                    caption = f"""✨ Wisdom from {quote_data['tradition']} ✨
+
+            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)
+                        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")
+                    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()
 
-    
-    with tab3:  # Make sure this is at the same indentation level as 'with tab2:'
-        display_debug_logs()   
 
 if __name__ == "__main__":
-    main()
+    main()