Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 
 # ------------------ PAGE CONFIG ------------------
 st.set_page_config(
-    page_title="🚀 AI LinkedIn Post Generator",
+    page_title="AI LinkedIn Post Generator",
     page_icon="💼",
     layout="centered"
 )
@@ -13,38 +13,43 @@ st.set_page_config(
 st.markdown("""
     <style>
     body {
-        background: linear-gradient(135deg, #1f4037, #99f2c8);
+        background: linear-gradient(135deg, #4facfe, #00f2fe);
     }
     .main {
-        background-color: rgba(0,0,0,0.6);
-        padding: 20px;
+        background-color: white;
+        padding: 25px;
         border-radius: 15px;
     }
-    h1, h2, h3, h4, h5, h6, p, label {
-        color: white !important;
-        font-weight: bold !important;
+    h1, h2, h3 {
+        color: #1f2937;
+        font-weight: 800;
+    }
+    p, label {
+        color: #374151;
+        font-weight: 600;
+        font-size: 16px;
     }
     .stButton>button {
-        background-color: #ff4b2b;
+        background-color: #2563eb;
         color: white;
         font-weight: bold;
-        border-radius: 10px;
+        border-radius: 8px;
         padding: 10px;
     }
     .stTextInput>div>div>input, .stTextArea textarea {
-        background-color: #222;
-        color: white;
-        font-weight: bold;
+        background-color: #f9fafb;
+        color: black;
+        font-weight: 600;
     }
     </style>
 """, unsafe_allow_html=True)
 
 # ------------------ TITLE ------------------
 st.title("💼 AI LinkedIn Post Generator")
-st.write("✨ Create powerful, engaging LinkedIn posts step by step!")
+st.write("Create high-quality LinkedIn posts step-by-step 🚀")
 
-# ------------------ API KEY ------------------
-api_key = st.text_input("GROQ_API_KEY")
+# ------------------ LOAD API KEY FROM ENV ------------------
+api_key = os.getenv("GROQ_API_KEY")
 
 # ------------------ SESSION STATE ------------------
 if "step" not in st.session_state:
@@ -69,8 +74,9 @@ elif st.session_state.step == 2:
 elif st.session_state.step == 3:
     audience = st.text_input("👥 Target audience")
     if st.button("Next ➡️"):
-        st.session_state.audience = audience
-        st.session_state.step = 4
+        if audience:
+            st.session_state.audience = audience
+            st.session_state.step = 4
 
 # ------------------ STEP 4 ------------------
 elif st.session_state.step == 4:
@@ -81,34 +87,34 @@ elif st.session_state.step == 4:
 
 # ------------------ GENERATE ------------------
 elif st.session_state.step == 5:
+
     if not api_key:
-        st.error("⚠️ Please enter your GROQ 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 the following details:
+        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}
 
-        Make it engaging, include hooks, emojis, and call-to-action.
+        Make it engaging, clear, and professional. Add hook and call-to-action.
         """
 
-        with st.spinner("✨ Generating your post..."):
-            
+        with st.spinner("Generating your post... ✨"):
             response = client.chat.completions.create(
-                model="openai/gpt-oss-120b",
-                messages=[{"role": "user", "content": prompt}],
-                temperature=0.7,
-                max_tokens=500
+                model="llama3-70b-8192",
+                messages=[{"role": "user", "content": prompt}]
             )
 
             post = response.choices[0].message.content
 
         st.success("✅ Your LinkedIn Post is Ready!")
-        st.markdown(f"### 📢 Generated Post:\n\n*{post}*")
+
+        st.markdown("### 📢 Generated Post")
+        st.write(post)
 
         if st.button("🔄 Create Another"):
             st.session_state.step = 1