Update app.py

app.py

@@ -1,111 +1,71 @@
 import streamlit as st
 import openai
 
-ai_model = "gpt-4-turbo"
-token = 4096
+# Constants
+AI_MODEL = "gpt-4-turbo"
+TOKEN_COUNT = 4096
+MAX_USES = 3
 
 # Set the page title and favicon
 st.set_page_config(page_title="Daily Reminder", page_icon=":bar_chart:")
 
+# Load API key from secrets
 openai.api_key = st.secrets['OPENAI_API_KEY']
-prompt = st.secrets['PROMPT_DAILY_REMINDER']
+prompt_template = st.secrets['PROMPT_DAILY_REMINDER']
 
+# Page title
 st.title('Daily Reminder')
 
+# Initialize or retrieve the usage count from session state
 if 'usage_count' not in st.session_state:
-    st.session_state['usage_count'] = 0  # Usage counter
+    st.session_state['usage_count'] = 0
 
-max_uses = 3
-
-if st.session_state['usage_count'] < max_uses:
-    # Main Contents Start from here -------------------------------
-
-    st.subheader('English')
-    en_input_topic = 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"),
-        index=0,  # Setting a default index if desired, or use None for no default selection
-        key="en_input_topic"
-    )
-
-    en_input_reason = st.text_input(
-        "What's the reason for this feeling? (e.g., 'I lost my motivation')",
-        key="en_input_reason"
-    )
-
-    en_input_plan = st.text_input(
-        "What's your plan for today? (e.g., 'I'm going to meet my friends.')",
-        key="en_input_plan"
-    )
-
-    if st.button("Generate a daily reminder", key="en_generate_quote"):
-        st.session_state['usage_count'] += 1  # Increment the usage counter
-        # Create a prompt based on the user input
-        en_prompt = f"""
-        - Task: {prompt}.
-        - Output language: Japanese.
-        - Feeling: {en_input_topic}.
-        - Reason for the feeling: {en_input_reason}.
-        - Plan for the day: {en_input_plan}.
-        """
-        # Make a request to the API to generate text
-        en_response = openai.ChatCompletion.create(
-            model=ai_model,  # Use the engine of your choice
-            messages=[{"role": "user", "content": en_prompt}],
-            max_tokens=token
-        )
-        st.write(en_response["choices"][0]["message"]["content"])
-
-    st.text(" ")
-    st.text(" ")
-
-
-    st.subheader('日本語')
-    ja_input_topic = st.selectbox(
-        "今日の気分を教えてください?",
-        ("モチベーションを上げたい", "悲しい", "怒っている", "今はリラックスしたい"),
-        index=0,  # Setting a default index if desired, or use None for no default selection
-        key="ja_input_topic"
-    )
-
-    ja_input_reason = st.text_input(
-        "その理由は何ですか？（例：「モチベーションが上がらない」）",
-        key="ja_input_reason"
-    )
-
-    ja_input_plan = st.text_input(
-        "今日の予定は何ですか？（例：「友達と会う予定です」）",
-        key="ja_input_plan"
-    )
-
-    if st.button("今日の名言を生成する", key="ja_generate_quote"):
+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
-        # Create a prompt based on the user input
-        ja_prompt = f"""
-        - Task: {prompt}
-        - Output language: Japanese.
-        - Feeling: {ja_input_topic}.
-        - Reason for the feeling: {ja_input_reason}.
-        - Plan for the day: {ja_input_plan}.
+        reminder_prompt = f"""
+        - Task: {prompt_template}.
+        - Output language: {language}.
+        - Feeling: {feeling}.
+        - Reason for the feeling: {reason}.
+        - Plan for the day: {plan}.
         """
-        # Make a request to the API to generate text
-        ja_response = openai.ChatCompletion.create(
-            model=ai_model,  # Use the engine of your choice
-            messages=[{"role": "user", "content": ja_prompt}],
-            max_tokens=token
-        )
-        st.write(ja_response["choices"][0]["message"]["content"])
-else:
-    st.error("You have reached your maximum usage limit.")
-
-st.markdown(
-"""
-<style>
-/* ---- Padding on header ---- */
-.st-emotion-cache-iiif1v {
-    display: none !important;
-}
-</style>
-""",
-    unsafe_allow_html=True,
-)
+        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
+
+# 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)
+
+# 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)