Update app.py

app.py

@@ -3,34 +3,39 @@ import os
 from datetime import datetime
 import json
 from crewai import Agent, Task, Crew
-from langchain.tools import Tool
+from langchain.tools import BaseTool
 from langchain_community.tools import DuckDuckGoSearchRun
 from langchain_openai import OpenAI
 from openai import OpenAI as OpenAIClient
+from typing import Optional, Type
+from pydantic import Field
 import logging
 import requests
 import replicate
+from io import BytesIO
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-def create_search_tool():
-    """Create a properly formatted search tool for CrewAI"""
-    search = DuckDuckGoSearchRun()
+class SearchTool(BaseTool):
+    name: str = "Search"
+    description: str = "Search the internet for information"
+    return_direct: bool = False
+    search: DuckDuckGoSearchRun = Field(default_factory=DuckDuckGoSearchRun)
     
-    return Tool(
-        name="Search",
-        func=search.run,
-        description="Search the internet for information and verification"
-    )
+    def _run(self, query: str) -> str:
+        return self.search.run(query)
+    
+    def _arun(self, query: str) -> str:
+        raise NotImplementedError("Async not implemented")
 
 class ContentGenerator:
     def __init__(self, openai_api_key, replicate_api_key):
         self.llm = OpenAI(api_key=openai_api_key, temperature=0.7)
         self.openai_client = OpenAIClient(api_key=openai_api_key)
         self.replicate_client = replicate.Client(api_token=replicate_api_key)
-        self.search_tool = create_search_tool()
+        self.search_tool = SearchTool()
 
     def generate_quote(self):
         """Generate a new spiritual quote using CrewAI"""
@@ -165,8 +170,130 @@ def main():
         st.error("Replicate API token not found in secrets!")
         st.stop()
 
-    # Rest of the main function remains the same...
-    # (Previous main function code continues here)
+    # Tabs
+    tab1, tab2 = st.tabs(["Generate Content", "Instagram Preview"])
+
+    with tab1:
+        st.title("🕉️ Spiritual Content Generator")
+        
+        if st.button("Generate New Content", key="generate_button"):
+            try:
+                with st.spinner("Generating content..."):
+                    generator = ContentGenerator(
+                        st.secrets["OPENAI_API_KEY"],
+                        st.secrets["REPLICATE_API_TOKEN"]
+                    )
+                    
+                    # Generate quote
+                    with st.status("Generating quote...") as status:
+                        quote_data = generator.generate_quote()
+                        st.session_state.generated_content = quote_data
+                        status.update(label="Quote generated!", state="complete")
+                    
+                    # Generate image
+                    with st.status("Generating image...") as status:
+                        image_url = generator.generate_image(
+                            quote_data["text"],
+                            quote_data["tradition"]
+                        )
+                        st.session_state.generated_image = image_url
+                        status.update(label="Image generated!", state="complete")
+                    
+                    # Generate audio
+                    with st.status("Generating audio...") as status:
+                        audio_url = generator.generate_audio(quote_data["tradition"])
+                        st.session_state.generated_audio = audio_url
+                        status.update(label="Audio generated!", state="complete")
+                    
+                    st.success("All content generated successfully!")
+
+            except Exception as e:
+                st.error(f"Error during generation: {str(e)}")
+
+        # Display generated content
+        if st.session_state.generated_content:
+            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;
+                '>
+                    <h2>{quote['text']}</h2>
+                    <p>- {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
+                st.subheader("📝 Caption")
+                hashtags = generate_hashtags(quote['tradition'])
+                caption = f"""✨ Wisdom from {quote['tradition']} ✨
+
+"{quote['text']}"
+- {quote['author']}
+
+{' '.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. Copy the generated caption
+                3. Best posting times: 8-10 AM or 6-8 PM
+                4. Engage with your audience about the quote's meaning
+                5. Consider creating a Reel with the image and background music
+                """)
+
+        else:
+            st.info("Generate content first to see the Instagram preview")
 
 if __name__ == "__main__":
     main()