Spaces:
No application file
No application file
File size: 2,291 Bytes
26ce7c2 4874a99 26ce7c2 4874a99 4cad79d 064ac70 26ce7c2 4874a99 26ce7c2 4874a99 26ce7c2 4874a99 26ce7c2 4874a99 26ce7c2 4874a99 26ce7c2 4874a99 26ce7c2 4874a99 26ce7c2 4874a99 26ce7c2 064ac70 26ce7c2 4874a99 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
import os
import streamlit as st
from groq import Groq
# Initialize Groq client
API_KEY = os.getenv("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") # Ensure GROQ_API_KEY is set in your environment variables
if not API_KEY:
st.error("API key is missing. Please set the GROQ_API_KEY environment variable.")
st.stop()
client = Groq(api_key=API_KEY)
# Streamlit app configuration
st.set_page_config(page_title="AI Chatbot", page_icon="🤖", layout="wide")
st.title("AI-Powered Chatbot with Groq")
# Sidebar for user input
st.sidebar.header("Chatbot Settings")
chat_model = st.sidebar.selectbox("Select AI Model", ["llama-3.3-70b-versatile", "llama-2.0-50b-creative"])
# Chat interface
st.write("## Chat with AI")
if "messages" not in st.session_state:
st.session_state.messages = []
# Display conversation history
for msg in st.session_state.messages:
if msg["role"] == "user":
st.write(f"**You:** {msg['content']}")
else:
st.write(f"**AI:** {msg['content']}")
# User input for the chatbot
user_input = st.text_input("Enter your message:", key="user_input")
# On submit, process the input
if st.button("Send"):
if user_input:
# Add user input to the chat history
st.session_state.messages.append({"role": "user", "content": user_input})
# Call Groq API
with st.spinner("Thinking..."):
try:
chat_completion = client.chat.completions.create(
messages=st.session_state.messages,
model=chat_model,
)
ai_response = chat_completion.choices[0].message.content
# Add AI response to the chat history
st.session_state.messages.append({"role": "assistant", "content": ai_response})
# Display the AI's response
st.write(f"**AI:** {ai_response}")
except Exception as e:
st.error(f"Error: {e}")
else:
st.warning("Please enter a message before sending.")
# Clear conversation button
if st.button("Clear Conversation"):
st.session_state.messages = []
# Footer
st.markdown("""
---
Powered by [Groq](https://groq.com) | Deployed on [Hugging Face Spaces](https://huggingface.co/spaces)
""")
|