Update app.py

app.py

@@ -29,6 +29,13 @@ label_mapping = {
 # Store classified article for QA
 context_storage = {"context": "", "bulk_context": "", "num_articles": 0}
 
+# Detect device type
+device_type = st.radio("Are you using a mobile device or a PC?", ["Mobile", "PC"])
+if device_type == "Mobile":
+    st.set_page_config(layout="centered")
+else:
+    st.set_page_config(layout="wide")
+
 # Text Cleaning Functions
 def clean_text(text):
     text = text.lower()
@@ -91,7 +98,6 @@ def chatbot_response(history, user_input, text_input=None, file_input=None):
     return history, answer
 
 # Streamlit App Layout
-st.set_page_config(page_title="News Classifier", page_icon="📰")
 cover_image = Image.open("cover.png")  # Ensure this image exists
 st.image(cover_image, caption="News Classifier 📢", use_container_width=True)
 
@@ -133,3 +139,4 @@ if st.button("✉ Send"):
     for q, a in history:
         st.write(f"*Q:* {q}")
         st.write(f"*A:* {a}")
+