File size: 1,435 Bytes
b501887 a4994af | 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 | import streamlit as st
import cohere
from dotenv import load_dotenv
import os
#load_dotenv(dotenv_path="../.env") # Adjust path if needed
load_dotenv()
api_key = os.getenv("cohere_api_key")
if not api_key:
st.error("API Key not found. Check your .env file and path.")
st.stop()
co = cohere.ClientV2(api_key=api_key)
st.title("GENAI SDP Chatbot using Cohere")
if "chat_history" not in st.session_state:
st.session_state.chat_history = []
user_input = st.text_input("You:", key="user_input")
if st.button("Send") and user_input:
st.session_state.chat_history.append({
"role": "user",
"content": [
{
"type": "text",
"text": user_input
}
]
})
response = co.chat(
messages=st.session_state.chat_history,
temperature=0.3,
model="command-a-03-2025",
)
assistant_reply = response.message.content[0].text
st.session_state.chat_history.append({
"role": "assistant",
"content": [
{
"type": "text",
"text": assistant_reply
}
]
})
# Display chat history
for msg in st.session_state.chat_history:
#print("msg##",msg['message']['content'][0]['text'])
role = msg["role"].capitalize()
text = msg["content"][0]["text"]
#llm_output['message']['content'][0]['text']
st.markdown(f"**{role}:** {text}") |