Update app.py

app.py

@@ -1,81 +1,65 @@
 import streamlit as st
 import openai
 
-ai_model = "gpt-4-turbo"
-token = 4096
+# Constants and configurations
+AI_MODEL = "gpt-4-turbo"
+TOKEN_COUNT = 4096
+MAX_USES = 3
 
 # Set the page title and favicon
 st.set_page_config(page_title="Text Correction", page_icon=":bar_chart:")
 
+# Load the API key
 openai.api_key = st.secrets['OPENAI_API_KEY']
-st.title('Sentence Correction')
-
 
+# Page title
+st.title('Sentence Correction')
 
+# Initialize usage count in session state
 if 'usage_count' not in st.session_state:
-    st.session_state['usage_count'] = 0  # Usage counter
-
-
-max_uses = 3
-
-if st.session_state['usage_count'] < max_uses:
-    # Main Contents Start from here -------------------------------
-
-    st.subheader('English')
-    en_input = st.text_area("Enter your English text here:", key="en_input")
-    if st.button("Correct", key="en_correction"):
-        st.session_state['usage_count'] += 1  # Increment the usage counter
-        # Create a prompt based on the user input
-        en_prompt = f"Correct the following English sentence. The output language should be English. Here is the text: {en_input}"
-        # 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 = st.text_area("日本語を入力ください", key="ja_input")
-    if st.button("校正する", key="ja_correction"):
-        st.session_state['usage_count'] += 1  # Increment the usage counter
-        # Create a prompt based on the user input
-        ja_prompt = f"Correct the following Japanese sentence. The output language should be Japanese. Here is the text: {ja_input}"
-        # 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.session_state['usage_count'] = 0
+
+def correct_text(language, input_text):
+    """Generate corrected text using OpenAI API based on the input text and language."""
+    if st.session_state['usage_count'] < MAX_USES:
+        st.session_state['usage_count'] += 1
+        prompt = f"Correct the following {language} sentence. The output language should be {language}. Here is the text: {input_text}"
+        with st.spinner('Loading... Please wait.'):
+            response = openai.ChatCompletion.create(
+                model=AI_MODEL,
+                messages=[{"role": "user", "content": prompt}],
+                max_tokens=TOKEN_COUNT
+            )
+        return response["choices"][0]["message"]["content"]
+    else:
+        st.error("You have reached your maximum usage limit.")
+        return None
+
+# Text correction form for English
+st.subheader('English')
+en_input = st.text_area("Enter your English text here:", key="en_input")
+if st.button("Correct", key="en_correction"):
+    result = correct_text("English", en_input)
+    if result:
+        st.write(result)
+
+st.text(" ")  # Spacing between sections
+
+# Text correction form for Japanese
+st.subheader('日本語')
+ja_input = st.text_area("日本語を入力ください", key="ja_input")
+if st.button("校正する", key="ja_correction"):
+    result = correct_text("Japanese", ja_input)
+    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; }
+</style>
+""",
+    unsafe_allow_html=True,
+)