Spaces:
Sleeping
Sleeping
import streamlit as st | |
import openai | |
# Constants | |
AI_MODEL = "gpt-4o" | |
TOKEN_COUNT = 4096 | |
MAX_USES = 3 | |
# Set the page configuration | |
st.set_page_config(page_title="Translation", page_icon=":bar_chart:") | |
# Load the API key from Streamlit's secrets | |
openai.api_key = st.secrets['OPENAI_API_KEY'] | |
# Page title | |
st.title('Translation') | |
# Style adjustments (optional, remove if not needed) | |
st.markdown( | |
""" | |
<style> | |
/* Custom style adjustments */ | |
.st-emotion-cache-iiif1v { display: none !important; } | |
.st-emotion-cache-gh2jqd {padding: 6rem 1rem 0rem;} | |
@media (max-width: 50.5rem) { | |
.st-emotion-cache-gh2jqd { | |
max-width: calc(0rem + 100vw); | |
} | |
} | |
</style> | |
""", | |
unsafe_allow_html=True, | |
) | |
# 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'] = 'Japanese to English' | |
def translate_text(input_text, source_language, target_language): | |
"""Function to handle text translation between two languages.""" | |
if st.session_state['usage_count'] < MAX_USES: | |
st.session_state['usage_count'] += 1 # Increment the counter | |
prompt = f"Translate from {source_language} to {target_language}. Here is the text: {input_text}" | |
with st.spinner(f'Translating from {source_language} to {target_language}... 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 | |
# Determine button text based on current language | |
if st.session_state['language'] == 'Japanese to English': | |
switch_button_text = 'English to Japanese(英語から日本語)' | |
else: | |
switch_button_text = 'Japanese to English(日本語から英語)' | |
# Language switcher button | |
if st.button(switch_button_text): | |
if st.session_state['language'] == 'Japanese to English': | |
st.session_state['language'] = 'English to Japanese' | |
else: | |
st.session_state['language'] = 'Japanese to English' | |
st.experimental_rerun() | |
# Display form based on selected language | |
if st.session_state['language'] == 'Japanese to English': | |
st.subheader('Japanese to English') | |
ja_input = st.text_area("Enter your Japanese text here:", key="ja_input") | |
if st.button("Translate", key="ja_translate"): | |
result = translate_text(ja_input, "Japanese", "English") | |
if result: | |
st.write(result) | |
else: | |
st.subheader('English to Japanese') | |
en_input = st.text_area("Enter your English text here:", key="en_input") | |
if st.button("Translate", key="en_translate"): | |
result = translate_text(en_input, "English", "Japanese") | |
if result: | |
st.write(result) | |