Ai-chatbot / app.py
Sapna36's picture
Update app.py
7e73039 verified
import requests
import gradio as gr
import os
# Get Groq API Key from Hugging Face Secrets (best practice)
GROQ_API_KEY = os.getenv("APIKEY")
# If you are testing locally, you can uncomment this line and hardcode your key (NOT recommended on Spaces)
# GROQ_API_KEY = "your-groq-api-key-here"
# Define the chatbot function
def chat_with_groq(message):
url = "https://api.groq.com/openai/v1/chat/completions"
headers = {
"Authorization": f"Bearer {GROQ_API_KEY}",
"Content-Type": "application/json"
}
payload = {
"model": "llama3-70b-8192",
"messages": [
{"role": "system", "content": "You are a helpful and friendly chatbot."},
{"role": "user", "content": message}
]
}
response = requests.post(url, headers=headers, json=payload)
if response.status_code == 200:
result = response.json()
return result['choices'][0]['message']['content']
else:
return f"Error: {response.status_code}, {response.text}"
# Create Gradio interface
iface = gr.Interface(
fn=chat_with_groq,
inputs="text",
outputs="text",
title="Groq Chatbot",
description="A simple chatbot powered by Groq API. Type your message and get a reply!",
theme="default"
)
# Launch interface
if __name__ == "__main__":
iface.launch()