|
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() |
|
|