Update app.py

app.py

@@ -11,59 +11,67 @@ prompt = st.secrets['PROMPT_EMAIL_REVIEW_GENERATOR']
 # Title of the page
 st.title('Email Reply Generator')
 
-# Initialize or retrieve the usage count from session state
+# Initialize or retrieve the usage count and language from 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'
 
 max_uses = 3
 
-# Check usage count to limit API calls
-if st.session_state['usage_count'] < max_uses:
-    # Define a function to handle API requests and increase modularity
-    def generate_reply(input_sender, input_subject, input_message,input_reply, input_length, language):
-        st.session_state['usage_count'] += 1
-        with st.spinner('Loading... Please wait.'):
-            user_prompt = f"- Task： {prompt} - Output language: {language}. - Sender：{input_sender}. - Email Subject： {input_subject}. - Email Message： {input_message}. - What you want to say: {input_reply}. - Length:{input_length}"
-            response = openai.ChatCompletion.create(
-                model="gpt-4o",
-                messages=[{"role": "user", "content": user_prompt}],
-                max_tokens=4096
-            )
-            return response["choices"][0]["message"]["content"]
-
-    # English Input Section
-    with st.form(key='en_form'):
-        st.subheader('English')
-        en_input_sender = st.text_input("Sender (e.g., boss, client, Co-worker,etc.)", key="en_input_sender")
-        en_input_subject = st.text_input("Email Subject (e.g., About scheduling a meeting)", key="en_input_subject")
-        en_input_message = st.text_area("Content of the recipient's email: (e.g., I would like to adjust the time for tomorrow's meeting, are you available in the afternoon?)", key="en_input_message")
-        en_input_reply = st.text_input("What you want to say: (e.g., I am available after 2 PM.)", key="en_input_reply")
-        en_input_length = st.text_input("Number of characters: (e.g., Within 100 characters)", key="en_input_length")
-
-
-
-        submit_en = st.form_submit_button("Generate an Email Reply")
-
-
-        if submit_en:
-            result = generate_reply(en_input_sender, en_input_subject, en_input_message,en_input_reply,en_input_length, "English")
-            st.write(result)
-
-    # Japanese Input Section
-    with st.form(key='ja_form'):
-        st.subheader('日本語')
-        ja_input_sender = st.text_input("送信主（例：上司、クライアント、同僚、など）", key="ja_input_sender")
-        ja_input_subject = st.text_input("メールの件名 (例：ミーティングの日程調整について)", key="ja_input_subject")
-        ja_input_message = st.text_area("相手のメールの内容： (例：明日、ミーティングの時間を調整したいのですが、午後の時間帯は空いていますか？)", key="ja_input_message")
-        ja_input_reply = st.text_input("伝えたいこと： (例：午後2時以降は空いています。)", key="ja_input_reply")
-        ja_input_length = st.text_input("文字数： (例：100文字以内)", key="ja_input_length")
-
+# Define a function to handle API requests and increase modularity
+def generate_reply(input_sender, input_subject, input_message, input_reply, language):
+    st.session_state['usage_count'] += 1
+    with st.spinner('Loading... Please wait.'):
+        user_prompt = f"- Task: {prompt} - Output language: {language}. - Sender: {input_sender}. - Email Subject: {input_subject}. - Email Message: {input_message}. - What you want to say: {input_reply}."
+        response = openai.ChatCompletion.create(
+            model="gpt-4o",
+            messages=[{"role": "user", "content": user_prompt}],
+            max_tokens=4096
+        )
+        return response["choices"][0]["message"]["content"]
+
+# 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'
 
+# 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()
 
-        submit_ja = st.form_submit_button("Eメールの返信を生成する")
-        if submit_ja:
-            result = generate_reply(ja_input_sender, ja_input_subject, ja_input_message,ja_input_reply,ja_input_length, "Japanese")
-            st.write(result)
+# Check usage count to limit API calls
+if st.session_state['usage_count'] < max_uses:
+    # Display form based on selected language
+    if st.session_state['language'] == 'English':
+        # English Input Section
+        with st.form(key='en_form'):
+            st.subheader('English')
+            en_input_sender = st.selectbox("Sender",("Co-worker", "Boss", "Client", "Friend"),key="en_input_sender")
+            en_input_subject = st.text_input("Email Subject (e.g., About scheduling a meeting)", key="en_input_subject")
+            en_input_message = st.text_area("Content of the recipient's email: (e.g., I would like to adjust the time for tomorrow's meeting, are you available in the afternoon?)", key="en_input_message")
+            en_input_reply = st.text_input("What you want to say: (e.g., I am available after 2 PM.)", key="en_input_reply")
+            submit_en = st.form_submit_button("Generate an Email Reply")
+            if submit_en:
+                result = generate_reply(en_input_sender, en_input_subject, en_input_message, en_input_reply, en_input_length, "English")
+                st.write(result)
+    else:
+        # Japanese Input Section
+        with st.form(key='ja_form'):
+            st.subheader('日本語')
+            ja_input_sender = st.selectbox("送信主",("同僚", "上司", "クライアント", "友達"),key="ja_input_sender")
+            ja_input_subject = st.text_input("メールの件名 (例：ミーティングの日程調整について)", key="ja_input_subject")
+            ja_input_message = st.text_area("相手のメールの内容： (例：明日、ミーティングの時間を調整したいのですが、午後の時間帯は空いていますか？)", key="ja_input_message")
+            ja_input_reply = st.text_input("伝えたいこと： (例：午後2時以降は空いています。)", key="ja_input_reply")
+            submit_ja = st.form_submit_button("Eメールの返信を生成する")
+            if submit_ja:
+                result = generate_reply(ja_input_sender, ja_input_subject, ja_input_message, ja_input_reply, ja_input_length, "Japanese")
+                st.write(result)
 else:
     st.error("You have reached your maximum usage limit.")