import streamlit as st
import ollama

def get_llama_response(user_input):
    try:
        response = ollama.chat(model='llama2', messages=[{'role': 'user', 'content': user_input}])
        return response['message']['content']
    except Exception as e:
        st.error(f"An error occurred: {str(e)}")
        return "Sorry, I couldn't process your request at the moment. Please try again later."

def main():
    st.title("Llama Chatbot")

    user_input = st.text_area("You:", placeholder="Enter your message here...")

    if st.button("Ask"):
        with st.spinner("Thinking..."):
            response = get_llama_response(user_input)
        st.success("Llama says:")
        st.write(response)

if __name__ == "__main__":
    main()