Update app.py

app.py

@@ -8,31 +8,34 @@ from nltk.stem import WordNetLemmatizer
 from transformers import pipeline
 from PIL import Image
 
-# JavaScript to detect screen width
-def get_device_type():
-    return st.session_state.get("device_type", None)
-
+# ✅ Set page config as the first Streamlit command (default to wide)
+st.set_page_config(page_title="News Classifier", page_icon="📰", layout="wide")
+
+# ✅ Use JavaScript to detect screen size and update session state
+device_type_script = """
+<script>
+    let screen_width = window.innerWidth;
+    let device_type = screen_width < 768 ? "Mobile" : "PC";
+    let streamlitVar = window.parent;
+    streamlitVar.postMessage({device_type: device_type}, "*");
+</script>
+"""
+st.markdown(device_type_script, unsafe_allow_html=True)
+
+# ✅ Initialize session state
 if "device_type" not in st.session_state:
-    device_type_script = """
-    <script>
-        let screen_width = window.innerWidth;
-        let device_type = screen_width < 768 ? "Mobile" : "PC";
-        window.parent.postMessage(device_type, "*");
-    </script>
-    """
-    st.markdown(device_type_script, unsafe_allow_html=True)
-
-    # Wait for response from JavaScript
-    device_type = st.selectbox("Select your device type", ["Mobile", "PC"])
-    st.session_state.device_type = device_type
-
-# Set the layout only ONCE at the start
-if get_device_type() == "Mobile":
-    st.set_page_config(page_title="News Classifier", page_icon="📰", layout="centered")
+    st.session_state.device_type = "PC"  # Default to PC
+
+# ✅ Allow user to manually select device type (Optional)
+st.session_state.device_type = st.radio("Select your device type", ["Mobile", "PC"], index=0 if st.session_state.device_type == "PC" else 1)
+
+# ✅ Apply UI adjustments based on device type
+if st.session_state.device_type == "Mobile":
+    st.write("📱 Mobile Mode Activated (UI adjusted)")
 else:
-    st.set_page_config(page_title="News Classifier", page_icon="📰", layout="wide")
+    st.write("💻 PC Mode Activated (Full-width UI)")
 
-# Now continue with the rest of the app
+# ✅ Continue with the rest of the app
 nltk.download('stopwords')
 nltk.download('wordnet')
 nltk.download('omw-1.4')