Update app.py

app.py

@@ -8,23 +8,18 @@ from nltk.stem import WordNetLemmatizer
 from transformers import pipeline
 from PIL import Image
 
-# Move set_page_config to the first line with a default layout
-st.set_page_config(page_title="News Classifier", page_icon="📰", layout="wide")
-
-# Function to detect device type
-def detect_device():
-    user_agent = st.experimental_get_query_params().get("user_agent", [""])[0]
-    if "mobi" in user_agent.lower():
-        return "Mobile"
-    return "PC"
-
-# Ask user for device preference (only if not already set)
+# Check if device type is already stored
 if "device_type" not in st.session_state:
-    st.session_state.device_type = st.radio("Are you using a mobile device or a PC?", ["Mobile", "PC"])
+    # Ask user for device type
+    device_type = st.radio("Are you using a mobile device or a PC?", ["Mobile", "PC"])
+    st.session_state.device_type = device_type
+    st.experimental_rerun()  # Refresh the app to apply layout change
 
-# Adjust layout based on user selection
+# Set layout based on user selection
 if st.session_state.device_type == "Mobile":
-    st.set_page_config(layout="centered")  # Not needed if already set, but here for clarity
+    st.set_page_config(page_title="News Classifier", page_icon="📰", layout="centered")
+else:
+    st.set_page_config(page_title="News Classifier", page_icon="📰", layout="wide")
 
 # Continue with the rest of the code
 nltk.download('stopwords')