Update app.py

app.py

@@ -1,4 +1,3 @@
-# ------------------ GENERATE ------------------
 import streamlit as st
 from groq import Groq
 import os
@@ -47,43 +46,87 @@ st.markdown("""
 
 # ------------------ TITLE ------------------
 st.title("💼 AI LinkedIn Post Generator")
-st.write("Create high-quality LinkedIn posts step-by-step 🚀")
+st.write("Let's craft an engaging LinkedIn post step-by-step 🚀")
 
-    api_key = os.getenv("GROQ_API_KEY")
+# Initialize session state variables
+if 'step' not in st.session_state:
+    st.session_state['step'] = 1
+if 'topic' not in st.session_state:
+    st.session_state['topic'] = ''
+if 'tone' not in st.session_state:
+    st.session_state['tone'] = ''
+if 'audience' not in st.session_state:
+    st.session_state['audience'] = ''
+if 'length' not in st.session_state:
+    st.session_state['length'] = 150
+if 'post' not in st.session_state:
+    st.session_state['post'] = ''
 
-    if not api_key:
-        st.error("⚠️ API key not found! Add GROQ_API_KEY in Hugging Face Secrets.")
-    else:
-        client = Groq(api_key=api_key)
+# Step 1: Input Topic
+if st.session_state['step'] == 1:
+    st.subheader("Step 1: What's the topic?")
+    topic_input = st.text_input("Enter the topic for your LinkedIn post", value=st.session_state['topic'])
+    if st.button("Next"):
+        st.session_state['topic'] = topic_input
+        st.session_state['step'] = 2
 
-        prompt = f"""
-        Write a LinkedIn post with:
-        Topic: {st.session_state.topic}
-        Tone: {st.session_state.tone}
-        Audience: {st.session_state.audience}
-        Length: {st.session_state.length}
+# Step 2: Input Tone
+elif st.session_state['step'] == 2:
+    st.subheader("Step 2: What's the tone?")
+    tone_input = st.text_input("E.g., professional, friendly, inspiring", value=st.session_state['tone'])
+    if st.button("Next"):
+        st.session_state['tone'] = tone_input
+        st.session_state['step'] = 3
+    if st.button("Back"):
+        st.session_state['step'] = 1
 
-        Make it engaging, clear, and professional. Add hook and call-to-action.
-        """
+# Step 3: Input Audience
+elif st.session_state['step'] == 3:
+    st.subheader("Step 3: Who's the audience?")
+    audience_input = st.text_input("E.g., marketers, developers", value=st.session_state['audience'])
+    if st.button("Next"):
+        st.session_state['audience'] = audience_input
+        st.session_state['step'] = 4
+    if st.button("Back"):
+        st.session_state['step'] = 2
 
-        with st.spinner("Generating your post... ✨"):
-            try:
-                response = client.chat.completions.create(
-                    model="mixtral-8x7b-32768",
-                    messages=[
-                        {"role": "user", "content": prompt}
-                    ],
-                    max_tokens=500
-                )
+# Step 4: Input Length
+elif st.session_state['step'] == 4:
+    st.subheader("Step 4: How long should the post be?")
+    length_input = st.slider("Number of words", min_value=50, max_value=500, value=st.session_state['length'])
+    if st.button("Generate Post"):
+        st.session_state['length'] = length_input
+        # Proceed to generate the post
+        # Fetch API key
+        api_key = os.getenv("GROQ_API_KEY")
+        if not api_key:
+            st.error("⚠️ API key not found! Add GROQ_API_KEY in Hugging Face Secrets.")
+        else:
+            client = Groq(api_key=api_key)
+            prompt = f"""
+            Write a LinkedIn post with:
+            Topic: {st.session_state['topic']}
+            Tone: {st.session_state['tone']}
+            Audience: {st.session_state['audience']}
+            Length: {st.session_state['length']} words
+            Make it engaging, clear, and professional. Add hook and call-to-action.
+            """
 
-                post = response.choices[0].message.content
-
-                st.success("✅ Your LinkedIn Post is Ready!")
-                st.markdown("### 📢 Generated Post")
-                st.write(post)
-
-                if st.button("🔄 Create Another"):
-                    st.session_state.step = 1
-
-            except Exception as e:
-                st.error(f"Error: {str(e)}")
+            with st.spinner("Generating your post... ✨"):
+                try:
+                    response = client.chat.completions.create(
+                        model="mixtral-8x7b-32768",
+                        messages=[{"role": "user", "content": prompt}],
+                        max_tokens=600
+                    )
+                    post = response.choices[0].message.content
+                    st.session_state['post'] = post
+                    st.success("✅ Your LinkedIn Post is Ready!")
+                    st.markdown("### 📢 Generated Post")
+                    st.write(post)
+                    if st.button("Create Another Post"):
+                        st.session_state['step'] = 1
+                except Exception as e:
+                    st.error(f"Error: {str(e)}")
+    if st.button("Back"):
+        st.session_state['step'] = 3