Update app.py

app.py

@@ -4,34 +4,37 @@ from transformers import pipeline
 from datetime import datetime
 
 # ================================
-# Cache the translation pipelines
+# Streamlit Page Configuration
+# ================================
+st.set_page_config(
+    page_title="🌐 Multi-Language Translator",
+    layout="centered",
+    initial_sidebar_state="auto",
+)
+
+# ================================
+# Cache the Translation Pipelines
 # ================================
 @st.cache_resource
 def load_translation_pipelines():
     """
     Load and cache translation pipelines to avoid reloading on every interaction.
     """
-    try:
-        enja = pipeline("translation", model="staka/fugumt-en-ja")
-        jaen = pipeline("translation", model="staka/fugumt-ja-en")
-        zhja = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-zh-ja")
-        return {'enja': enja, 'jaen': jaen, 'zhja': zhja}
-    except Exception as e:
-        st.error(f"Error loading translation models: {e}")
-        return {}
+    enja = pipeline("translation", model="staka/fugumt-en-ja")
+    jaen = pipeline("translation", model="staka/fugumt-ja-en")
+    zhja = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-zh-ja")
+    return {'enja': enja, 'jaen': jaen, 'zhja': zhja}
 
 # Load the translation models
-session_models = load_translation_pipelines()
+try:
+    session_models = load_translation_pipelines()
+except Exception as e:
+    st.error(f"Error loading translation models: {e}")
+    session_models = {}
 
 # ================================
 # Streamlit Application Layout
 # ================================
-st.set_page_config(
-    page_title="Multi-Language Translator",
-    layout="centered",
-    initial_sidebar_state="auto",
-)
-
 st.title("🌐 Multi-Language Translator")
 
 # Initialize session state for CSV creation flag