Update app.py

app.py

@@ -1,55 +1,7 @@
-import streamlit as st
-from groq import Groq
-import os
+# ------------------ GENERATE ------------------
+elif st.session_state.step == 5:
 
-# ------------------ PAGE CONFIG ------------------
-st.set_page_config(
-    page_title="AI LinkedIn Post Generator",
-    page_icon="💼",
-    layout="centered"
-)
-
-# ------------------ CUSTOM CSS ------------------
-st.markdown("""
-    <style>
-    body {
-        background: linear-gradient(135deg, #4facfe, #00f2fe);
-    }
-    .main {
-        background-color: white;
-        padding: 25px;
-        border-radius: 15px;
-    }
-    h1, h2, h3 {
-        color: #1f2937;
-        font-weight: 800;
-    }
-    p, label {
-        color: #374151;
-        font-weight: 600;
-        font-size: 16px;
-    }
-    .stButton>button {
-        background-color: #2563eb;
-        color: white;
-        font-weight: bold;
-        border-radius: 8px;
-        padding: 10px;
-    }
-    .stTextInput>div>div>input, .stTextArea textarea {
-        background-color: #f9fafb;
-        color: black;
-        font-weight: 600;
-    }
-    </style>
-""", unsafe_allow_html=True)
-
-# ------------------ TITLE ------------------
-st.title("💼 AI LinkedIn Post Generator")
-st.write("Create high-quality LinkedIn posts step-by-step 🚀")
-
-# ------------------ LOAD API KEY FROM ENV ------------------
-   api_key = os.getenv("GROQ_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.")
@@ -67,7 +19,7 @@ st.write("Create high-quality LinkedIn posts step-by-step 🚀")
         """
 
         with st.spinner("Generating your post... ✨"):
-            
+            try:
                 response = client.chat.completions.create(
                     model="mixtral-8x7b-32768",
                     messages=[