Chatbot2 / app.py
tayy786's picture
Create app.py
9dbc035 verified
Raw
History Blame Contribute Delete
1.14 kB
import os
import gradio as gr
from groq import Groq
# ==============================
# Add your Groq API Key here
# (or set in HuggingFace Secrets)
# ==============================
GROQ_API_KEY = os.getenv("Access")
client = Groq(api_key=GROQ_API_KEY)
# ------------------------------
# Chat Function
# ------------------------------
def chatbot(message, history):
messages = []
# previous conversation
for user, bot in history:
messages.append({"role": "user", "content": user})
messages.append({"role": "assistant", "content": bot})
# new message
messages.append({"role": "user", "content": message})
# Groq API call
completion = client.chat.completions.create(
model="llama-3.1-8b-instant",
messages=messages,
temperature=0.7,
max_tokens=1024,
)
reply = completion.choices[0].message.content
return reply
# ------------------------------
# Gradio Interface
# ------------------------------
demo = gr.ChatInterface(
fn=chatbot,
title="🤖 Groq AI Chatbot",
description="Chatbot powered by Groq API + Gradio",
)
demo.launch()