Update app.py

app.py

@@ -3,20 +3,15 @@ from few_shot import FewShotPosts
 from post_generator import generate_post
 import json
 
-# Improved app with more functionality
+
 def main():
     st.set_page_config(page_title="LinkedIn Post Generator", layout="wide")
     st.title("🚀 LinkedIn Post Generator")
-    st.markdown(
-        """
-        **Create impactful LinkedIn posts effortlessly.**  
-        Customize your post with advanced options, analyze generated content, and save your creations.
-        """
-    )
+    st.markdown("**Create impactful LinkedIn posts effortlessly.** Customize your post with advanced options, analyze generated content, and save your creations.")
 
-    # Sidebar for advanced options
+    # Sidebar for settings
     st.sidebar.title("🔧 Customization Options")
-    st.sidebar.markdown("Use these settings to personalize your generated post.")
+    st.sidebar.markdown("Use these settings to personalize your post.")
 
     fs = FewShotPosts()
     tags = fs.get_tags()
@@ -26,71 +21,82 @@ def main():
     selected_tag = st.sidebar.selectbox("Topic (Tag):", options=tags)
     selected_length = st.sidebar.radio("Post Length:", ["Short", "Medium", "Long"])
     selected_language = st.sidebar.radio("Language:", ["English", "Hinglish"])
-    selected_tone = st.sidebar.selectbox(
-        "Tone/Style:", ["Motivational", "Professional", "Informal", "Neutral"]
-    )
-    custom_context = st.sidebar.text_area(
-        "Additional Context (Optional):",
-        placeholder="Provide extra details or a specific direction for your post...",
-    )
+    selected_tone = st.sidebar.selectbox("Tone/Style:", ["Motivational", "Professional", "Informal", "Neutral"])
+    custom_context = st.sidebar.text_area("Additional Context (Optional):", placeholder="Provide extra details...")
 
-    # History tracking
+    # Initialize session state
     if "generated_posts" not in st.session_state:
         st.session_state.generated_posts = []
 
-    # Main content
-    st.subheader("Generate Your LinkedIn Post")
-
-    if st.button("Generate Post"):
-        with st.spinner("Generating your post..."):
-            try:
-                # Generate post
-                post = generate_post(selected_length, selected_language, selected_tag)
-
-                # Add tone and context to the prompt dynamically
-                if selected_tone:
-                    post = f"**Tone**: {selected_tone}\n\n{post}"
-                if custom_context:
-                    post += f"\n\n**Context Added**: {custom_context}"
-
-                # Save to session history
-                st.session_state.generated_posts.append(post)
-
-                st.success("Post generated successfully!")
-                st.markdown("### Your LinkedIn Post:")
-                st.write(post)
-
-                # Post analysis
-                st.markdown("### Post Analysis:")
-                post_analysis(post)
-
-            except Exception as e:
-                st.error(f"An error occurred: {e}")
-
-    # Display session history
-    if st.session_state.generated_posts:
-        st.markdown("### Generated Posts History:")
-        for i, history_post in enumerate(st.session_state.generated_posts):
-            st.markdown(f"**Post {i + 1}:**")
-            st.write(history_post)
-
-    # Save and Download
-    st.markdown("---")
-    st.subheader("Save or Share Your Post")
-    if st.session_state.generated_posts:
-        if st.button("Download All Posts"):
-            download_posts(st.session_state.generated_posts)
-        if st.button("Clear History"):
-            st.session_state.generated_posts = []
-            st.info("History cleared!")
-
-    # Footer
-    st.markdown(
-        """
-        ---
-        """
-    )
-
+    # Tabs for different modes
+    tabs = st.tabs(["Generate Post", "Edit Post", "Analyze Post", "History", "Save/Share"])
+
+    # Tab 1: Generate Post
+    with tabs[0]:
+        st.subheader("Create a Post")
+        if st.button("Generate Post"):
+            with st.spinner("Generating your post..."):
+                try:
+                    post = generate_post(selected_length, selected_language, selected_tag, selected_tone)
+
+                    # Add tone and context dynamically
+                    if selected_tone:
+                        post = f"**Tone**: {selected_tone}\n\n{post}"
+                    if custom_context:
+                        post += f"\n\n**Additional Context**: {custom_context}"
+
+                    # Save to session history
+                    st.session_state.generated_posts.append(post)
+                    st.success("Post generated successfully!")
+                    st.markdown("### Your LinkedIn Post:")
+                    st.markdown(f"<div style='background-color: #f9f9f9; padding: 10px; border-radius: 5px;'>{post}</div>", unsafe_allow_html=True)
+
+                except Exception as e:
+                    st.error(f"An error occurred: {e}")
+
+    # Tab 2: Edit Post
+    with tabs[1]:
+        st.subheader("Edit Your Post")
+        if st.session_state.generated_posts:
+            editable_post = st.text_area("Edit your post:", st.session_state.generated_posts[-1])
+            if st.button("Save Changes"):
+                st.session_state.generated_posts[-1] = editable_post
+                st.success("Post updated!")
+        else:
+            st.info("Generate a post to edit.")
+
+    # Tab 3: Analyze Post
+    with tabs[2]:
+        st.subheader("Analyze Your Post")
+        if st.session_state.generated_posts:
+            post_to_analyze = st.session_state.generated_posts[-1]
+            st.markdown("### Post for Analysis:")
+            st.write(post_to_analyze)
+            post_analysis(post_to_analyze)
+        else:
+            st.info("Generate a post to analyze.")
+
+    # Tab 4: Post History
+    with tabs[3]:
+        st.subheader("Generated Posts History")
+        if st.session_state.generated_posts:
+            for i, post in enumerate(reversed(st.session_state.generated_posts), start=1):
+                with st.expander(f"Post {i}"):
+                    st.write(post)
+        else:
+            st.info("No posts generated yet.")
+
+    # Tab 5: Save or Share
+    with tabs[4]:
+        st.subheader("Save or Share Your Post")
+        if st.session_state.generated_posts:
+            if st.button("Download All Posts"):
+                download_posts(st.session_state.generated_posts)
+            if st.button("Clear History"):
+                st.session_state.generated_posts = []
+                st.info("History cleared!")
+        else:
+            st.info("No posts to save or share.")
 
 def post_analysis(post):
     """Perform basic analysis of the generated post."""
@@ -102,8 +108,8 @@ def post_analysis(post):
         "Informal": ["hey", "cool", "fun", "awesome"],
     }
 
-    st.write(f"**Word Count:** {word_count}")
-    st.write(f"**Character Count:** {char_count}")
+    st.metric("Word Count", word_count)
+    st.metric("Character Count", char_count)
 
     # Detect tone based on keywords (simple heuristic)
     detected_tone = "Neutral"
@@ -112,7 +118,7 @@ def post_analysis(post):
             detected_tone = tone
             break
 
-    st.write(f"**Detected Tone:** {detected_tone}")
+    st.metric("Detected Tone", detected_tone)
 
 
 def download_posts(posts):
@@ -126,6 +132,5 @@ def download_posts(posts):
     )
 
 
-# Run the app
 if __name__ == "__main__":
     main()