chatbot / app.py
Maryam587's picture
Update app.py
c57f7a3 verified
import gradio as gr
from groq import Groq
import os
# 🔐 Set your Groq API key here OR use Hugging Face Secrets
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
# Initialize client
client = Groq(api_key=GROQ_API_KEY)
# Chat function
def chatbot(message, history):
messages = []
# Add previous conversation
for user, bot in history:
messages.append({"role": "user", "content": user})
messages.append({"role": "assistant", "content": bot})
# Add current message
messages.append({"role": "user", "content": message})
# Call Groq API
response = client.chat.completions.create(
model="llama3-70b-8192", # fast + powerful
messages=messages,
temperature=0.7,
max_tokens=512,
)
reply = response.choices[0].message.content
return reply
# Gradio UI
demo = gr.ChatInterface(
fn=chatbot,
title="💬 AI Chatbot (Groq + Hugging Face)",
description="Ask anything! Powered by Groq API ⚡",
)
# Run app
if __name__ == "__main__":
demo.launch()