| import streamlit as st |
| from transformers import pipeline |
|
|
| |
| def generate_message(): |
| return "Hello, this is a generated message!" |
|
|
| |
| def load_chatbot_model(): |
| pipe = pipeline("conversational", model="alpindale/goliath-120b") |
| return pipeline |
| |
|
|
|
|
| |
|
|
| |
| def main(): |
| st.title("Main Page with Chatbot") |
| st.write("Welcome to the Main Page! Type your message to chat with our virtual therapist.") |
|
|
| |
| chatbot_model = load_chatbot_model() |
|
|
| |
| user_input = st.text_input("You: ") |
|
|
| |
| 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) |
|
|
| |
| if st.button("Open New Page"): |
| open_new_page() |
|
|
| |
| if st.button("Generate Message"): |
| generate_and_show_message() |
|
|
| |
| def open_new_page(): |
| st.title("New Page") |
| st.write("This is the New Page!") |
|
|
| |
| def generate_and_show_message(): |
| message = generate_message() |
| st.title("Generated Message") |
| st.write(f"This message was generated: {message}") |
|
|
| if __name__ == "__main__": |
| main() |
|
|