Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from langchain_groq import ChatGroq | |
| from langchain.schema import SystemMessage, HumanMessage | |
| import os | |
| # Securely fetch API Key from environment variable | |
| GROQ_API_KEY = os.getenv("GROQ_API_KEY") | |
| if not GROQ_API_KEY: | |
| raise ValueError("โ ๏ธ GROQ_API_KEY is missing! Set it in your environment variables.") | |
| # Initialize the Groq Model | |
| llm = ChatGroq( | |
| temperature=0, | |
| groq_api_key=GROQ_API_KEY, | |
| model_name="llama3-8b-8192" | |
| ) | |
| def blog_generator(Blog_Topic, Blog_Keyword, Blog_Tone, Blog_Numberofwords, Blog_Target_audience): | |
| """Generate a blog using Groq AI""" | |
| system_prompt = f""" | |
| You are an expert blog writer. Follow these instructions: | |
| 1. Write a blog on: {Blog_Topic}. | |
| 2. Include keywords: {Blog_Keyword}. | |
| 3. Use tone: {Blog_Tone}. | |
| 4. Limit to {Blog_Numberofwords} words. | |
| 5. Target audience: {Blog_Target_audience}. | |
| """ | |
| user_prompt = f"Write a blog on {Blog_Topic}, with keywords {Blog_Keyword}, tone {Blog_Tone}, {Blog_Numberofwords} words, for audience {Blog_Target_audience}." | |
| response = llm.invoke([SystemMessage(content=system_prompt), HumanMessage(content=user_prompt)]) | |
| return response.content | |
| # Gradio Interface | |
| with gr.Blocks(theme=gr.themes.Soft()) as demo: | |
| gr.Markdown("# ๐ AI Blog Generator") | |
| with gr.Row(): | |
| topic = gr.Textbox(label="Blog Topic", placeholder="Enter the topic of your blog...") | |
| keyword = gr.Textbox(label="Keywords (comma-separated)", placeholder="e.g., AI, Blogging, Technology") | |
| with gr.Row(): | |
| tone = gr.Dropdown(["Formal", "Casual", "Professional"], label="Blog Tone", value="Professional") | |
| words = gr.Slider(100, 1000, step=50, label="Number of Words", value=500) | |
| audience = gr.Dropdown(["General Public", "Tech Enthusiasts", "Business Professionals", "Students"], label="Target Audience", value="General Public") | |
| generate_btn = gr.Button("Generate Blog", variant="primary") | |
| output = gr.Textbox(label="Generated Blog", lines=10, placeholder="Your generated blog will appear here...") | |
| generate_btn.click(blog_generator, inputs=[topic, keyword, tone, words, audience], outputs=output) | |
| demo.launch() |