|
import gradio as gr |
|
import openai |
|
import os |
|
|
|
openai.api_key = os.getenv("GROQ_API_KEY") |
|
openai.api_base = "https://api.groq.com/openai/v1" |
|
|
|
def get_groq_response(message): |
|
try: |
|
response = openai.ChatCompletion.create( |
|
model = "llama-3.3-70b-versatile", |
|
messages = [ |
|
{"role": "system", "content": "Precise answer"}, |
|
{"role": "user", "content": message} |
|
] |
|
) |
|
return response.choices[0].message["content"] |
|
except Exception as e: |
|
return f"Error: {str(e)}" |
|
|
|
def chatbot(user_input, history = []): |
|
bot_response = get_groq_response(user_input) |
|
history.append((user_input, bot_response)) |
|
return history, history |
|
|
|
chat_interface = gr.Interface( |
|
fn=chatbot, |
|
inputs= ["text", "state"], |
|
outputs=["chatbot", "state"], |
|
live=False, |
|
title="My Chatbot", |
|
description="Mom: We have ChatGPT at Home, \n ChatGPT at Home:" |
|
) |
|
chat_interface.launch() |
|
|