| import streamlit as st |
| import requests |
| import os |
|
|
| |
| API_KEY = os.getenv("GROQ_API_KEY_ChatBot") |
|
|
| API_URL = "https://api.groq.com/openai/v1/chat/completions" |
|
|
| st.set_page_config(page_title="Groq AI Chatbot", layout="centered") |
|
|
| st.title("AI Chatbot") |
|
|
| if "messages" not in st.session_state: |
| st.session_state.messages = [] |
|
|
| |
| for msg in st.session_state.messages: |
| with st.chat_message(msg["role"]): |
| st.write(msg["content"]) |
|
|
| def get_response(messages): |
| headers = { |
| "Authorization": f"Bearer {API_KEY}", |
| "Content-Type": "application/json" |
| } |
|
|
| payload = { |
| "model": "llama-3.1-8b-instant", |
| "messages": messages, |
| "temperature": 0.7 |
| } |
|
|
| res = requests.post(API_URL, headers=headers, json=payload) |
|
|
| if res.status_code == 200: |
| return res.json()["choices"][0]["message"]["content"] |
| else: |
| return res.text |
|
|
| user_input = st.chat_input("Type your message...") |
|
|
| if user_input: |
| st.session_state.messages.append({"role": "user", "content": user_input}) |
|
|
| reply = get_response(st.session_state.messages) |
|
|
| st.session_state.messages.append({"role": "assistant", "content": reply}) |
|
|
| st.rerun() |
