Update app.py

app.py

@@ -2,102 +2,85 @@ import streamlit as st
 import requests
 import os
 
-# ----------------------------
+# ------------------------
 # Page Config
-# ----------------------------
+# ------------------------
 st.set_page_config(
     page_title="LinkedIn Post Generator",
-    page_icon="🚀",
-    layout="centered"
+    page_icon="🚀"
 )
 
-st.title("🚀 AI LinkedIn Post Generator")
-st.write("Generate viral-style LinkedIn posts using AI")
+st.title("🚀 AI LinkedIn Post Generator (Groq Powered)")
+st.write("Generate high-engagement LinkedIn posts instantly.")
 
-# ----------------------------
-# Get HF API Key from Secrets
-# ----------------------------
-HF_API_KEY = os.getenv("HF_API_KEY")
+# ------------------------
+# Load Groq API Key from Secrets
+# ------------------------
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
-if not HF_API_KEY:
-    st.error("⚠️ Please add your HF_API_KEY in Hugging Face Space Secrets.")
+if not GROQ_API_KEY:
+    st.error("Please add GROQ_API_KEY in Hugging Face Space Secrets.")
     st.stop()
 
-API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
+API_URL = "https://api.groq.com/openai/v1/chat/completions"
 
 headers = {
-    "Authorization": f"Bearer {HF_API_KEY}"
+    "Authorization": f"Bearer {GROQ_API_KEY}",
+    "Content-Type": "application/json"
 }
 
-# ----------------------------
+# ------------------------
 # User Inputs
-# ----------------------------
-topic = st.text_input("Enter your LinkedIn post topic")
+# ------------------------
+topic = st.text_input("Enter Topic")
 
 tone = st.selectbox(
     "Select Tone",
-    [
-        "Professional",
-        "Motivational",
-        "Storytelling",
-        "Bold & Controversial",
-        "Friendly"
-    ]
+    ["Professional", "Motivational", "Storytelling", "Bold", "Friendly"]
 )
 
 generate = st.button("Generate Post")
 
-# ----------------------------
-# Prompt Template
-# ----------------------------
-def build_prompt(topic, tone):
-    return f"""
-You are a LinkedIn content expert.
+# ------------------------
+# Generate Content
+# ------------------------
+if generate:
+    if not topic:
+        st.warning("Please enter a topic.")
+    else:
+        with st.spinner("Generating your post..."):
 
-Write a viral LinkedIn post about: "{topic}"
+            prompt = f"""
+Write a high-engagement LinkedIn post about "{topic}".
 
 Tone: {tone}
 
 Requirements:
 - Strong hook in first line
-- Short punchy paragraphs
-- Use spacing for readability
-- Add bullet points if relevant
-- End with engagement question
-- Add 3-5 relevant hashtags
-
-Make it authentic and high engagement.
+- Short readable paragraphs
+- Add spacing for clarity
+- End with an engaging question
+- Include 3 relevant hashtags
 """
 
-
-# ----------------------------
-# Generate Response
-# ----------------------------
-if generate:
-    if not topic:
-        st.warning("Please enter a topic.")
-    else:
-        with st.spinner("Generating your viral post..."):
-
-            prompt = build_prompt(topic, tone)
-
             payload = {
-                "inputs": prompt,
-                "parameters": {
-                    "max_new_tokens": 400,
-                    "temperature": 0.7,
-                    "return_full_text": False
-                }
+                "model": "llama3-8b-8192",
+                "messages": [
+                    {"role": "system", "content": "You are a LinkedIn content expert."},
+                    {"role": "user", "content": prompt}
+                ],
+                "temperature": 0.7,
+                "max_tokens": 500
             }
 
             response = requests.post(API_URL, headers=headers, json=payload)
 
             if response.status_code == 200:
                 result = response.json()
-                generated_text = result[0]["generated_text"]
-
-                st.success("✅ Your LinkedIn Post is Ready!")
-                st.text_area("Generated Post", generated_text, height=400)
+                content = result["choices"][0]["message"]["content"]
 
+                st.success("✅ Post Generated!")
+                st.text_area("Your LinkedIn Post", content, height=400)
             else:
-                st.error("Error generating content. Try again later.")
+                st.error("Error generating content.")
+                st.write(response.text)