Update app.py

app.py

@@ -105,20 +105,27 @@ elif st.session_state.step == 5:
 
         with st.spinner("Generating your post... ✨"):
            
-              response = client.chat.completions.create(
+         response = client.chat.completions.create(
         model="mixtral-8x7b-32768",
         messages=[
             {"role": "user", "content": user_input}
         ],
         max_tokens=500
     )
-    
-    st.write(response.choices[0].message.content)
 
-        st.success("✅ Your LinkedIn Post is Ready!")
+    # ✅ Extract response properly
+    post = response.choices[0].message.content
 
-        st.markdown("### 📢 Generated Post")
-        st.write(post)
+    # ✅ Show success
+    st.success("✅ Your LinkedIn Post is Ready!")
 
-        if st.button("🔄 Create Another"):
-            st.session_state.step = 1
+    # ✅ Display output
+    st.markdown("### 📢 Generated Post")
+    st.write(post)
+
+    # ✅ Reset button
+    if st.button("🔄 Create Another"):
+        st.session_state.step = 1
+
+except Exception as e:
+    st.error(f"Error: {str(e)}")