import streamlit as st
from transformers import pipeline

# Load Hugging Face chatbot model
chatbot_model = pipeline("conversational", model="alpindale/goliath-120b")

# Main Page with Chatbot
def main():
    st.title("Virtual Therapist Chatbot")
    st.write("Feel free to chat with our virtual therapist!")

    # User input for the chatbot
    user_input = st.text_input("You: ")

    # Generate response when user enters input
    if user_input:
        response = chatbot_model(user_input, max_length=50, num_return_sequences=1)[0]['generated_text']
        st.text_area("Therapist:", response, height=100)

# Run the main function
if __name__ == "__main__":
    main()