Update app.py

app.py

@@ -1,64 +1,142 @@
-import streamlit as st
-from agents import web_summarizer_company, web_summarizer_person, company_linkedin_agent, person_linkedin_agent
-from tools import scrape_website, post_on_linkedin, replace_i_with_you
-
-def main():
-    st.title('LinkedIn Post Generator')
-
-    with st.form(key='linkedin_form'):
-        topic = st.text_input('Topic')
-        url = st.text_input('Website URL')
-        mood = st.text_input('Mood')
-        post_company = st.checkbox('For Company')
-
-        generate_button = st.form_submit_button(label='Generate Post')
-
-        if generate_button:
-            if url:
-                
-                scraped_data = scrape_website(url)
-
-                if post_company:
-                    summary = web_summarizer_company(scraped_data)
-                else:
-                    summary = web_summarizer_person(scraped_data)
-                summary = replace_i_with_you(summary.content)
-                print(summary)
-
-                if post_company:
-                    post_content = company_linkedin_agent(topic, summary, mood)
-                else:
-                    post_content = person_linkedin_agent(topic, summary, mood)
-                
-                print(post_content.content)
-                post_content = post_content.content
-                
-                if post_content.endswith(':'):
-                    post_content = post_content[:-5]
-
-                st.session_state.post_content = post_content
-                st.session_state.post_generated = True
-
-                st.markdown(f"**Generated Post Content:**\n\n{st.session_state.post_content}")
-
-                # Text area for reviewing generated content
-                # st.text_area('Post Content (for review)', st.session_state.post_content, height=200)
-
-            else:
-                st.error('Please provide a URL.')
-
-    # Button to post on LinkedIn outside the form
-    if st.session_state.get('post_generated'):
-        post_linkedin = st.checkbox('Post on LinkedIn')
-        
-        if post_linkedin:
-            token = st.text_input('LinkedIn Token', type='password')
-            post_button = st.button('Post')
-            if post_button:
-                image_path = None
-                post_on_linkedin(token, 'linkedin post', st.session_state.post_content, image_path)
-                st.success('Post has been successfully published on LinkedIn!')
-                st.session_state.post_generated = False  # Reset state after posting
-
-if __name__ == "__main__":
-    main()
+import streamlit as st
+from agents import web_summarizer_company, web_summarizer_person, company_linkedin_agent, person_linkedin_agent, image_prompt_agent
+from tools import scrape_website, post_on_linkedin, replace_i_with_you, generate_image
+
+def main():
+    st.title('LinkedIn Post Generator')
+
+    with st.form(key='linkedin_form'):
+        topic = st.text_input('Topic')
+        url = st.text_input('Website URL')
+        mood = st.text_input('Mood')
+        post_company = st.checkbox('For Company')
+        ai_image = st.checkbox('Add AI Generated Image')
+
+        generate_button = st.form_submit_button(label='Generate Post')
+
+        if generate_button:
+            if url:
+                
+                scraped_data = scrape_website(url)
+
+                if post_company:
+                    summary = web_summarizer_company(scraped_data)
+                else:
+                    summary = web_summarizer_person(scraped_data)
+                summary = replace_i_with_you(summary.content)
+                print(summary)
+
+                if post_company:
+                    post_content = company_linkedin_agent(topic, summary, mood)
+                else:
+                    post_content = person_linkedin_agent(topic, summary, mood)
+                
+                print(post_content.content)
+                post_content = post_content.content
+                
+                if post_content.endswith(':'):
+                    post_content = post_content[:-5]
+
+                image_path = None
+                if ai_image:
+                    prompt = image_prompt_agent(post_content)
+                    image_path = generate_image(prompt)
+                    print(image_path)
+
+                st.session_state.post_content = post_content
+                st.session_state.image_path = image_path
+                st.session_state.post_generated = True
+
+                st.markdown(f"**Generated Post Content:**\n\n{st.session_state.post_content}")
+                if ai_image:
+                    st.image(image_path)
+
+                # Text area for reviewing generated content
+                # st.text_area('Post Content (for review)', st.session_state.post_content, height=200)
+
+            else:
+                st.error('Please provide a URL.')
+
+    # Button to post on LinkedIn outside the form
+    if st.session_state.get('post_generated'):
+        post_linkedin = st.checkbox('Post on LinkedIn')
+        
+        if post_linkedin:
+            token = st.text_input('LinkedIn Token', type='password')
+            post_button = st.button('Post')
+            if post_button:
+                image_path = st.session_state.image_path
+                post_on_linkedin(token, 'linkedin post', st.session_state.post_content, image_path)
+                st.success('Post has been successfully published on LinkedIn!')
+                st.session_state.post_generated = False  # Reset state after posting
+
+if __name__ == "__main__":
+    main()
+
+
+
+
+# import streamlit as st
+# from agents import web_summarizer_company, web_summarizer_person, company_linkedin_agent, person_linkedin_agent
+# from tools import scrape_website, post_on_linkedin, replace_i_with_you
+
+# def main():
+#     st.title('LinkedIn Post Generator')
+
+#     with st.form(key='linkedin_form'):
+#         topic = st.text_input('Topic')
+#         url = st.text_input('Website URL')
+#         mood = st.text_input('Mood')
+#         post_company = st.checkbox('For Company')
+
+#         generate_button = st.form_submit_button(label='Generate Post')
+
+#         if generate_button:
+#             if url:
+                
+#                 scraped_data = scrape_website(url)
+
+#                 if post_company:
+#                     summary = web_summarizer_company(scraped_data)
+#                 else:
+#                     summary = web_summarizer_person(scraped_data)
+#                 summary = replace_i_with_you(summary.content)
+#                 print(summary)
+
+#                 if post_company:
+#                     post_content = company_linkedin_agent(topic, summary, mood)
+#                 else:
+#                     post_content = person_linkedin_agent(topic, summary, mood)
+                
+#                 print(post_content.content)
+#                 post_content = post_content.content
+                
+#                 if post_content.endswith(':'):
+#                     post_content = post_content[:-5]
+
+#                 st.session_state.post_content = post_content
+#                 st.session_state.post_generated = True
+
+#                 st.markdown(f"**Generated Post Content:**\n\n{st.session_state.post_content}")
+
+#                 # Text area for reviewing generated content
+#                 # st.text_area('Post Content (for review)', st.session_state.post_content, height=200)
+
+#             else:
+#                 st.error('Please provide a URL.')
+
+#     # Button to post on LinkedIn outside the form
+#     if st.session_state.get('post_generated'):
+#         post_linkedin = st.checkbox('Post on LinkedIn')
+        
+#         if post_linkedin:
+#             token = st.text_input('LinkedIn Token', type='password')
+#             post_button = st.button('Post')
+#             if post_button:
+#                 image_path = None
+#                 post_on_linkedin(token, 'linkedin post', st.session_state.post_content, image_path)
+#                 st.success('Post has been successfully published on LinkedIn!')
+#                 st.session_state.post_generated = False  # Reset state after posting
+
+# if __name__ == "__main__":
+#     main()