Update app.py

app.py

@@ -16,16 +16,17 @@ prompt_template = st.secrets['PROMPT_DAILY_REMINDER']
 # Page title
 st.title('Daily Reminder')
 
-# Initialize or retrieve the usage count from session state
+# 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}.
@@ -42,30 +43,44 @@ def generate_reminder(language, feeling, reason, plan):
         st.error("You have reached your maximum usage limit.")
         return None
 
-# Form for English input
-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)
+# Determine button text based on current language
+if st.session_state['language'] == 'English':
+    switch_button_text = 'Switch Language to Japanese'
+else:
+    switch_button_text = 'Switch Language to English'
 
-# Form for Japanese input
-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)
+# Language switcher button
+if st.button(switch_button_text):
+    if st.session_state['language'] == 'English':
+        st.session_state['language'] = 'Japanese'
+    else:
+        st.session_state['language'] = '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)