| import gradio as gr |
| from groq import Groq |
| import os |
|
|
| |
| api_key = os.environ.get("GROQ_API_KEY") |
|
|
| |
| if api_key: |
| client = Groq(api_key=api_key) |
| else: |
| print("Error: GROQ_API_KEY not found in environment variables.") |
|
|
| |
| def generate_linkedin_post(topic, tone): |
| if not api_key: |
| return "Error: API Key is missing. Please add GROQ_API_KEY to your Space Secrets." |
| |
| system_prompt = f""" |
| You are an expert social media manager and copywriter. |
| Write an engaging, high-quality LinkedIn post about the following topic. |
| The tone of the post should be: {tone}. |
| Include engaging formatting, line breaks, and 2-3 relevant hashtags at the end. |
| Do not include any pleasantries like "Here is your post:", just output the post itself. |
| """ |
|
|
| try: |
| response = client.chat.completions.create( |
| model="llama-3.1-8b-instant", |
| messages=[ |
| {"role": "system", "content": system_prompt}, |
| {"role": "user", "content": f"Topic: {topic}"} |
| ], |
| temperature=0.7, |
| max_tokens=500, |
| ) |
| return response.choices[0].message.content |
| |
| except Exception as e: |
| return f"An error occurred: {str(e)}" |
|
|
| |
| demo = gr.Interface( |
| fn=generate_linkedin_post, |
| inputs=[ |
| gr.Textbox( |
| label="What is the topic of your post?", |
| placeholder="e.g., Why AI will not replace software engineers, but make them faster.", |
| lines=2 |
| ), |
| gr.Dropdown( |
| label="Select the Tone", |
| choices=["Professional", "Casual & Friendly", "Inspirational", "Controversial/Hot Take", "Educational"], |
| value="Professional" |
| ) |
| ], |
| outputs=gr.Textbox(label="Your Generated LinkedIn Post", lines=10), |
| title="🚀 AI LinkedIn Post Generator", |
| description="Enter a topic and choose a tone to generate a ready-to-publish LinkedIn post instantly using Groq's Llama 3.1 model.", |
| theme=gr.themes.Soft() |
| ) |
|
|
| |
| if __name__ == "__main__": |
| demo.launch() |
