Update app.py

app.py

@@ -49,44 +49,7 @@ st.title("💼 AI LinkedIn Post Generator")
 st.write("Create high-quality LinkedIn posts step-by-step 🚀")
 
 # ------------------ LOAD API KEY FROM ENV ------------------
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
-
-# ------------------ SESSION STATE ------------------
-if "step" not in st.session_state:
-    st.session_state.step = 1
-
-# ------------------ STEP 1 ------------------
-if st.session_state.step == 1:
-    topic = st.text_input("📌 Enter your post topic")
-    if st.button("Next ➡️"):
-        if topic:
-            st.session_state.topic = topic
-            st.session_state.step = 2
-
-# ------------------ STEP 2 ------------------
-elif st.session_state.step == 2:
-    tone = st.selectbox("🎯 Select tone", ["Professional", "Casual", "Inspirational", "Storytelling"])
-    if st.button("Next ➡️"):
-        st.session_state.tone = tone
-        st.session_state.step = 3
-
-# ------------------ STEP 3 ------------------
-elif st.session_state.step == 3:
-    audience = st.text_input("👥 Target audience")
-    if st.button("Next ➡️"):
-        if audience:
-            st.session_state.audience = audience
-            st.session_state.step = 4
-
-# ------------------ STEP 4 ------------------
-elif st.session_state.step == 4:
-    length = st.selectbox("📏 Post length", ["Short", "Medium", "Long"])
-    if st.button("Generate 🚀"):
-        st.session_state.length = length
-        st.session_state.step = 5
-
-# ------------------ GENERATE ------------------
-elif st.session_state.step == 5:
+   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.")
@@ -104,24 +67,23 @@ elif st.session_state.step == 5:
         """
 
         with st.spinner("Generating your post... ✨"):
-           
-       try:
-    response = client.chat.completions.create(
-        model="mixtral-8x7b-32768",
-        messages=[
-            {"role": "user", "content": user_input}
-        ],
-        max_tokens=500
-    )
-
-    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)}")
+            try:
+                response = client.chat.completions.create(
+                    model="mixtral-8x7b-32768",
+                    messages=[
+                        {"role": "user", "content": prompt}
+                    ],
+                    max_tokens=500
+                )
+
+                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)}")