AI_Chatbot / app.py
saherPervaiz's picture
Update app.py
01a92dc verified
raw
history blame contribute delete
874 Bytes
import gradio as gr
import os
from groq import Groq
# Initialize Groq client
client = Groq(api_key="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")
# Chat function
def chat_with_groq(user_input):
try:
# Create a chat completion request
chat_completion = client.chat.completions.create(
messages=[
{"role": "user", "content": user_input},
],
model="llama-3.3-70b-versatile",
)
# Return the AI's response
return chat_completion.choices[0].message.content
except Exception as e:
return f"Error: {e}"
# Gradio interface
iface = gr.Interface(
fn=chat_with_groq,
inputs="text",
outputs="text",
title="Groq Chatbot",
description="A chatbot powered by Groq API and Gradio."
)
# Launch locally
if __name__ == "__main__":
iface.launch()