Update main.py

main.py

@@ -8,7 +8,7 @@ st.set_page_config(page_title="Kurdish Translator", layout="wide")
 
 # Initialize session state
 if "theme" not in st.session_state:
-    st.session_state.theme = "light"
+    st.session_state.theme = "dark"
 if "src_lang" not in st.session_state:
     st.session_state.src_lang = "English"
 if "tgt_lang" not in st.session_state:
@@ -65,6 +65,35 @@ h1, h2, h3, p, label {{
     from {{ transform: translateY(0); }}
     to {{ transform: translateY(-12px); }}
 }}
+.model-loader {{
+    display: flex;
+    flex-direction: column;
+    justify-content: center;
+    align-items: center;
+    height: 60vh;
+}}
+.spinner {{
+    width: 60px;
+    height: 60px;
+    border: 4px solid {card_bg};
+    border-top: 4px solid #4A90E2;
+    border-radius: 50%;
+    animation: spin 1s linear infinite;
+}}
+@keyframes spin {{
+    0% {{ transform: rotate(0deg); }}
+    100% {{ transform: rotate(360deg); }}
+}}
+.loading-text {{
+    margin-top: 20px;
+    font-size: 18px;
+    color: #4A90E2;
+    animation: pulse 1.5s ease-in-out infinite;
+}}
+@keyframes pulse {{
+    0%, 100% {{ opacity: 1; }}
+    50% {{ opacity: 0.5; }}
+}}
 .output-box {{
     padding: 20px;
     border-radius: 12px;
@@ -85,7 +114,24 @@ h1, h2, h3, p, label {{
 def get_model():
     return load_model()
 
-model = get_model()
+# Show custom loader while model is loading
+if "model_loaded" not in st.session_state:
+    loader_container = st.empty()
+    loader_container.markdown(
+        """
+        <div class='model-loader'>
+            <div class='spinner'></div>
+            <div class='loading-text'>Loading translation model...</div>
+        </div>
+        """,
+        unsafe_allow_html=True
+    )
+    model = get_model()
+    st.session_state.model_loaded = True
+    loader_container.empty()
+    st.rerun()
+else:
+    model = get_model()
 
 # -----------------------------
 # CALLBACK FUNCTIONS
@@ -171,4 +217,4 @@ if st.session_state.output_text:
     
     # Copy section
     st.code(st.session_state.output_text, language=None)
-    st.caption("👆 Select and copy the text above")
+    st.caption("👆 Select and copy the text above (Ctrl+C / Cmd+C)")