Update app.py

app.py

@@ -1,59 +1,51 @@
 import streamlit as st
 from few_shot import FewShotPosts
 from post_generator import generate_post
-from preprocess import process_posts_for_persona
-
 
 # Options for length and language
 length_options = ["Short", "Medium", "Long"]
-language_options = ["English", "Hindi","Kannada"]
-
+language_options = ["English", "Hindi", "Kannada"]
 
 # Main app layout
 def main():
-    # Get persona name from UI or user input
-    persona = input("Enter Persona Name: ").strip()
-
-    try:
-        # Create instance of FewShotPosts with the selected persona
-        fs = FewShotPosts(persona)
-
-        # Print available tags for debugging
-        print(f"Available Tags for {persona}: {fs.get_tags()}")
-
-        # Example: Fetch posts based on user-defined criteria
-        posts = fs.get_filtered_posts(length="Short", language="English", tag="Economy")
-        print(posts)
-
-    except FileNotFoundError as e:
-        print(e)
-    st.subheader("LinkedIn Post Generator: Codebasics")
+    st.title("LinkedIn Post Generator: Codebasics")
 
-    # Create three columns for the dropdowns
-    col1, col2, col3 = st.columns(3)
+    # User Input for Persona
+    persona = st.text_input("Enter Persona Name:", "").strip()
 
-    fs = FewShotPosts(persona)
-    tags = fs.get_tags()
-    with col1:
-        # Dropdown for Topic (Tags)
-        selected_tag = st.selectbox("Topic", options=tags)
+    if persona:
+        try:
+            # Create an instance of FewShotPosts
+            fs = FewShotPosts(persona)
 
-    with col2:
-        # Dropdown for Length
-        selected_length = st.selectbox("Length", options=length_options)
+            # Fetch available tags dynamically
+            tags = fs.get_tags()
+            
+            if not tags:
+                st.warning("No tags found for this persona. Check the dataset.")
+                return
+            
+            # Create three columns for dropdowns
+            col1, col2, col3 = st.columns(3)
 
-    with col3:
-        # Dropdown for Language
-        selected_language = st.selectbox("Language", options=language_options)
+            with col1:
+                selected_tag = st.selectbox("Topic", options=tags)
 
+            with col2:
+                selected_length = st.selectbox("Length", options=length_options)
 
+            with col3:
+                selected_language = st.selectbox("Language", options=language_options)
 
-    # Generate Button
-    if st.button("Generate"):
-        post = generate_post(persona,selected_length, selected_language, selected_tag)
-        st.write(post)
+            # Generate Button
+            if st.button("Generate"):
+                post = generate_post(persona, selected_length, selected_language, selected_tag)
+                st.subheader("Generated Post:")
+                st.write(post)
 
+        except FileNotFoundError as e:
+            st.error(f"Error: {e}")
 
 # Run the app
 if __name__ == "__main__":
- main()
+    main()