Update app.py

app.py

@@ -19,17 +19,19 @@ st.title('Rap Generator')
 # Initialize or retrieve the usage count 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'
 
 def generate_rap(language, topic, occupation, message):
     """Generate a rap based on the user's inputs."""
     if st.session_state['usage_count'] < MAX_USES:
         st.session_state['usage_count'] += 1  # Increment the usage counter
         rap_prompt = f"""
-        - Task： {prompt_template}.
+        - Task: {prompt_template}.
         - Output language: {language}.
-        - Topic：{topic}.
-        - Occupation： {occupation}.
-        - What you want to say： {message}.
+        - Topic: {topic}.
+        - Occupation: {occupation}.
+        - What you want to say: {message}.
         """
         with st.spinner('Generating your rap... Please wait.'):
             response = openai.ChatCompletion.create(
@@ -42,27 +44,39 @@ def generate_rap(language, topic, occupation, message):
         st.error("You have reached your maximum usage limit.")
         return None
 
-# English rap generation form
-st.subheader('English')
-en_topic = st.text_input("Topic (e.g., Sunday)", key="en_topic")
-en_occupation = st.text_input("Your Occupation (e.g., Data Scientist)", key="en_occupation")
-en_message = st.text_input("What do you want to say? (e.g., prepare for tomorrow)", key="en_message")
-if st.button("Generate a rap", key="en_generate_rap"):
-    result = generate_rap("English", en_topic, en_occupation, en_message)
-    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'
 
-st.text(" ")  # Spacing
+# 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()
 
-# Japanese rap generation form
-st.subheader('日本語')
-ja_topic = st.text_input("トピック（例：日曜日）", key="ja_topic")
-ja_occupation = st.text_input("あなたの仕事（例：データサイエンティスト）", key="ja_occupation")
-ja_message = st.text_input("言いたいこと（例：明日に向けて気持ちを準備したい）", key="ja_message")
-if st.button("ラップを生成する", key="ja_generate_rap"):
-    result = generate_rap("Japanese", ja_topic, ja_occupation, ja_message)
-    if result:
-        st.write(result)
+# Display form based on selected language
+if st.session_state['language'] == 'English':
+    st.subheader('English')
+    en_topic = st.text_input("Topic (e.g., Sunday)", key="en_topic")
+    en_occupation = st.text_input("Your Occupation (e.g., Data Scientist)", key="en_occupation")
+    en_message = st.text_input("What do you want to say? (e.g., prepare for tomorrow)", key="en_message")
+    if st.button("Generate a rap", key="en_generate_rap"):
+        result = generate_rap("English", en_topic, en_occupation, en_message)
+        if result:
+            st.write(result)
+else:
+    st.subheader('日本語')
+    ja_topic = st.text_input("トピック（例：日曜日）", key="ja_topic")
+    ja_occupation = st.text_input("あなたの仕事（例：データサイエンティスト）", key="ja_occupation")
+    ja_message = st.text_input("言いたいこと（例：明日に向けて気持ちを準備したい）", key="ja_message")
+    if st.button("ラップを生成する", key="ja_generate_rap"):
+        result = generate_rap("Japanese", ja_topic, ja_occupation, ja_message)
+        if result:
+            st.write(result)
 
 # Style adjustments (optional, remove if not needed)
 st.markdown(
@@ -73,4 +87,4 @@ st.markdown(
 </style>
 """,
     unsafe_allow_html=True,
-)
+)