import streamlit as st
import cohere

st.title("My First Cohere Chat App on HF Space")

co = cohere.Client("3xjTiuBZlBOnNWZPTAsAVZzUeA9l6Z2i0MZtQlZW")

user_question = st.text_input("Ask me anything!")

if user_question:  # Check if user entered a question
    response = co.generate(
        prompt=user_question,
    )
    st.write(response)  # Display Cohere's response