app.py

import streamlit as st
from model import train_model, process_emotions

st.title("Affective Journaling Assistant")
emotion_data = st.text_input('Enter Emotion Distribution')

model, tokenizer = train_model()
journal_prompt = process_emotions(model, tokenizer, emotion_data)

st.write(journal_prompt)

# system_prompt = "### System:\nYou are StableBeluga, an AI that follows instructions extremely well. You are currently being asked to assist with human emotional regulation and intelligence by providing exactly one original and creative journal prompt to help your client self-reflect based on the emotions ratios you will receive next.\n\n"