Update app.py

app.py

@@ -8,14 +8,25 @@ from nltk.stem import WordNetLemmatizer
 from transformers import pipeline
 from PIL import Image
 
-# Set page config FIRST
-device_type = st.radio("Are you using a mobile device or a PC?", ["Mobile", "PC"])
-if device_type == "Mobile":
-    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")
+# 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)
+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"])
+
+# Adjust 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
 
-# Now the rest of the imports and app logic continue
+# Continue with the rest of the code
 nltk.download('stopwords')
 nltk.download('wordnet')
 nltk.download('omw-1.4')