Update app.py

app.py

@@ -1,4 +1,3 @@
-# to-do: add credits - https://www.youtube.com/watch?v=qZ_J-Xg0QM4
 import streamlit as st
 from few_shot import FewShotPosts
 from post_generator import generate_post
@@ -40,12 +39,14 @@ def main():
                 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}"
+                    # Add additional context if provided
                     if custom_context:
                         post += f"\n\n**Additional Context**: {custom_context}"
 
+                    # Append hashtags at the end
+                    hashtags = f"\n\n#LinkedIn #{selected_tag} #{selected_tone}".replace(" ", "")
+                    post += hashtags
+
                     # Save to session history
                     st.session_state.generated_posts.append(post)
                     st.success("Post generated successfully!")
@@ -99,8 +100,9 @@ def main():
         else:
             st.info("No posts to save or share.")
 
+    # Add reference in the sidebar
     st.sidebar.markdown("---")
-    st.sidebar.markdown("Reference:")
+    st.sidebar.markdown("### Reference:")
     st.sidebar.markdown("[Linkedin Post Generator - Codebasics](https://www.youtube.com/watch?v=qZ_J-Xg0QM4)")
 
 
@@ -137,6 +139,5 @@ def download_posts(posts):
         mime="application/json",
     )
 
-
 if __name__ == "__main__":
     main()