daily-reminder / app.py
heysho's picture
Update app.py
0d28552 verified
raw
history blame
3.96 kB
import streamlit as st
import openai
# Constants
AI_MODEL = "gpt-4o"
TOKEN_COUNT = 2000
MAX_USES = 3
# Set the page title and favicon
st.set_page_config(page_title="Daily Reminder", page_icon=":bar_chart:")
# Load API key and prompt template from secrets
try:
openai.api_key = st.secrets['OPENAI_API_KEY']
prompt_template = st.secrets['PROMPT_DAILY_REMINDER']
except KeyError:
st.error("API key or prompt template is missing in the secrets.")
st.stop()
# Page title
st.title('Daily Reminder')
# Initialize usage counter and language in session state
if 'usage_count' not in st.session_state:
st.session_state['usage_count'] = 0
if 'language' not in st.session_state:
st.session_state['language'] = 'English'
def generate_reminder(language, feeling, reason, plan):
"""Generates a daily reminder based on the user's mood and plans."""
if st.session_state['usage_count'] < MAX_USES:
st.session_state['usage_count'] += 1 # Increment the usage counter
reminder_prompt = f"""
- Task: {prompt_template}
- Output language: {language}.
- Feeling: {feeling}.
- Reason for the feeling: {reason}.
- Plan for the day: {plan}.
- Explain within 100 words in English, or 150 characters in Japanese.
- Display with only single language. Don't mix Japanese and English.
"""
with st.spinner('Generating your daily reminder...'):
response = openai.ChatCompletion.create(
model=AI_MODEL,
messages=[{"role": "user", "content": reminder_prompt}],
max_tokens=TOKEN_COUNT
)
return response["choices"][0]["message"]["content"]
else:
st.error("You have reached your maximum usage limit.")
return None
# Determine button text based on current language
switch_button_text = 'Switch Language to Japanese' if st.session_state['language'] == 'English' else 'Switch Language to English'
# Language switcher button
if st.button(switch_button_text):
st.session_state['language'] = 'Japanese' if st.session_state['language'] == 'English' else 'English'
st.experimental_rerun()
# Display form based on selected language
if st.session_state['language'] == 'English':
st.subheader('English')
en_feeling = st.selectbox(
"How are you feeling today?",
("I want to be more motivated", "I'm feeling sad", "I'm angry", "I just want to relax today", "I feel excited", "I'm feeling anxious"),
key="en_feeling"
)
en_reason = st.text_input("What's the reason for this feeling?", key="en_reason")
en_plan = st.text_input("What's your plan for today?", key="en_plan")
if st.button("Generate a daily reminder", key="en_generate"):
result = generate_reminder("English", en_feeling, en_reason, en_plan)
if result:
st.write(result)
else:
st.subheader('日本語')
ja_feeling = st.selectbox(
"今日の気分を教えてください?",
("モチベーションを上げたい", "悲しい", "怒っている", "今はリラックスしたい", "ワクワクしている", "不安を感じている"),
key="ja_feeling"
)
ja_reason = st.text_input("その理由は何ですか?", key="ja_reason")
ja_plan = st.text_input("今日の予定は何ですか?", key="ja_plan")
if st.button("今日の名言を生成する", key="ja_generate"):
result = generate_reminder("Japanese", ja_feeling, ja_reason, ja_plan)
if result:
st.write(result)
# Style adjustments (optional, remove if not needed)
st.markdown(
"""
<style>
/* Custom style adjustments */
.st-emotion-cache-iiif1v { display: none !important; }
.st-emotion-cache-13ln4jf {padding: 6rem 1rem 0rem;}
@media (max-width: 50.5rem) {
.st-emotion-cache-13ln4jf {
max-width: calc(0rem + 100vw);
}
}
</style>
""",
unsafe_allow_html=True,
)